Switched to timezone aware schema for database

src/alpine_bits_python/util/fix_postgres_sequences.py

@@ -1,10 +1,16 @@
 #!/usr/bin/env python3
-"""Fix PostgreSQL sequence values after migration from SQLite.
+"""Fix PostgreSQL sequences and migrate datetime columns after SQLite migration.
 
-This script resets all ID sequence values to match the current maximum ID
-in each table. This is necessary because the migration script inserts records
+This script performs two operations:
+1. Migrates DateTime columns to TIMESTAMP WITH TIME ZONE for timezone-aware support
+2. Resets all ID sequence values to match the current maximum ID in each table
+
+The sequence reset is necessary because the migration script inserts records
 with explicit IDs, which doesn't automatically advance PostgreSQL sequences.
 
+The datetime migration ensures proper handling of timezone-aware datetimes,
+which is required by the application code.
+
 Usage:
     # Using default config.yaml
     uv run python -m alpine_bits_python.util.fix_postgres_sequences
@@ -42,14 +48,40 @@ from alpine_bits_python.logging_config import get_logger, setup_logging
 _LOGGER = get_logger(__name__)
 
 
+async def migrate_datetime_columns(session: AsyncSession) -> None:
+    """Migrate DateTime columns to TIMESTAMP WITH TIME ZONE.
+
+    This updates the columns to properly handle timezone-aware datetimes.
+    """
+    _LOGGER.info("\nMigrating DateTime columns to timezone-aware...")
+
+    datetime_columns = [
+        ("hashed_customers", "created_at"),
+        ("reservations", "created_at"),
+        ("acked_requests", "timestamp"),
+    ]
+
+    for table_name, column_name in datetime_columns:
+        _LOGGER.info(f"  {table_name}.{column_name}: Converting to TIMESTAMPTZ")
+        await session.execute(
+            text(
+                f"ALTER TABLE {table_name} "
+                f"ALTER COLUMN {column_name} TYPE TIMESTAMP WITH TIME ZONE"
+            )
+        )
+
+    await session.commit()
+    _LOGGER.info("✓ DateTime columns migrated to timezone-aware")
+
+
 async def fix_sequences(database_url: str) -> None:
-    """Fix PostgreSQL sequences to match current max IDs.
+    """Fix PostgreSQL sequences to match current max IDs and migrate datetime columns.
 
     Args:
         database_url: PostgreSQL database URL
     """
     _LOGGER.info("=" * 70)
-    _LOGGER.info("PostgreSQL Sequence Fix")
+    _LOGGER.info("PostgreSQL Migration & Sequence Fix")
     _LOGGER.info("=" * 70)
     _LOGGER.info("Database: %s", database_url.split("@")[-1] if "@" in database_url else database_url)
     _LOGGER.info("=" * 70)
@@ -59,6 +91,11 @@ async def fix_sequences(database_url: str) -> None:
     SessionMaker = async_sessionmaker(engine, expire_on_commit=False)
 
     try:
+        # Migrate datetime columns first
+        async with SessionMaker() as session:
+            await migrate_datetime_columns(session)
+
+        # Then fix sequences
         async with SessionMaker() as session:
             # List of tables and their sequence names
             tables = [
@@ -105,9 +142,12 @@ async def fix_sequences(database_url: str) -> None:
             await session.commit()
 
         _LOGGER.info("\n" + "=" * 70)
-        _LOGGER.info("✓ Sequences fixed successfully!")
+        _LOGGER.info("✓ Migration completed successfully!")
         _LOGGER.info("=" * 70)
-        _LOGGER.info("\nYou can now insert new records without ID conflicts.")
+        _LOGGER.info("\nChanges applied:")
+        _LOGGER.info("  1. DateTime columns are now timezone-aware (TIMESTAMPTZ)")
+        _LOGGER.info("  2. Sequences are reset to match current max IDs")
+        _LOGGER.info("\nYou can now insert new records without conflicts.")
 
     except Exception as e:
         _LOGGER.exception("Failed to fix sequences: %s", e)