pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
richard-dds 2018-09-21 11:35:32 -04:00
parent 112f85a10e
commit 5a4cbb56bb
3 changed files with 8 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
atst/domain/auth.py
View File

@ -34,6 +34,7 @@ def get_current_user():
else: else:
return False return False
def logout(): def logout():
if session.get("user_id"): if session.get("user_id"):
del (session["user_id"]) del (session["user_id"])

5
atst/models/audit_event.py
View File

@ -38,7 +38,4 @@ class AuditEvent(Base, TimestampsMixin):
def save(self, connection): def save(self, connection):
attrs = inspect(self).dict attrs = inspect(self).dict
connection.execute( connection.execute(self.__table__.insert(), **attrs)
self.__table__.insert(),
**attrs
)

13
atst/models/mixins/auditable.py
View File

@ -11,18 +11,17 @@ ACTION_DELETE = "delete"
def getattr_path(obj, path, default=None): def getattr_path(obj, path, default=None):
_obj = obj _obj = obj
for item in path.split('.'): for item in path.split("."):
_obj = getattr(_obj, item, default) _obj = getattr(_obj, item, default)
return _obj return _obj
def camel_to_snake(camel_cased): def camel_to_snake(camel_cased):
s1 = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', camel_cased) s1 = re.sub("(.)([A-Z][a-z]+)", r"\1_\2", camel_cased)
return re.sub('([a-z0-9])([A-Z])', r'\1_\2', s1).lower() return re.sub("([a-z0-9])([A-Z])", r"\1_\2", s1).lower()
class AuditableMixin(object): class AuditableMixin(object):
@staticmethod @staticmethod
def create_audit_event(connection, resource, action): def create_audit_event(connection, resource, action):
user_id = getattr_path(g, "current_user.id") user_id = getattr_path(g, "current_user.id")
@ -41,9 +40,9 @@ class AuditableMixin(object):
@classmethod @classmethod
def __declare_last__(cls): def __declare_last__(cls):
event.listen(cls, 'after_insert', cls.audit_insert) event.listen(cls, "after_insert", cls.audit_insert)
event.listen(cls, 'after_delete', cls.audit_delete) event.listen(cls, "after_delete", cls.audit_delete)
event.listen(cls, 'after_update', cls.audit_update) event.listen(cls, "after_update", cls.audit_update)
@staticmethod @staticmethod
def audit_insert(mapper, connection, target): def audit_insert(mapper, connection, target):