Upload core/models/user.py with huggingface_hub

core/models/user.py

@@ -31,8 +31,13 @@ class User(Base):
     mfa_enabled = Column(Boolean, default=False)
     mfa_secret = Column(EncryptedString, nullable=True)
 
+    # Preferences - Store as JSON string for compatibility
+    preferences = Column(String, default=lambda: json.dumps({}))
+
     # Relationships
-    cases = relationship("Case", back_populates="assignee", foreign_keys="Case.assignee_id")
+    cases = relationship(
+        "Case", back_populates="assignee", foreign_keys="Case.assignee_id"
+    )
     activities = relationship("CaseActivity", back_populates="user")
     devices = relationship("UserDevice", backref="user")
     onboarding_state = relationship("UserOnboardingState", backref="user")
@@ -75,6 +80,54 @@ class UserDevice(Base):
     created_at = Column(DateTime, default=utc_now)
 
 
+class UserSession(Base):
+    __tablename__ = "user_sessions"
+
+    id = Column(String, primary_key=True, index=True, default=lambda: str(uuid.uuid4()))
+    user_id = Column(String, ForeignKey("users.id"), index=True, nullable=False)
+    session_token = Column(String, unique=True, index=True, nullable=False)
+    device_info = Column(String)  # JSON string with device/browser info
+    ip_address = Column(String, nullable=False)
+    user_agent = Column(String)
+    location = Column(String)  # Geolocation data
+    is_active = Column(Boolean, default=True, index=True)
+    is_current = Column(Boolean, default=False)  # Mark current session
+    expires_at = Column(DateTime, index=True)
+    last_activity = Column(DateTime, default=utc_now, index=True)
+    created_at = Column(DateTime, default=utc_now)
+
+    # Relationships
+    user = relationship("User", backref="sessions")
+
+    __table_args__ = (
+        Index("idx_user_sessions_active", "user_id", "is_active"),
+        Index("idx_user_sessions_expires", "expires_at"),
+    )
+
+
+class LoginAttempt(Base):
+    __tablename__ = "login_attempts"
+
+    id = Column(String, primary_key=True, index=True, default=lambda: str(uuid.uuid4()))
+    email = Column(String, index=True)  # Store email for non-existent users
+    user_id = Column(String, ForeignKey("users.id"), index=True, nullable=True)
+    ip_address = Column(String, nullable=False, index=True)
+    user_agent = Column(String)
+    location = Column(String)
+    device_type = Column(String, default="desktop")
+    success = Column(Boolean, default=False, index=True)
+    failure_reason = Column(String)
+    attempted_at = Column(DateTime, default=utc_now, index=True)
+
+    # Relationships
+    user = relationship("User", backref="login_attempts")
+
+    __table_args__ = (
+        Index("idx_login_attempts_user_time", "user_id", "attempted_at"),
+        Index("idx_login_attempts_ip_time", "ip_address", "attempted_at"),
+    )
+
+
 class UserOnboardingState(Base):
     __tablename__ = "user_onboarding_states"
 
@@ -106,9 +159,13 @@ class TrainingRecord(Base):
 
     id = Column(String, primary_key=True, index=True)
     user_id = Column(String, index=True)
-    training_type = Column(String, index=True)  # security_awareness, compliance, technical
+    training_type = Column(
+        String, index=True
+    )  # security_awareness, compliance, technical
     training_module = Column(String, index=True)  # specific course or topic
-    completion_status = Column(String, default="not_started", index=True)  # not_started, in_progress, completed, failed
+    completion_status = Column(
+        String, default="not_started", index=True
+    )  # not_started, in_progress, completed, failed
     score = Column(Float)  # Test score if applicable
     completion_date = Column(DateTime, index=True)
     expiry_date = Column(DateTime, index=True)
@@ -134,10 +191,16 @@ class AccessReview(Base):
     reviewer_id = Column(String, index=True)
     review_period_start = Column(DateTime, index=True)
     review_period_end = Column(DateTime, index=True)
-    review_status = Column(String, default="pending", index=True)  # pending, in_progress, completed, overdue
+    review_status = Column(
+        String, default="pending", index=True
+    )  # pending, in_progress, completed, overdue
     overall_risk_assessment = Column(String, index=True)  # low, medium, high, critical
-    findings = Column(String, default=lambda: json.dumps([]))  # Specific access issues found
-    recommendations = Column(String, default=lambda: json.dumps([]))  # Remediation actions
+    findings = Column(
+        String, default=lambda: json.dumps([])
+    )  # Specific access issues found
+    recommendations = Column(
+        String, default=lambda: json.dumps([])
+    )  # Remediation actions
     approval_status = Column(String, default="pending")  # pending, approved, rejected
     approved_by = Column(String)
     approved_at = Column(DateTime)
@@ -151,6 +214,8 @@ __all__ = [
     "Team",
     "Project",
     "UserDevice",
+    "UserSession",
+    "LoginAttempt",
     "UserOnboardingState",
     "RookieChecklist",
     "TrainingRecord",