Add timestamps to workspace

alembic/versions/a2b499a1dd62_workspace_timestamps.py

@@ -0,0 +1,30 @@
+"""workspace timestamps
+
+Revision ID: a2b499a1dd62
+Revises: f549c7cee17c
+Create Date: 2018-08-17 10:43:13.165829
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'a2b499a1dd62'
+down_revision = 'f549c7cee17c'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('workspaces', sa.Column('time_created', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False))
+    op.add_column('workspaces', sa.Column('time_updated', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False))
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('workspaces', 'time_updated')
+    op.drop_column('workspaces', 'time_created')
+    # ### end Alembic commands ###

atst/models/mixins.py

@@ -0,0 +1,7 @@
+from sqlalchemy import Column, func, TIMESTAMP
+
+
+class TimestampsMixin(object):
+    time_created = Column(TIMESTAMP(timezone=True), nullable=False, server_default=func.now())
+    time_updated = Column(TIMESTAMP(timezone=True), nullable=False, server_default=func.now(), onupdate=func.current_timestamp())
+

atst/models/workspace.py

@@ -3,9 +3,10 @@ from sqlalchemy.orm import relationship
 
 from atst.models import Base
 from atst.models.types import Id
+from atst.models.mixins import TimestampsMixin
 
 
-class Workspace(Base):
+class Workspace(Base, TimestampsMixin):
     __tablename__ = "workspaces"
 
     id = Id()