Squashed commit of the following:
commit 1eeea9f83ad9230a5c1f7a75662770eaab0df837 Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 21:15:41 2025 +0200 Disable resuming of old time entries. commit 3e3ec2f01cb7943622b819a19179388078ae1315 Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 20:59:19 2025 +0200 Refactor db migrations. commit 15a51a569da36c6b7c9e01ab17b6fdbdee6ad994 Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 19:58:04 2025 +0200 Apply new style for Time Tracking view. commit 77e5278b303e060d2b03853b06277f8aa567ae68 Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 18:06:04 2025 +0200 Allow direct registrations as a Company. commit 188a8772757cbef374243d3a5f29e4440ddecabe Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 18:04:45 2025 +0200 Add email invitation feature. commit d9ebaa02aa01b518960a20dccdd5a327d82f30c6 Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 17:12:32 2025 +0200 Apply common style for Company, User, Team management pages. commit 81149caf4d8fc6317e2ab1b4f022b32fc5aa6d22 Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 16:44:32 2025 +0200 Move export functions to own module. commit 1a26e19338e73f8849c671471dd15cc3c1b1fe82 Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 15:51:15 2025 +0200 Split up models.py. commit 61f1ccd10f721b0ff4dc1eccf30c7a1ee13f204d Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 12:05:28 2025 +0200 Move utility function into own modules. commit 84b341ed35e2c5387819a8b9f9d41eca900ae79f Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 11:44:24 2025 +0200 Refactor auth functions use. commit 923e311e3da5b26d85845c2832b73b7b17c48adb Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 11:35:52 2025 +0200 Refactor route nameing and fix bugs along the way. commit f0a5c4419c340e62a2615c60b2a9de28204d2995 Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 10:34:33 2025 +0200 Fix URL endpoints in announcement template. commit b74d74542a1c8dc350749e4788a9464d067a88b5 Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 09:25:53 2025 +0200 Move announcements to own module. commit 9563a28021ac46c82c04fe4649b394dbf96f92c7 Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 09:16:30 2025 +0200 Combine Company view and edit templates. commit 6687c373e681d54e4deab6b2582fed5cea9aadf6 Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 08:17:42 2025 +0200 Move Users, Company and System Administration to own modules. commit 8b7894a2e3eb84bb059f546648b6b9536fea724e Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 07:40:57 2025 +0200 Move Teams and Projects to own modules. commit d11bf059d99839ecf1f5d7020b8c8c8a2454c00b Author: Jens Luedicke <jens@luedicke.me> Date: Mon Jul 7 07:09:33 2025 +0200 Move Tasks and Sprints to own modules.
This commit is contained in:
145
migrations/old_migrations/08_fix_work_region_enum.py
Executable file
145
migrations/old_migrations/08_fix_work_region_enum.py
Executable file
@@ -0,0 +1,145 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Fix work region enum values in the database
|
||||
"""
|
||||
|
||||
import os
|
||||
import psycopg2
|
||||
from psycopg2 import sql
|
||||
from urllib.parse import urlparse
|
||||
|
||||
# Get database URL from environment
|
||||
DATABASE_URL = os.environ.get('DATABASE_URL', 'postgresql://timetrack:timetrack123@localhost:5432/timetrack')
|
||||
|
||||
def fix_work_region_enum():
|
||||
"""Update work region enum values in database"""
|
||||
# Parse database URL
|
||||
parsed = urlparse(DATABASE_URL)
|
||||
|
||||
# Connect to database
|
||||
conn = psycopg2.connect(
|
||||
host=parsed.hostname,
|
||||
port=parsed.port or 5432,
|
||||
user=parsed.username,
|
||||
password=parsed.password,
|
||||
database=parsed.path[1:] # Remove leading slash
|
||||
)
|
||||
|
||||
try:
|
||||
with conn.cursor() as cur:
|
||||
print("Starting work region enum migration...")
|
||||
|
||||
# First check if work_region column is using enum type
|
||||
cur.execute("""
|
||||
SELECT data_type
|
||||
FROM information_schema.columns
|
||||
WHERE table_name = 'company_work_config'
|
||||
AND column_name = 'work_region'
|
||||
""")
|
||||
data_type = cur.fetchone()
|
||||
|
||||
if data_type and data_type[0] == 'USER-DEFINED':
|
||||
# It's an enum, we need to update it
|
||||
print("work_region is an enum type, migrating...")
|
||||
|
||||
# Create temporary column
|
||||
print("1. Creating temporary column...")
|
||||
cur.execute("ALTER TABLE company_work_config ADD COLUMN work_region_temp VARCHAR(50)")
|
||||
|
||||
# Copy and map values
|
||||
print("2. Copying and mapping values...")
|
||||
cur.execute("""
|
||||
UPDATE company_work_config SET work_region_temp = CASE
|
||||
WHEN work_region::text = 'GERMANY' THEN 'EU'
|
||||
WHEN work_region::text = 'DE' THEN 'EU'
|
||||
WHEN work_region::text = 'UNITED_STATES' THEN 'USA'
|
||||
WHEN work_region::text = 'US' THEN 'USA'
|
||||
WHEN work_region::text = 'UNITED_KINGDOM' THEN 'UK'
|
||||
WHEN work_region::text = 'GB' THEN 'UK'
|
||||
WHEN work_region::text = 'FRANCE' THEN 'EU'
|
||||
WHEN work_region::text = 'FR' THEN 'EU'
|
||||
WHEN work_region::text = 'EUROPEAN_UNION' THEN 'EU'
|
||||
WHEN work_region::text = 'CUSTOM' THEN 'OTHER'
|
||||
ELSE COALESCE(work_region::text, 'OTHER')
|
||||
END
|
||||
""")
|
||||
print(f" Updated {cur.rowcount} rows")
|
||||
|
||||
# Drop old column
|
||||
print("3. Dropping old work_region column...")
|
||||
cur.execute("ALTER TABLE company_work_config DROP COLUMN work_region")
|
||||
|
||||
# Check if enum type exists and drop it
|
||||
cur.execute("""
|
||||
SELECT EXISTS (
|
||||
SELECT 1 FROM pg_type WHERE typname = 'workregion'
|
||||
)
|
||||
""")
|
||||
if cur.fetchone()[0]:
|
||||
print("4. Dropping old workregion enum type...")
|
||||
cur.execute("DROP TYPE IF EXISTS workregion CASCADE")
|
||||
|
||||
# Create new enum type
|
||||
print("5. Creating new workregion enum type...")
|
||||
cur.execute("""
|
||||
CREATE TYPE workregion AS ENUM (
|
||||
'USA',
|
||||
'CANADA',
|
||||
'UK',
|
||||
'EU',
|
||||
'AUSTRALIA',
|
||||
'OTHER'
|
||||
)
|
||||
""")
|
||||
|
||||
# Add new column with enum type
|
||||
print("6. Adding new work_region column...")
|
||||
cur.execute("ALTER TABLE company_work_config ADD COLUMN work_region workregion DEFAULT 'OTHER'")
|
||||
|
||||
# Copy data back
|
||||
print("7. Copying data to new column...")
|
||||
cur.execute("UPDATE company_work_config SET work_region = work_region_temp::workregion")
|
||||
|
||||
# Drop temporary column
|
||||
print("8. Dropping temporary column...")
|
||||
cur.execute("ALTER TABLE company_work_config DROP COLUMN work_region_temp")
|
||||
|
||||
else:
|
||||
# It's already a varchar, just update the values
|
||||
print("work_region is already a varchar, updating values...")
|
||||
cur.execute("""
|
||||
UPDATE company_work_config SET work_region = CASE
|
||||
WHEN work_region = 'GERMANY' THEN 'EU'
|
||||
WHEN work_region = 'DE' THEN 'EU'
|
||||
WHEN work_region = 'UNITED_STATES' THEN 'USA'
|
||||
WHEN work_region = 'US' THEN 'USA'
|
||||
WHEN work_region = 'UNITED_KINGDOM' THEN 'UK'
|
||||
WHEN work_region = 'GB' THEN 'UK'
|
||||
WHEN work_region = 'FRANCE' THEN 'EU'
|
||||
WHEN work_region = 'FR' THEN 'EU'
|
||||
WHEN work_region = 'EUROPEAN_UNION' THEN 'EU'
|
||||
WHEN work_region = 'CUSTOM' THEN 'OTHER'
|
||||
ELSE COALESCE(work_region, 'OTHER')
|
||||
END
|
||||
""")
|
||||
print(f"Updated {cur.rowcount} rows")
|
||||
|
||||
# Commit changes
|
||||
conn.commit()
|
||||
print("\nWork region enum migration completed successfully!")
|
||||
|
||||
# Verify the results
|
||||
print("\nCurrent work_region values in database:")
|
||||
cur.execute("SELECT DISTINCT work_region FROM company_work_config ORDER BY work_region")
|
||||
for row in cur.fetchall():
|
||||
print(f" - {row[0]}")
|
||||
|
||||
except Exception as e:
|
||||
print(f"Error during migration: {e}")
|
||||
conn.rollback()
|
||||
raise
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
if __name__ == "__main__":
|
||||
fix_work_region_enum()
|
||||
Reference in New Issue
Block a user