100 lines
3.9 KiB
Python
100 lines
3.9 KiB
Python
from flask_sqlalchemy import SQLAlchemy
|
|
from werkzeug.security import generate_password_hash, check_password_hash
|
|
|
|
from datetime import datetime
|
|
import bcrypt
|
|
|
|
db = SQLAlchemy()
|
|
|
|
class User(db.Model):
|
|
__tablename__ = 'users'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
username = db.Column(db.String(50), unique=True, nullable=False)
|
|
password_hash = db.Column(db.String(255), nullable=False)
|
|
email = db.Column(db.String(100), unique=True, nullable=False)
|
|
uid = db.Column(db.String(50)) # 站点UID
|
|
role = db.Column(db.String(20), default='user') # admin/user
|
|
status = db.Column(db.String(20), default='pending') # pending/active/disabled
|
|
tags = db.Column(db.String(200)) # 用户标签，逗号分隔
|
|
note = db.Column(db.Text) # 管理员备注
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
approved_at = db.Column(db.DateTime) # 审核通过时间
|
|
approved_by = db.Column(db.Integer, db.ForeignKey('users.id')) # 审核人
|
|
|
|
def set_password(self, password):
|
|
self.password_hash = generate_password_hash(password, method='pbkdf2:sha256')
|
|
|
|
def check_password(self, password):
|
|
return check_password_hash(self.password_hash, password)
|
|
|
|
|
|
|
|
class Group(db.Model):
|
|
__tablename__ = 'groups'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
group_name = db.Column(db.String(50), nullable=False)
|
|
group_key = db.Column(db.String(50), unique=True, nullable=False)
|
|
description = db.Column(db.Text)
|
|
status = db.Column(db.String(20), default='active')
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
|
|
|
class UserGroup(db.Model):
|
|
__tablename__ = 'user_groups'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
|
|
group_id = db.Column(db.Integer, db.ForeignKey('groups.id'), nullable=False)
|
|
is_admin = db.Column(db.Boolean, default=False) # 是否为组管理员
|
|
join_date = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
|
user = db.relationship('User', backref='user_groups')
|
|
group = db.relationship('Group', backref='user_groups')
|
|
|
|
|
|
class Task(db.Model):
|
|
__tablename__ = 'tasks'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
group_id = db.Column(db.Integer, db.ForeignKey('groups.id'), nullable=False)
|
|
|
|
# 发布组特定字段
|
|
series_name = db.Column(db.String(200), nullable=False)
|
|
series_link = db.Column(db.String(500))
|
|
series_date = db.Column(db.Date, nullable=False)
|
|
priority = db.Column(db.String(20)) # 高/中/低
|
|
|
|
# 流程字段
|
|
status = db.Column(db.String(20), default='pending') # pending/claimed/completed/cancelled
|
|
claimed_by = db.Column(db.Integer, db.ForeignKey('users.id'))
|
|
claimed_at = db.Column(db.DateTime)
|
|
claim_note = db.Column(db.Text)
|
|
|
|
torrent_id = db.Column(db.String(100))
|
|
complete_note = db.Column(db.Text)
|
|
completed_at = db.Column(db.DateTime)
|
|
|
|
created_by = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
|
# 关系
|
|
group = db.relationship('Group', backref='tasks')
|
|
claimer = db.relationship('User', foreign_keys=[claimed_by], backref='claimed_tasks')
|
|
creator = db.relationship('User', foreign_keys=[created_by], backref='created_tasks')
|
|
|
|
|
|
class TaskLog(db.Model):
|
|
__tablename__ = 'task_logs'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
task_id = db.Column(db.Integer, db.ForeignKey('tasks.id'), nullable=False)
|
|
user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
|
|
action = db.Column(db.String(50), nullable=False) # create/claim/complete/cancel
|
|
comment = db.Column(db.Text)
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
|
task = db.relationship('Task', backref='logs')
|
|
user = db.relationship('User', backref='task_logs')
|