72 lines
2.8 KiB
Python
72 lines
2.8 KiB
Python
"""
|
|
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
|
|
}
|