joaquin
f5d6745ed0
Full-stack IPTV management system with Xtream Codes API compatibility, multi-account provider fan-out, Jellyfin local library integration, real-time dashboard, EPG support, and A/V-sync-safe stream recovery.
67 lines
2.8 KiB
Python
67 lines
2.8 KiB
Python
from sqlalchemy import Column, Integer, String, Boolean, DateTime, ForeignKey, UniqueConstraint, func
|
|
from sqlalchemy.orm import relationship
|
|
import bcrypt as _bcrypt
|
|
from ..database import Base
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
username = Column(String(100), unique=True, nullable=False, index=True)
|
|
password_hash = Column(String(200), nullable=False)
|
|
max_connections = Column(Integer, default=1, nullable=False)
|
|
expiry_date = Column(DateTime(timezone=True), nullable=True)
|
|
is_active = Column(Boolean, default=True, nullable=False)
|
|
is_priority = Column(Boolean, default=False, nullable=False)
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
catalog_entries = relationship("UserCatalogEntry", back_populates="user", cascade="all, delete-orphan")
|
|
provider_preferences = relationship("UserProviderPreference", back_populates="user", cascade="all, delete-orphan", order_by="UserProviderPreference.priority")
|
|
|
|
def set_password(self, password: str) -> None:
|
|
hashed = _bcrypt.hashpw(password.encode(), _bcrypt.gensalt())
|
|
self.password_hash = hashed.decode()
|
|
|
|
def verify_password(self, password: str) -> bool:
|
|
return _bcrypt.checkpw(password.encode(), self.password_hash.encode())
|
|
|
|
|
|
class UserCatalogEntry(Base):
|
|
__tablename__ = "user_catalog_entries"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
|
|
category_id = Column(Integer, ForeignKey("categories.id", ondelete="CASCADE"), nullable=False)
|
|
|
|
user = relationship("User", back_populates="catalog_entries")
|
|
category = relationship("Category", back_populates="catalog_entries")
|
|
|
|
|
|
class UserCustomCatalogEntry(Base):
|
|
__tablename__ = "user_custom_catalog_entries"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
|
|
custom_category_id = Column(Integer, ForeignKey("custom_categories.id", ondelete="CASCADE"), nullable=False)
|
|
|
|
__table_args__ = (
|
|
UniqueConstraint("user_id", "custom_category_id", name="uq_user_custom_cat"),
|
|
)
|
|
|
|
|
|
class UserProviderPreference(Base):
|
|
__tablename__ = "user_provider_preferences"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
|
|
provider_account_id = Column(Integer, ForeignKey("provider_accounts.id", ondelete="CASCADE"), nullable=False)
|
|
priority = Column(Integer, default=0, nullable=False)
|
|
|
|
user = relationship("User", back_populates="provider_preferences")
|
|
provider_account = relationship("ProviderAccount")
|
|
|
|
__table_args__ = (
|
|
UniqueConstraint("user_id", "provider_account_id", name="uq_user_provider_pref"),
|
|
)
|