feat: Alembic setup, User and AuditLog models with migration

Co-Authored-By: Claude <noreply@anthropic.com>
2026-07-22 16:25:53 +08:00
parent 83112ed36d
commit 5a69cea3ae
7 changed files with 184 additions and 0 deletions
@@ -0,0 +1,58 @@
+"""create users and audit_logs tables
+
+Revision ID: ab0c56ae4242
+Revises:
+Create Date: 2026-07-22 16:25:22.698619
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'ab0c56ae4242'
+down_revision: Union[str, Sequence[str], None] = None
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    op.create_table(
+        'users',
+        sa.Column('id', sa.String(36), primary_key=True),
+        sa.Column('username', sa.String(50), unique=True, nullable=False, index=True),
+        sa.Column('email', sa.String(255), unique=True, nullable=False),
+        sa.Column('hashed_password', sa.String(255), nullable=False),
+        sa.Column(
+            'role',
+            sa.Enum('owner', 'operator', 'viewer', name='userrole'),
+            nullable=False,
+            server_default='viewer',
+        ),
+        sa.Column('is_active', sa.Boolean, nullable=False, server_default=sa.true()),
+        sa.Column('is_approved', sa.Boolean, nullable=False, server_default=sa.false()),
+        sa.Column('login_attempts', sa.Integer, nullable=False, server_default='0'),
+        sa.Column('locked_until', sa.DateTime, nullable=True),
+        sa.Column('created_at', sa.DateTime, nullable=False, server_default=sa.func.now()),
+        sa.Column('updated_at', sa.DateTime, nullable=False, server_default=sa.func.now()),
+    )
+
+    op.create_table(
+        'audit_logs',
+        sa.Column('id', sa.String(36), primary_key=True),
+        sa.Column('user_id', sa.String(36), nullable=False, index=True),
+        sa.Column('username', sa.String(50), nullable=False),
+        sa.Column('action', sa.String(100), nullable=False, index=True),
+        sa.Column('target', sa.String(255), nullable=False),
+        sa.Column('detail', sa.Text, nullable=True),
+        sa.Column('ip_address', sa.String(45), nullable=True),
+        sa.Column('created_at', sa.DateTime, nullable=False, server_default=sa.func.now()),
+    )
+
+
+def downgrade() -> None:
+    op.drop_table('audit_logs')
+    op.drop_table('users')
+    op.execute('DROP TYPE IF EXISTS userrole')