diff --git a/app/models.py b/app/models.py index 298d244..faa1c64 100644 --- a/app/models.py +++ b/app/models.py @@ -37,14 +37,17 @@ class Attachment(Base): class User(Base): __tablename__ = "users" - id = Column(String, primary_key=True) - email = Column(String, unique=True) + #id = Column(String, primary_key=True) + #email = Column(String, unique=True) + email = Column(String, primary_key=True) name = Column(String) + password = Column(String) + totp_secret = Column(String, nullable=True) twofa_enabled = Column(Boolean, default=False) twofa_method = Column(String, default="totp") twofa_secret = Column(String) twofa_sensitive_actions = Column(Boolean, default=False) - email = Column(String, primary_key=True) + role = Column(String, default="user") # "user", "admin" can_use_audio = Column(Boolean, default=False) can_use_vision = Column(Boolean, default=False) @@ -53,7 +56,9 @@ class User(Base): can_export_pdf = Column(Boolean, default=True) can_export_md = Column(Boolean, default=True) - + + + class Preferences(Base):