From a7b87786f79495a36cae5afabd858f63d896338f Mon Sep 17 00:00:00 2001 From: MoonTestUse1 Date: Sat, 28 Dec 2024 07:28:20 +0600 Subject: [PATCH] =?UTF-8?q?=D0=98=D0=BD=D1=82=D0=B5=D0=B3=D1=80=D0=B0?= =?UTF-8?q?=D1=86=D0=B8=D1=8F=20=D0=B1=D0=B0=D0=B7=D1=8B=20=D0=B4=D0=B0?= =?UTF-8?q?=D0=BD=D0=BD=D1=8B=D1=85=20Postgresql=20v7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/alembic/versions/initial_migration.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) diff --git a/backend/alembic/versions/initial_migration.py b/backend/alembic/versions/initial_migration.py index 26f2061..fab8f5f 100644 --- a/backend/alembic/versions/initial_migration.py +++ b/backend/alembic/versions/initial_migration.py @@ -14,6 +14,14 @@ branch_labels = None depends_on = None def upgrade() -> None: + # Create RequestStatus enum type + request_status = sa.Enum(RequestStatus, name='requeststatus') + request_status.create(op.get_bind(), checkfirst=True) + + # Create RequestPriority enum type + request_priority = sa.Enum(RequestPriority, name='requestpriority') + request_priority.create(op.get_bind(), checkfirst=True) + # Create employees table op.create_table( 'employees', @@ -36,8 +44,8 @@ def upgrade() -> None: sa.Column('employee_id', sa.Integer(), nullable=True), sa.Column('department', sa.String(), nullable=False), sa.Column('request_type', sa.String(), nullable=False), - sa.Column('priority', sa.Enum(RequestPriority), nullable=False), - sa.Column('status', sa.Enum(RequestStatus), nullable=False, server_default='new'), + sa.Column('priority', 'requestpriority', nullable=False), + sa.Column('status', 'requeststatus', nullable=False, server_default=sa.text("'NEW'")), sa.Column('description', sa.String(), nullable=True), sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()')), sa.ForeignKeyConstraint(['employee_id'], ['employees.id'], ), @@ -47,4 +55,8 @@ def upgrade() -> None: def downgrade() -> None: op.drop_table('requests') - op.drop_table('employees') \ No newline at end of file + op.drop_table('employees') + + # Drop enum types + sa.Enum(RequestStatus, name='requeststatus').drop(op.get_bind(), checkfirst=True) + sa.Enum(RequestPriority, name='requestpriority').drop(op.get_bind(), checkfirst=True) \ No newline at end of file