author | sylvain.thenault@logilab.fr |
Tue, 12 May 2009 11:56:12 +0200 | |
branch | tls-sprint |
changeset 1739 | 78b0819162a8 |
parent 1726 | 08918409815e |
child 1740 | 2292ae32c98f |
permissions | -rw-r--r-- |
1152 | 1 |
"""relation tags store |
2 |
||
3 |
:organization: Logilab |
|
4 |
:copyright: 2001-2009 LOGILAB S.A. (Paris, FRANCE), all rights reserved. |
|
5 |
:contact: http://www.logilab.fr/ -- mailto:contact@logilab.fr |
|
6 |
""" |
|
7 |
__docformat__ = "restructuredtext en" |
|
1148
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
8 |
|
1533 | 9 |
|
1148
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
10 |
class RelationTags(object): |
1533 | 11 |
"""a tag store for full relation definitions : |
1148
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
12 |
|
1533 | 13 |
(subject type, relation type, object type, tagged) |
1152 | 14 |
|
15 |
allowing to set tags using wildcard (eg '*') as subject type / object type |
|
16 |
||
1533 | 17 |
This class associates a single tag to each key. |
1152 | 18 |
""" |
1739 | 19 |
_allowed_values = None |
20 |
def __init__(self, initfunc=None, allowed_values=None): |
|
1148
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
21 |
self._tagdefs = {} |
1739 | 22 |
if allowed_values is not None: |
23 |
self._allowed_values = allowed_values |
|
24 |
self._initfunc = initfunc |
|
1451 | 25 |
|
1528
864ae7c15ef5
other fixlets
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents:
1451
diff
changeset
|
26 |
def __repr__(self): |
864ae7c15ef5
other fixlets
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents:
1451
diff
changeset
|
27 |
return repr(self._tagdefs) |
864ae7c15ef5
other fixlets
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents:
1451
diff
changeset
|
28 |
|
1533 | 29 |
# dict compat |
30 |
def __getitem__(self, key): |
|
31 |
return self.get(*key) |
|
32 |
__contains__ = __getitem__ |
|
1451 | 33 |
|
1739 | 34 |
def _get_keys(self, stype, rtype, otype, tagged): |
1533 | 35 |
keys = [(rtype, tagged, '*', '*'), |
36 |
(rtype, tagged, '*', otype), |
|
37 |
(rtype, tagged, stype, '*'), |
|
38 |
(rtype, tagged, stype, otype)] |
|
39 |
if stype == '*' or otype == '*': |
|
40 |
keys.remove((rtype, tagged, '*', '*')) |
|
41 |
if stype == '*': |
|
42 |
keys.remove((rtype, tagged, '*', otype)) |
|
43 |
if otype == '*': |
|
44 |
keys.remove((rtype, tagged, stype, '*')) |
|
45 |
return keys |
|
1451 | 46 |
|
1739 | 47 |
def init(self, schema): |
48 |
# XXX check existing keys against schema |
|
49 |
for rtype, tagged, stype, otype in self._tagdefs: |
|
50 |
assert rtype in schema |
|
51 |
if stype != '*': |
|
52 |
assert stype in schema |
|
53 |
if otype != '*': |
|
54 |
assert otype in schema |
|
55 |
if self._initfunc is not None: |
|
56 |
for eschema in schema.entities(): |
|
57 |
for rschema, tschemas, role in eschema.relation_definitions(True): |
|
58 |
for tschema in tschemas: |
|
59 |
if role == 'subject': |
|
60 |
stype, otype = eschema, tschema |
|
61 |
else: |
|
62 |
stype, otype = tschema, eschema |
|
63 |
self._initfunc(stype, rtype, otype, role) |
|
64 |
||
65 |
# rtag declaration api #################################################### |
|
66 |
||
67 |
def tag_attribute(self, key, tag): |
|
68 |
key = list(key) |
|
69 |
key.append('*') |
|
70 |
self.tag_subject_of(key, tag) |
|
1548
bd225e776739
new tag_attribute convenience method
sylvain.thenault@logilab.fr
parents:
1533
diff
changeset
|
71 |
|
1739 | 72 |
def tag_subject_of(self, key, tag): |
73 |
key = list(key) |
|
74 |
key.append('subject') |
|
75 |
self.tag_relation(key, tag) |
|
76 |
||
77 |
def tag_object_of(self, key, tag): |
|
78 |
key = list(key) |
|
79 |
key.append('object') |
|
80 |
self.tag_relation(key, tag) |
|
1533 | 81 |
|
1739 | 82 |
def tag_relation(self, key, tag): |
83 |
#if isinstance(key, basestring): |
|
84 |
# stype, rtype, otype = key.split() |
|
85 |
#else: |
|
86 |
stype, rtype, otype, tagged = [str(k) for k in key] |
|
87 |
if self._allowed_values is not None: |
|
88 |
assert tag in self._allowed_values |
|
89 |
self._tagdefs[(rtype, tagged, stype, otype)] = tag |
|
1533 | 90 |
|
1739 | 91 |
# rtag runtime api ######################################################## |
92 |
||
93 |
def del_rtag(self, stype, rtype, otype, tagged): |
|
94 |
del self._tagdefs[(rtype, tagged, stype, otype)] |
|
95 |
||
96 |
def get(self, stype, rtype, otype, tagged): |
|
1721
694f6a50e138
final rtags api (eventually :$)
sylvain.thenault@logilab.fr
parents:
1548
diff
changeset
|
97 |
for key in reversed(self._get_keys(stype, rtype, otype, tagged)): |
1148
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
98 |
try: |
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
99 |
return self._tagdefs[key] |
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
100 |
except KeyError: |
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
101 |
continue |
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
102 |
return None |
1451 | 103 |
|
1721
694f6a50e138
final rtags api (eventually :$)
sylvain.thenault@logilab.fr
parents:
1548
diff
changeset
|
104 |
def etype_get(self, etype, rtype, role, ttype='*'): |
694f6a50e138
final rtags api (eventually :$)
sylvain.thenault@logilab.fr
parents:
1548
diff
changeset
|
105 |
if role == 'subject': |
694f6a50e138
final rtags api (eventually :$)
sylvain.thenault@logilab.fr
parents:
1548
diff
changeset
|
106 |
return self.get(etype, rtype, ttype, role) |
694f6a50e138
final rtags api (eventually :$)
sylvain.thenault@logilab.fr
parents:
1548
diff
changeset
|
107 |
return self.get(ttype, rtype, etype, role) |
1533 | 108 |
|
109 |
||
1451 | 110 |
|
1533 | 111 |
class RelationTagsSet(RelationTags): |
112 |
"""This class associates a set of tags to each key.""" |
|
113 |
||
1739 | 114 |
def tag_relation(self, key, tag): |
115 |
stype, rtype, otype, tagged = [str(k) for k in key] |
|
1533 | 116 |
rtags = self._tagdefs.setdefault((rtype, tagged, stype, otype), set()) |
1148
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
117 |
rtags.add(tag) |
1451 | 118 |
|
1739 | 119 |
def get(self, stype, rtype, otype, tagged): |
1148
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
120 |
rtags = set() |
1721
694f6a50e138
final rtags api (eventually :$)
sylvain.thenault@logilab.fr
parents:
1548
diff
changeset
|
121 |
for key in self._get_keys(stype, rtype, otype, tagged): |
1148
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
122 |
try: |
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
123 |
rtags.update(self._tagdefs[key]) |
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
124 |
except KeyError: |
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
125 |
continue |
55a8238f8f7c
keep notion of relation tags, tough with simplier implementation and usage
sylvain.thenault@logilab.fr
parents:
diff
changeset
|
126 |
return rtags |
1739 | 127 |
|
128 |
||
129 |
class RelationTagsBool(RelationTags): |
|
130 |
_allowed_values = frozenset((True, False)) |
|
131 |
||
132 |
def tag_subject_of(self, key, tag=True): |
|
133 |
super(RelationTagsBool, self).tag_subject_of(key, tag) |
|
134 |
||
135 |
def tag_object_of(self, key, tag=True): |
|
136 |
super(RelationTagsBool, self).tag_object_of(key, tag) |
|
137 |
||
138 |
def tag_attribute(self, key, tag=True): |
|
139 |
super(RelationTagsBool, self).tag_attribute(key, tag) |