voussoir/etiquette
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add method get_author to all objects.

Browse source
This commit is contained in:
voussoir 2018-03-18 15:58:51 -07:00
parent dffde094e8
commit 60049c777f
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
etiquette/objects.py
View file

@ -46,6 +46,14 @@ class ObjectBase:
def __hash__(self): def __hash__(self):
return hash(self.id) return hash(self.id)
def get_author(self):
'''
Return the User who created this object, or None if it is unassigned.
'''
if self.author_id is None:
return None
return self.photodb.get_user(id=self.author_id)
class GroupableMixin: class GroupableMixin:
group_getter = None group_getter = None
@ -788,14 +796,6 @@ class Photo(ObjectBase):
self.__reinit__() self.__reinit__()
return self.thumbnail return self.thumbnail
def get_author(self):
'''
Return the User object who owns this photo, or None if it is unassigned.
'''
if self.author_id is None:
return None
return self.photodb.get_user(id=self.author_id)
def get_containing_albums(self): def get_containing_albums(self):
''' '''
Return the albums of which this photo is a member. Return the albums of which this photo is a member.