from sqlalchemy import Column, Integer, String, Text, DateTime, JSON
from sqlalchemy.sql import func
from app.database import Base

class LoginLog(Base):
    __tablename__ = "login_log"
    __table_args__ = {'schema': 'aigcspace'}
    
    id = Column(Integer, primary_key=True, index=True)
    user_id = Column(String(50), nullable=False, index=True)
    username = Column(String(100))
    user_type = Column(String(20), nullable=False)
    login_result = Column(String(20), nullable=False, index=True)
    fail_reason = Column(String(200))
    ip_address = Column(String(50))
    user_agent = Column(Text)
    created_at = Column(DateTime, default=func.now(), index=True)

class APILog(Base):
    __tablename__ = "api_log"
    __table_args__ = {'schema': 'aigcspace'}
    
    id = Column(Integer, primary_key=True, index=True)
    user_id = Column(String(50), index=True)
    username = Column(String(100))
    api_path = Column(String(500), nullable=False, index=True)
    request_method = Column(String(10), nullable=False)
    request_params = Column(JSON)
    response_status = Column(Integer)
    response_time = Column(Integer)
    created_at = Column(DateTime, default=func.now(), index=True)