diff options
author | René 'Necoro' Neumann <necoro@necoro.net> | 2017-03-12 13:38:51 +0100 |
---|---|---|
committer | René 'Necoro' Neumann <necoro@necoro.net> | 2017-03-12 13:38:51 +0100 |
commit | 01a80c04fa72da63e6cd4f1973d299479fbe5566 (patch) | |
tree | 9cf0cbfaf01900d8f0408457bf64bcb0cf41aef6 /archivist | |
parent | 3392f9ab0a0e2c2a5f2bf2db79973ea9a5110593 (diff) | |
download | archivist-01a80c04fa72da63e6cd4f1973d299479fbe5566.tar.gz archivist-01a80c04fa72da63e6cd4f1973d299479fbe5566.tar.bz2 archivist-01a80c04fa72da63e6cd4f1973d299479fbe5566.zip |
Misc fixes
Diffstat (limited to '')
-rw-r--r-- | archivist/model.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/archivist/model.py b/archivist/model.py index e672d29..5af743b 100644 --- a/archivist/model.py +++ b/archivist/model.py @@ -39,16 +39,16 @@ class BaseModel(Model): @classmethod def create_or_get(cls, **kwargs): - if not __keys__: + if not cls.__keys__: raise ValueError("create_or_get not supported on %s" % cls.__name__) - for key in __keys__: + for key in cls.__keys__: if key not in kwargs: raise ValueError("required key '%s' missing" % key) try: return cls.create(**kwargs), True except IntegrityError: - sel = {k:kwargs[k] for k in __keys__} + sel = {k:kwargs[k] for k in cls.__keys__} return cls.get(**sel), False class Meta: @@ -151,6 +151,9 @@ class DocumentTag(BaseModel): class Meta: primary_key = CompositeKey('document', 'tag') + def __repr__(self): + return "<%s %d --> %d>" % (self.__class__.__name__, self.document_id, self.tag_id) + @table class TagImplications(BaseModel): tag = ForeignKeyField(Tag, related_name = 'implications') |