87 lines
3.0 KiB
Python
87 lines
3.0 KiB
Python
from sqlalchemy import Column, Integer, String, ForeignKey, Text, Table, Boolean, DateTime
|
|
from sqlalchemy.orm import relationship
|
|
|
|
from .database import Base
|
|
from datetime import datetime
|
|
|
|
|
|
|
|
class SubscriptionPlans(Base):
|
|
__tablename__ = "subscription_plans"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
name = Column(String, unique=True, index=True) # "Free", "Pro", "Ultra"
|
|
max_heirs = Column(Integer, default=1)
|
|
weekly_token_limit = Column(Integer, default=1000)
|
|
max_assets = Column(Integer, default=5)
|
|
max_storage_mb = Column(Integer, default=10)
|
|
can_use_ai_proxy = Column(Boolean, default=False)
|
|
description = Column(Text, nullable=True)
|
|
|
|
users = relationship("User", back_populates="subscription_plans")
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
username = Column(String, index=True)
|
|
email = Column(String, unique=True, index=True)
|
|
hashed_password = Column(String)
|
|
|
|
|
|
tier = Column(String, ForeignKey("subscription_plans.name"), default="Free")
|
|
tier_expires_at = Column(DateTime)
|
|
last_active_at = Column(DateTime)
|
|
|
|
subscription_plans = relationship("SubscriptionPlans", back_populates="users")
|
|
|
|
# System keys
|
|
public_key = Column(String)
|
|
private_key = Column(String) # Encrypted or raw? Storing raw for now as per req
|
|
is_admin = Column(Boolean, default=False)
|
|
guale = Column(Boolean, default=False)
|
|
|
|
assets = relationship("Asset", foreign_keys="Asset.author_id", back_populates="author")
|
|
inherited_assets = relationship("Asset", foreign_keys="Asset.heir_id", back_populates="heir")
|
|
|
|
|
|
|
|
class Asset(Base):
|
|
__tablename__ = "assets"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
title = Column(String, index=True)
|
|
content_outer_encrypted = Column(Text)
|
|
author_id = Column(Integer, ForeignKey("users.id"))
|
|
heir_id = Column(Integer, ForeignKey("users.id"))
|
|
type = Column(String, index=True, nullable=True)
|
|
|
|
# Key shard for this asset
|
|
private_key_shard = Column(String)
|
|
|
|
author = relationship("User", foreign_keys=[author_id], back_populates="assets")
|
|
heir = relationship("User", foreign_keys=[heir_id], back_populates="inherited_assets")
|
|
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
|
|
|
class AIConfig(Base):
|
|
__tablename__ = "ai_configs"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
provider_name = Column(String, unique=True, index=True)
|
|
api_key = Column(String)
|
|
api_url = Column(String)
|
|
default_model = Column(String)
|
|
is_active = Column(Boolean, default=True)
|
|
|
|
class UserTokenUsage(Base):
|
|
__tablename__ = "user_token_usage"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id"), unique=True)
|
|
tokens_used = Column(Integer, default=0)
|
|
last_reset_at = Column(DateTime)
|
|
|
|
user = relationship("User", backref="token_usage", uselist=False) |