1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
VERSION = (0, 3, 1, "final", 0)
def get_version():
if VERSION[3] == "final":
return "%s.%s.%s" % (VERSION[0], VERSION[1], VERSION[2])
elif VERSION[3] == "dev":
if VERSION[2] == 0:
return "%s.%s.%s%s" % (VERSION[0], VERSION[1], VERSION[3], VERSION[4])
return "%s.%s.%s.%s%s" % (VERSION[0], VERSION[1], VERSION[2], VERSION[3], VERSION[4])
else:
return "%s.%s.%s%s" % (VERSION[0], VERSION[1], VERSION[2], VERSION[3])
__version__ = get_version()
class AlreadyRegistered(Exception):
"""
An attempt was made to register a model more than once.
"""
pass
registry = []
def register(model, tag_descriptor_attr='tags',
tagged_item_manager_attr='tagged'):
"""
Sets the given model class up for working with tags.
"""
from tagging.managers import ModelTaggedItemManager, TagDescriptor
if model in registry:
raise AlreadyRegistered("The model '%s' has already been "
"registered." % model._meta.object_name)
if hasattr(model, tag_descriptor_attr):
raise AttributeError("'%s' already has an attribute '%s'. You must "
"provide a custom tag_descriptor_attr to register." % (
model._meta.object_name,
tag_descriptor_attr,
)
)
if hasattr(model, tagged_item_manager_attr):
raise AttributeError("'%s' already has an attribute '%s'. You must "
"provide a custom tagged_item_manager_attr to register." % (
model._meta.object_name,
tagged_item_manager_attr,
)
)
# Add tag descriptor
setattr(model, tag_descriptor_attr, TagDescriptor())
# Add custom manager
ModelTaggedItemManager().contribute_to_class(model, tagged_item_manager_attr)
# Finally register in registry
registry.append(model)
|
