clipperz/password-manager-mirror
1
0
Fork 0
mirror of http://git.whoc.org.uk/git/password-manager.git synced 2024-11-14 18:39:01 +01:00
Code Issues Releases Wiki Activity

Fix flask model initialization

Browse Source 
This fixes Records and RecordVersions to populate parent objects as part of
the initialization
This commit is contained in:
jokajak 2016-04-24 12:17:22 -04:00
parent 3574eacb25
commit e4e4de7c1d
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
backend/flask/src/clipperz/models.py
View File

@ -87,8 +87,11 @@ class RecordVersion(db.Model):
order_by=id,
cascade='all,delete'))
def __init__(self):
def __init__(self, record=None):
"""Initialize a record version."""
if record:
self.record = record
self.record_id = record.id
self.creation_date = datetime.datetime.utcnow()
def update(self, someData):
@ -128,8 +131,11 @@ class Record(db.Model):
uselist=False,
cascade='save-update, merge, delete, delete-orphan')
def __init__(self):
def __init__(self, user=None):
"""Initialize a record."""
if user:
self.user_id = user.id
self.version = 1
self.creation_date = datetime.datetime.utcnow()
def update(self, data, record_version):
@ -139,10 +145,7 @@ class Record(db.Model):
self.api_version = data['version']
self.update_date = datetime.datetime.now()
self.current_record_version = record_version
if self.version:
self.version += 1
else:
self.version = 1
self.version += 1
# ------------------------------------------------------------------------------