darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-01 15:46:07 -05:00
Code Issues Packages Projects Releases Wiki Activity

Refactored dao_update_organisation

Browse Source
This commit is contained in:
Ken Tsang
2017-07-11 18:18:23 +01:00
parent 106d59006b
commit 0c77f2d010
3 changed files with 19 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
app/dao/organisations_dao.py
View File

@@ -14,3 +14,10 @@ def dao_get_organisation_by_id(org_id):
@transactional
def dao_create_organisation(organisation):
db.session.add(organisation)
@transactional
def dao_update_organisation(organisation, **kwargs):
for key, value in kwargs.items():
setattr(organisation, key, value)
db.session.add(organisation)

4
app/organisation/rest.py
View File

@@ -4,6 +4,7 @@ from app.dao.organisations_dao import (
dao_create_organisation,
dao_get_organisations,
dao_get_organisation_by_id,
dao_update_organisation
)
from app.errors import register_errors
from app.models import Organisation
@@ -45,7 +46,6 @@ def update_organisation(organisation_id):
validate(data, post_update_organisation_schema)
fetched_organisation = dao_get_organisation_by_id(organisation_id)
for key in data.keys():
setattr(fetched_organisation, key, data[key])
dao_update_organisation(fetched_organisation, **data)
return jsonify(data=fetched_organisation.serialize()), 200