Add superinit to the objects.

master
voussoir 2017-04-22 21:16:30 -07:00
parent 72aaf3460d
commit 82ca1a08da
1 changed files with 9 additions and 5 deletions

View File

@ -13,6 +13,10 @@ from voussoirkit import spinal
class ObjectBase: class ObjectBase:
def __init__(self, photodb):
super().__init__()
self.photodb = photodb
def __eq__(self, other): def __eq__(self, other):
return ( return (
isinstance(other, type(self)) and isinstance(other, type(self)) and
@ -202,7 +206,7 @@ class Album(ObjectBase, GroupableMixin):
group_table = 'album_group_rel' group_table = 'album_group_rel'
def __init__(self, photodb, db_row): def __init__(self, photodb, db_row):
self.photodb = photodb super().__init__(photodb)
if isinstance(db_row, (list, tuple)): if isinstance(db_row, (list, tuple)):
db_row = helpers.parallel_to_dict(constants.SQL_ALBUM_COLUMNS, db_row) db_row = helpers.parallel_to_dict(constants.SQL_ALBUM_COLUMNS, db_row)
self.id = db_row['id'] self.id = db_row['id']
@ -360,7 +364,7 @@ class Album(ObjectBase, GroupableMixin):
class Bookmark(ObjectBase): class Bookmark(ObjectBase):
def __init__(self, photodb, db_row): def __init__(self, photodb, db_row):
self.photodb = photodb super().__init__(photodb)
if isinstance(db_row, (list, tuple)): if isinstance(db_row, (list, tuple)):
db_row = helpers.parallel_to_dict(constants.SQL_BOOKMARK_COLUMNS, db_row) db_row = helpers.parallel_to_dict(constants.SQL_BOOKMARK_COLUMNS, db_row)
@ -405,7 +409,7 @@ class Photo(ObjectBase):
Photos are not the actual image data, just the database entry. Photos are not the actual image data, just the database entry.
''' '''
def __init__(self, photodb, db_row): def __init__(self, photodb, db_row):
self.photodb = photodb super().__init__(photodb)
if isinstance(db_row, (list, tuple)): if isinstance(db_row, (list, tuple)):
db_row = helpers.parallel_to_dict(constants.SQL_PHOTO_COLUMNS, db_row) db_row = helpers.parallel_to_dict(constants.SQL_PHOTO_COLUMNS, db_row)
@ -882,7 +886,7 @@ class Tag(ObjectBase, GroupableMixin):
group_table = 'tag_group_rel' group_table = 'tag_group_rel'
def __init__(self, photodb, db_row): def __init__(self, photodb, db_row):
self.photodb = photodb super().__init__(photodb)
if isinstance(db_row, (list, tuple)): if isinstance(db_row, (list, tuple)):
db_row = helpers.parallel_to_dict(constants.SQL_TAG_COLUMNS, db_row) db_row = helpers.parallel_to_dict(constants.SQL_TAG_COLUMNS, db_row)
self.id = db_row['id'] self.id = db_row['id']
@ -1074,7 +1078,7 @@ class User(ObjectBase):
A dear friend of ours. A dear friend of ours.
''' '''
def __init__(self, photodb, db_row): def __init__(self, photodb, db_row):
self.photodb = photodb super().__init__(photodb)
if isinstance(db_row, (list, tuple)): if isinstance(db_row, (list, tuple)):
db_row = helpers.parallel_to_dict(constants.SQL_USER_COLUMNS, db_row) db_row = helpers.parallel_to_dict(constants.SQL_USER_COLUMNS, db_row)
self.id = db_row['id'] self.id = db_row['id']