Add migration services

services/pos/alembic.ini

@@ -29,7 +29,7 @@ revision_environment = false
 sourceless = false
 
 # version of a migration file's filename format
-version_num_format = %s
+version_num_format = %%s
 
 # version path separator
 version_path_separator = os

services/pos/app/main.py

@@ -5,6 +5,7 @@ Handles integration with external POS systems (Square, Toast, Lightspeed)
 
 import time
 from fastapi import FastAPI, Request
+from sqlalchemy import text
 from app.core.config import settings
 from app.api import pos_config, webhooks, sync
 from app.core.database import database_manager
@@ -14,6 +15,27 @@ from shared.service_base import StandardFastAPIService
 class POSService(StandardFastAPIService):
     """POS Integration Service with standardized setup"""
 
+    expected_migration_version = "001_initial_pos"
+
+    async def on_startup(self, app):
+        """Custom startup logic including migration verification"""
+        await self.verify_migrations()
+        await super().on_startup(app)
+
+    async def verify_migrations(self):
+        """Verify database schema matches the latest migrations."""
+        try:
+            async with self.database_manager.get_session() as session:
+                result = await session.execute(text("SELECT version_num FROM alembic_version"))
+                version = result.scalar()
+                if version != self.expected_migration_version:
+                    self.logger.error(f"Migration version mismatch: expected {self.expected_migration_version}, got {version}")
+                    raise RuntimeError(f"Migration version mismatch: expected {self.expected_migration_version}, got {version}")
+                self.logger.info(f"Migration verification successful: {version}")
+        except Exception as e:
+            self.logger.error(f"Migration verification failed: {e}")
+            raise
+
     def __init__(self):
         # Define expected database tables for health checks
         pos_expected_tables = [

services/pos/migrations/env.py

@@ -35,8 +35,24 @@ except ImportError as e:
 # this is the Alembic Config object
 config = context.config
 
-# Set database URL from settings if not already set
-database_url = os.getenv('DATABASE_URL') or getattr(settings, 'DATABASE_URL', None)
+# Set database URL from environment variables or settings
+# Try service-specific DATABASE_URL first, then fall back to generic
+database_url = os.getenv('POS_DATABASE_URL') or os.getenv('DATABASE_URL')
+
+# If DATABASE_URL is not set, construct from individual components
+if not database_url:
+    postgres_host = os.getenv('POSTGRES_HOST')
+    postgres_port = os.getenv('POSTGRES_PORT', '5432')
+    postgres_db = os.getenv('POSTGRES_DB')
+    postgres_user = os.getenv('POSTGRES_USER')
+    postgres_password = os.getenv('POSTGRES_PASSWORD')
+
+    if all([postgres_host, postgres_db, postgres_user, postgres_password]):
+        database_url = f"postgresql+asyncpg://{postgres_user}:{postgres_password}@{postgres_host}:{postgres_port}/{postgres_db}"
+    else:
+        # Fallback to settings
+        database_url = getattr(settings, 'DATABASE_URL', None)
+
 if database_url:
     config.set_main_option("sqlalchemy.url", database_url)
 

services/pos/migrations/env.py.bak

@@ -0,0 +1,111 @@
+"""Alembic environment configuration for pos service"""
+
+import asyncio
+import logging
+import os
+import sys
+from logging.config import fileConfig
+from sqlalchemy import pool
+from sqlalchemy.engine import Connection
+from sqlalchemy.ext.asyncio import async_engine_from_config
+from alembic import context
+
+# Add the service directory to the Python path
+service_path = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+if service_path not in sys.path:
+    sys.path.insert(0, service_path)
+
+# Add shared modules to path
+shared_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..", "shared"))
+if shared_path not in sys.path:
+    sys.path.insert(0, shared_path)
+
+try:
+    from app.core.config import settings
+    from shared.database.base import Base
+
+    # Import all models to ensure they are registered with Base.metadata
+    from app.models import *  # Import all models
+
+except ImportError as e:
+    print(f"Import error in migrations env.py: {e}")
+    print(f"Current Python path: {sys.path}")
+    raise
+
+# this is the Alembic Config object
+config = context.config
+
+# Set database URL from environment variables or settings
+database_url = os.getenv('POS_DATABASE_URL') or os.getenv('DATABASE_URL')
+
+# If DATABASE_URL is not set, construct from individual components
+if not database_url:
+    postgres_host = os.getenv('POSTGRES_HOST')
+    postgres_port = os.getenv('POSTGRES_PORT', '5432')
+    postgres_db = os.getenv('POSTGRES_DB')
+    postgres_user = os.getenv('POSTGRES_USER')
+    postgres_password = os.getenv('POSTGRES_PASSWORD')
+
+    if all([postgres_host, postgres_db, postgres_user, postgres_password]):
+        database_url = f"postgresql+asyncpg://{postgres_user}:{postgres_password}@{postgres_host}:{postgres_port}/{postgres_db}"
+    else:
+        # Fallback to settings
+        database_url = getattr(settings, 'DATABASE_URL', None)
+
+if database_url:
+    config.set_main_option("sqlalchemy.url", database_url)
+
+# Interpret the config file for Python logging
+if config.config_file_name is not None:
+    fileConfig(config.config_file_name)
+
+# Set target metadata
+target_metadata = Base.metadata
+
+def run_migrations_offline() -> None:
+    """Run migrations in 'offline' mode."""
+    url = config.get_main_option("sqlalchemy.url")
+    context.configure(
+        url=url,
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+        compare_type=True,
+        compare_server_default=True,
+    )
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+def do_run_migrations(connection: Connection) -> None:
+    context.configure(
+        connection=connection,
+        target_metadata=target_metadata,
+        compare_type=True,
+        compare_server_default=True,
+    )
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+async def run_async_migrations() -> None:
+    """Run migrations in 'online' mode."""
+    connectable = async_engine_from_config(
+        config.get_section(config.config_ini_section, {}),
+        prefix="sqlalchemy.",
+        poolclass=pool.NullPool,
+    )
+
+    async with connectable.connect() as connection:
+        await connection.run_sync(do_run_migrations)
+
+    await connectable.dispose()
+
+def run_migrations_online() -> None:
+    """Run migrations in 'online' mode."""
+    asyncio.run(run_async_migrations())
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()