from ..base import NodeAttribute, Node, CollectionAttribute, CollectionNode
from ...exceptions import SpeakerAttributeError, DiscourseAttributeError
[docs]
class SpeakerAttribute(NodeAttribute):
def __init__(self, node, label):
super(SpeakerAttribute, self).__init__(node, label)
def __repr__(self):
return '<SpeakerAttribute "{}">'.format(str(self))
[docs]
class SpeakerNode(Node):
def __init__(self, corpus=None, hierarchy=None):
super(SpeakerNode, self). __init__('Speaker', corpus, hierarchy)
def __repr__(self):
return '<SpeakerNode "{}">'.format(str(self))
def __getattr__(self, key):
if key == 'discourses':
return DiscourseCollectionNode(self)
if not self.hierarchy.has_speaker_property(key):
properties = [x[0] for x in self.hierarchy.speaker_properties]
raise SpeakerAttributeError(
'Speakers do not have a \'{}\' property (available: {}).'.format(key, ', '.join(properties)))
return SpeakerAttribute(self, key)
[docs]
class DiscourseNode(Node):
def __init__(self, corpus=None, hierarchy=None):
super(DiscourseNode, self). __init__('Discourse', corpus, hierarchy)
def __repr__(self):
return '<DiscourseNode "{}">'.format(str(self))
[docs]
class DiscourseCollectionNode(CollectionNode):
non_optional = True
subquery_match_template = '''({anchor_node_alias})-[speaks:speaks_in]->({def_collection_alias})'''
def __repr__(self):
return '<DiscourseCollectionNode "{}">'.format(str(self))
def __init__(self, speaker_node):
d = DiscourseNode(speaker_node.corpus, speaker_node.hierarchy)
super(DiscourseCollectionNode, self). __init__(speaker_node, d)
@property
def speaks_in_alias(self):
return 'speaks'
@property
def with_pre_collection(self):
return ', '.join([self.speaks_in_alias])
def with_statement(self):
withs = [self.collect_template.format(a=self.collection_alias),
self.collect_template.format(a=self.speaks_in_alias)
]
return ', '.join(withs)
@property
def withs(self):
withs = [self.collection_alias, self.speaks_in_alias]
return withs
def __getattr__(self, key):
if key.startswith('channel'):
return ChannelAttribute(self)
if not self.hierarchy.has_discourse_property(key):
properties = [x[0] for x in self.hierarchy.discourse_properties]
raise DiscourseAttributeError(
'Discourses do not have a \'{}\' property (available: {}).'.format(key, ', '.join(properties)))
return CollectionAttribute(self, key)
@property
def collection_alias(self):
return self.collected_node.alias
@property
def def_collection_alias(self):
return self.collected_node.alias
[docs]
class ChannelAttribute(CollectionAttribute):
def __init__(self, node):
super(ChannelAttribute, self).__init__(node, 'channel')
def __repr__(self):
return '<ChannelAttribute "{}">'.format(str(self))
def for_cypher(self):
return '{}.{}'.format(self.node.speaks_in_alias, 'channel')
def for_return(self):
return self.return_template.format(alias=self.node.speaks_in_alias, property=self.label)