Files
AI-Website/backend/app/models.py
2026-01-09 09:48:57 +08:00

54 lines
2.0 KiB
Python

from sqlalchemy import Column, Integer, String, Float, DateTime, ForeignKey, Boolean
from sqlalchemy.orm import relationship
from app.database import Base
from datetime import datetime
class User(Base):
__tablename__ = "users"
id = Column(Integer, primary_key=True, index=True)
username = Column(String, unique=True, index=True)
email = Column(String, unique=True, index=True, nullable=True)
hashed_password = Column(String)
is_active = Column(Boolean, default=True)
is_admin = Column(Boolean, default=False)
balance = Column(Float, default=0)
# 添加关系
orders = relationship("Order", back_populates="user")
class App(Base):
__tablename__ = "apps"
id = Column(Integer, primary_key=True, index=True)
name = Column(String, index=True)
desc = Column(String)
price = Column(Float)
status = Column(String, default="上架") # 上架、下架
created_at = Column(DateTime, default=datetime.utcnow)
# 添加关系
orders = relationship("Order", back_populates="app")
class Order(Base):
__tablename__ = "orders"
id = Column(Integer, primary_key=True, index=True)
user_id = Column(Integer, ForeignKey("users.id"))
app_id = Column(Integer, ForeignKey("apps.id")) # 添加 app_id 字段
type = Column(String, default="应用调用") # 订单类型
amount = Column(Float) # 订单金额
description = Column(String, nullable=True) # 订单描述
status = Column(String) # 订单状态:待支付、已完成、已取消等
created_at = Column(DateTime, default=datetime.utcnow)
# 添加关系
user = relationship("User", back_populates="orders")
app = relationship("App", back_populates="orders")
class History(Base):
__tablename__ = "history"
id = Column(Integer, primary_key=True, index=True)
user_id = Column(Integer, ForeignKey('users.id'))
type = Column(String) # 'recharge' or 'consume'
amount = Column(Float)
desc = Column(String)
created_at = Column(DateTime, default=datetime.utcnow)