Update atst to atat

atat/models/audit_event.py

@@ -0,0 +1,56 @@
+from sqlalchemy import String, Column, ForeignKey, inspect
+from sqlalchemy.dialects.postgresql import UUID, JSONB
+from sqlalchemy.orm import relationship
+
+from atat.models.base import Base
+import atat.models.types as types
+from atat.models.mixins.timestamps import TimestampsMixin
+
+
+class AuditEvent(Base, TimestampsMixin):
+    __tablename__ = "audit_events"
+
+    id = types.Id()
+
+    user_id = Column(UUID(as_uuid=True), ForeignKey("users.id"), index=True)
+    user = relationship("User", backref="audit_events")
+
+    portfolio_id = Column(UUID(as_uuid=True), ForeignKey("portfolios.id"), index=True)
+    portfolio = relationship("Portfolio", backref="audit_events")
+
+    application_id = Column(
+        UUID(as_uuid=True), ForeignKey("applications.id"), index=True
+    )
+    application = relationship("Application", backref="audit_events")
+
+    changed_state = Column(JSONB())
+    event_details = Column(JSONB())
+
+    resource_type = Column(String(), nullable=False)
+    resource_id = Column(UUID(as_uuid=True), index=True, nullable=False)
+
+    display_name = Column(String())
+    action = Column(String(), nullable=False)
+
+    @property
+    def log(self):
+        return {
+            "portfolio_id": str(self.portfolio_id),
+            "application_id": str(self.application_id),
+            "changed_state": self.changed_state,
+            "event_details": self.event_details,
+            "resource_type": self.resource_type,
+            "resource_id": str(self.resource_id),
+            "display_name": self.display_name,
+            "action": self.action,
+        }
+
+    def save(self, connection):
+        attrs = inspect(self).dict
+
+        connection.execute(self.__table__.insert(), **attrs)
+
+    def __repr__(self):  # pragma: no cover
+        return "<AuditEvent(name='{}', action='{}', id='{}')>".format(
+            self.display_name, self.action, self.id
+        )