Remnants/AdministrationItDepartmens
1
0
Fork 0
mirror of https://gitlab.com/MoonTestUse1/AdministrationItDepartmens.git synced 2025-08-14 00:25:46 +02:00
Code Activity

base5

Browse Source
This commit is contained in:
MoonTestUse1
2025-01-03 23:37:33 +06:00
parent 17e448ed67
commit 4953152119
1 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
backend/alembic/versions/initial_schema.py
View File

@@ -16,12 +16,22 @@ branch_labels = None
depends_on = None depends_on = None
def upgrade() -> None: def upgrade() -> None:
# Создаем enum типы # Создаем enum типы с проверкой существования
request_status = postgresql.ENUM('new', 'in_progress', 'completed', 'rejected', name='requeststatus') connection = op.get_bind()
request_status.create(op.get_bind())
request_priority = postgresql.ENUM('low', 'medium', 'high', name='requestpriority') # Проверяем существование типа requeststatus
request_priority.create(op.get_bind()) 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(
@@ -83,5 +93,5 @@ def downgrade() -> None:
op.drop_table('employees') op.drop_table('employees')
# Удаляем enum типы # Удаляем enum типы
op.execute('DROP TYPE requeststatus') op.execute('DROP TYPE IF EXISTS requeststatus')
op.execute('DROP TYPE requestpriority') op.execute('DROP TYPE IF EXISTS requestpriority')