Add DEMO feature to the project

services/demo_session/app/models/demo_session.py

@@ -0,0 +1,71 @@
+"""
+Demo Session Models
+Tracks ephemeral demo sessions for prospect users
+"""
+
+from sqlalchemy import Column, String, Boolean, DateTime, Integer, Enum as SQLEnum
+from sqlalchemy.dialects.postgresql import UUID, JSONB
+from datetime import datetime, timezone
+import uuid
+import enum
+
+from shared.database.base import Base
+
+
+class DemoSessionStatus(enum.Enum):
+    """Demo session status"""
+    ACTIVE = "active"
+    EXPIRED = "expired"
+    DESTROYED = "destroyed"
+
+
+class DemoSession(Base):
+    """Demo Session tracking model"""
+    __tablename__ = "demo_sessions"
+
+    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
+    session_id = Column(String(100), unique=True, nullable=False, index=True)
+
+    # Session ownership
+    user_id = Column(UUID(as_uuid=True), nullable=True)
+    ip_address = Column(String(45), nullable=True)
+    user_agent = Column(String(500), nullable=True)
+
+    # Demo tenant linking
+    base_demo_tenant_id = Column(UUID(as_uuid=True), nullable=False, index=True)
+    virtual_tenant_id = Column(UUID(as_uuid=True), nullable=False, index=True)
+    demo_account_type = Column(String(50), nullable=False)  # 'individual_bakery', 'central_baker'
+
+    # Session lifecycle
+    status = Column(SQLEnum(DemoSessionStatus, values_callable=lambda obj: [e.value for e in obj]), default=DemoSessionStatus.ACTIVE, index=True)
+    created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc), index=True)
+    expires_at = Column(DateTime(timezone=True), nullable=False, index=True)
+    last_activity_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
+    destroyed_at = Column(DateTime(timezone=True), nullable=True)
+
+    # Session metrics
+    request_count = Column(Integer, default=0)
+    data_cloned = Column(Boolean, default=False)
+    redis_populated = Column(Boolean, default=False)
+
+    # Session metadata
+    session_metadata = Column(JSONB, default=dict)
+
+    def __repr__(self):
+        return f"<DemoSession(session_id={self.session_id}, status={self.status.value})>"
+
+    def to_dict(self):
+        """Convert to dictionary"""
+        return {
+            "id": str(self.id),
+            "session_id": self.session_id,
+            "virtual_tenant_id": str(self.virtual_tenant_id),
+            "base_demo_tenant_id": str(self.base_demo_tenant_id),
+            "demo_account_type": self.demo_account_type,
+            "status": self.status.value,
+            "created_at": self.created_at.isoformat() if self.created_at else None,
+            "expires_at": self.expires_at.isoformat() if self.expires_at else None,
+            "last_activity_at": self.last_activity_at.isoformat() if self.last_activity_at else None,
+            "request_count": self.request_count,
+            "metadata": self.session_metadata
+        }