forked from ComputerTech/aprhodite
123 lines
4.4 KiB
Python
123 lines
4.4 KiB
Python
"""
|
||
models.py – SQLAlchemy ORM models for SexyChat.
|
||
|
||
Tables
|
||
------
|
||
users – Registered accounts
|
||
messages – Encrypted PM history (user↔user and user↔AI)
|
||
"""
|
||
|
||
from datetime import datetime, timezone
|
||
from database import db
|
||
|
||
|
||
def _utcnow():
|
||
return datetime.now(timezone.utc)
|
||
|
||
|
||
class User(db.Model):
|
||
__tablename__ = "users"
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
username = db.Column(db.String(20), unique=True, nullable=False, index=True)
|
||
password_hash = db.Column(db.String(128), nullable=False)
|
||
email = db.Column(db.String(255), unique=True, nullable=True)
|
||
role = db.Column(db.String(10), default="user", nullable=False) # root, admin, mod, user
|
||
has_ai_access = db.Column(db.Boolean, default=False, nullable=False)
|
||
ai_messages_used = db.Column(db.Integer, default=0, nullable=False)
|
||
is_verified = db.Column(db.Boolean, default=False, nullable=False)
|
||
created_at = db.Column(db.DateTime, default=_utcnow, nullable=False)
|
||
|
||
sent_messages = db.relationship(
|
||
"Message", foreign_keys="Message.sender_id",
|
||
backref="sender", lazy="dynamic",
|
||
)
|
||
received_messages = db.relationship(
|
||
"Message", foreign_keys="Message.recipient_id",
|
||
backref="recipient", lazy="dynamic",
|
||
)
|
||
|
||
# Persistent "Ignore" list
|
||
ignoring = db.relationship(
|
||
"User",
|
||
secondary="user_ignores",
|
||
primaryjoin="User.id == UserIgnore.ignorer_id",
|
||
secondaryjoin="User.id == UserIgnore.ignored_id",
|
||
backref=db.backref("ignored_by", lazy="dynamic"),
|
||
lazy="dynamic"
|
||
)
|
||
|
||
def __repr__(self):
|
||
return f"<User {self.username!r}>"
|
||
|
||
|
||
class UserIgnore(db.Model):
|
||
__tablename__ = "user_ignores"
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
ignorer_id = db.Column(db.Integer, db.ForeignKey("users.id"), nullable=False)
|
||
ignored_id = db.Column(db.Integer, db.ForeignKey("users.id"), nullable=False)
|
||
created_at = db.Column(db.DateTime, default=_utcnow, nullable=False)
|
||
|
||
__table_args__ = (
|
||
db.Index("ix_ignore_pair", "ignorer_id", "ignored_id", unique=True),
|
||
)
|
||
|
||
|
||
|
||
class Message(db.Model):
|
||
__tablename__ = "messages"
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
sender_id = db.Column(db.Integer, db.ForeignKey("users.id"), nullable=False)
|
||
recipient_id = db.Column(db.Integer, db.ForeignKey("users.id"), nullable=False)
|
||
# AES-GCM ciphertext – base64 encoded; server never stores plaintext
|
||
encrypted_content = db.Column(db.Text, nullable=False)
|
||
# AES-GCM nonce / IV – base64 encoded (12 bytes → 16 chars)
|
||
nonce = db.Column(db.String(64), nullable=False)
|
||
timestamp = db.Column(db.DateTime, default=_utcnow, nullable=False)
|
||
|
||
__table_args__ = (
|
||
# Composite indices for the two common query patterns
|
||
db.Index("ix_msg_recipient_ts", "recipient_id", "timestamp"),
|
||
db.Index("ix_msg_sender_ts", "sender_id", "timestamp"),
|
||
)
|
||
|
||
def __repr__(self):
|
||
return f"<Message {self.sender_id}→{self.recipient_id} @ {self.timestamp}>"
|
||
|
||
|
||
class Ban(db.Model):
|
||
"""Persisted ban entry – survives server restarts."""
|
||
__tablename__ = "bans"
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
username = db.Column(db.String(20), nullable=False, index=True)
|
||
ip = db.Column(db.String(45), nullable=True, index=True)
|
||
reason = db.Column(db.String(255), nullable=True)
|
||
created_at = db.Column(db.DateTime, default=_utcnow, nullable=False)
|
||
|
||
|
||
class Mute(db.Model):
|
||
"""Persisted mute entry – survives server restarts."""
|
||
__tablename__ = "mutes"
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
username = db.Column(db.String(20), unique=True, nullable=False, index=True)
|
||
created_at = db.Column(db.DateTime, default=_utcnow, nullable=False)
|
||
|
||
|
||
class VioletHistory(db.Model):
|
||
"""Per-user plaintext conversation history with Violet AI."""
|
||
__tablename__ = "violet_history"
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
user_id = db.Column(db.Integer, db.ForeignKey("users.id"), nullable=False)
|
||
role = db.Column(db.String(10), nullable=False) # 'user' or 'assistant'
|
||
text = db.Column(db.Text, nullable=False)
|
||
timestamp = db.Column(db.DateTime, default=_utcnow, nullable=False)
|
||
|
||
__table_args__ = (
|
||
db.Index("ix_violet_hist_user_ts", "user_id", "timestamp"),
|
||
)
|