mirror of
https://gitlab.com/MoonTestUse1/AdministrationItDepartmens.git
synced 2025-08-14 00:25:46 +02:00
base6
This commit is contained in:
@@ -7,7 +7,6 @@ Create Date: 2024-01-03 20:45:00.000000
|
|||||||
"""
|
"""
|
||||||
from alembic import op
|
from alembic import op
|
||||||
import sqlalchemy as sa
|
import sqlalchemy as sa
|
||||||
from sqlalchemy.dialects import postgresql
|
|
||||||
|
|
||||||
# revision identifiers, used by Alembic.
|
# revision identifiers, used by Alembic.
|
||||||
revision = 'initial_schema'
|
revision = 'initial_schema'
|
||||||
@@ -16,23 +15,6 @@ branch_labels = None
|
|||||||
depends_on = None
|
depends_on = None
|
||||||
|
|
||||||
def upgrade() -> None:
|
def upgrade() -> None:
|
||||||
# Создаем enum типы с проверкой существования
|
|
||||||
connection = op.get_bind()
|
|
||||||
|
|
||||||
# Проверяем существование типа requeststatus
|
|
||||||
result = connection.execute("SELECT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'requeststatus')")
|
|
||||||
exists = result.scalar()
|
|
||||||
if not exists:
|
|
||||||
request_status = postgresql.ENUM('new', 'in_progress', 'completed', 'rejected', name='requeststatus')
|
|
||||||
request_status.create(connection)
|
|
||||||
|
|
||||||
# Проверяем существование типа requestpriority
|
|
||||||
result = connection.execute("SELECT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'requestpriority')")
|
|
||||||
exists = result.scalar()
|
|
||||||
if not exists:
|
|
||||||
request_priority = postgresql.ENUM('low', 'medium', 'high', name='requestpriority')
|
|
||||||
request_priority.create(connection)
|
|
||||||
|
|
||||||
# Создаем таблицу employees
|
# Создаем таблицу employees
|
||||||
op.create_table(
|
op.create_table(
|
||||||
'employees',
|
'employees',
|
||||||
@@ -51,8 +33,8 @@ def upgrade() -> None:
|
|||||||
sa.Column('id', sa.Integer(), nullable=False),
|
sa.Column('id', sa.Integer(), nullable=False),
|
||||||
sa.Column('title', sa.String(), nullable=False),
|
sa.Column('title', sa.String(), nullable=False),
|
||||||
sa.Column('description', sa.String(), nullable=False),
|
sa.Column('description', sa.String(), nullable=False),
|
||||||
sa.Column('status', sa.Enum('new', 'in_progress', 'completed', 'rejected', name='requeststatus'), nullable=False),
|
sa.Column('status', sa.String(), nullable=False),
|
||||||
sa.Column('priority', sa.Enum('low', 'medium', 'high', name='requestpriority'), nullable=False),
|
sa.Column('priority', sa.String(), nullable=False),
|
||||||
sa.Column('employee_id', sa.Integer(), nullable=False),
|
sa.Column('employee_id', sa.Integer(), nullable=False),
|
||||||
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
||||||
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=True),
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=True),
|
||||||
@@ -90,8 +72,4 @@ def downgrade() -> None:
|
|||||||
# Удаляем таблицы
|
# Удаляем таблицы
|
||||||
op.drop_table('tokens')
|
op.drop_table('tokens')
|
||||||
op.drop_table('requests')
|
op.drop_table('requests')
|
||||||
op.drop_table('employees')
|
op.drop_table('employees')
|
||||||
|
|
||||||
# Удаляем enum типы
|
|
||||||
op.execute('DROP TYPE IF EXISTS requeststatus')
|
|
||||||
op.execute('DROP TYPE IF EXISTS requestpriority')
|
|
||||||
Reference in New Issue
Block a user