48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
"""create users table
|
|
|
|
Revision ID: a1b2c3d4e5f6
|
|
Revises:
|
|
Create Date: 2023-01-01 12:00:00.000000
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects.postgresql import ENUM
|
|
from datetime import datetime
|
|
import uuid
|
|
|
|
# revision identifiers
|
|
revision = 'a1b2c3d4e5f6'
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
def upgrade():
|
|
user_roles = ENUM('admin', 'lawyer', 'client', name='user_roles')
|
|
user_roles.create(op.get_bind())
|
|
|
|
op.create_table(
|
|
'users',
|
|
sa.Column('id', sa.String(), primary_key=True, default=lambda: str(uuid.uuid4())),
|
|
sa.Column('first_name', sa.String(100), nullable=False),
|
|
sa.Column('last_name', sa.String(100), nullable=False),
|
|
sa.Column('email', sa.String(255), nullable=False, unique=True),
|
|
sa.Column('password', sa.String(255), nullable=False),
|
|
sa.Column('phone_number', sa.String(20), nullable=False),
|
|
sa.Column('role', user_roles, nullable=False),
|
|
sa.Column('bar_number', sa.String(50), nullable=True),
|
|
sa.Column('is_active', sa.Boolean(), default=True),
|
|
sa.Column('address', sa.Text(), nullable=True),
|
|
sa.Column('profile_picture', sa.String(255), nullable=True),
|
|
sa.Column('last_login', sa.DateTime(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('updated_at', sa.DateTime(), server_default=sa.func.now(), onupdate=sa.func.now()),
|
|
)
|
|
|
|
op.create_index('ix_users_email', 'users', ['email'], unique=True)
|
|
op.create_index('ix_users_bar_number', 'users', ['bar_number'], unique=True)
|
|
|
|
def downgrade():
|
|
op.drop_index('ix_users_bar_number', 'users')
|
|
op.drop_index('ix_users_email', 'users')
|
|
op.drop_table('users')
|
|
ENUM(name='user_roles').drop(op.get_bind()) |