diff --git a/atst/domain/auth.py b/atst/domain/auth.py index 0d4844c9..74c6d2d8 100644 --- a/atst/domain/auth.py +++ b/atst/domain/auth.py @@ -34,6 +34,7 @@ def get_current_user(): else: return False + def logout(): if session.get("user_id"): del (session["user_id"]) diff --git a/atst/models/audit_event.py b/atst/models/audit_event.py index 7ed50d5b..52a88cf6 100644 --- a/atst/models/audit_event.py +++ b/atst/models/audit_event.py @@ -38,7 +38,4 @@ class AuditEvent(Base, TimestampsMixin): def save(self, connection): attrs = inspect(self).dict - connection.execute( - self.__table__.insert(), - **attrs - ) + connection.execute(self.__table__.insert(), **attrs) diff --git a/atst/models/mixins/auditable.py b/atst/models/mixins/auditable.py index 6079e6d5..0758fa10 100644 --- a/atst/models/mixins/auditable.py +++ b/atst/models/mixins/auditable.py @@ -11,18 +11,17 @@ ACTION_DELETE = "delete" def getattr_path(obj, path, default=None): _obj = obj - for item in path.split('.'): + for item in path.split("."): _obj = getattr(_obj, item, default) return _obj def camel_to_snake(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() + 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() class AuditableMixin(object): - @staticmethod def create_audit_event(connection, resource, action): user_id = getattr_path(g, "current_user.id") @@ -41,9 +40,9 @@ class AuditableMixin(object): @classmethod def __declare_last__(cls): - event.listen(cls, 'after_insert', cls.audit_insert) - event.listen(cls, 'after_delete', cls.audit_delete) - event.listen(cls, 'after_update', cls.audit_update) + event.listen(cls, "after_insert", cls.audit_insert) + event.listen(cls, "after_delete", cls.audit_delete) + event.listen(cls, "after_update", cls.audit_update) @staticmethod def audit_insert(mapper, connection, target):