init

2025-11-28 21:20:40 +08:00
commit f940b95b67
73 changed files with 15721 additions and 0 deletions
--- a/models/init.py
+++ b/models/init.py
@@ -0,0 +1,161 @@
+from datetime import datetime
+from flask_sqlalchemy import SQLAlchemy
+
+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)
+    email = db.Column(db.String(100), unique=True, nullable=False)
+    password = db.Column(db.String(255), nullable=False)
+    group_id = db.Column(db.Integer, db.ForeignKey('user_groups.id'), default=2)
+    register_ip = db.Column(db.String(45))
+    last_login_ip = db.Column(db.String(45))
+    total_parse_count = db.Column(db.Integer, default=0)
+    is_active = db.Column(db.Boolean, default=True)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+    group = db.relationship('UserGroup', backref='users')
+
+    @property
+    def is_authenticated(self):
+        return True
+
+    def get_id(self):
+        return str(self.id)
+
+class UserGroup(db.Model):
+    __tablename__ = 'user_groups'
+
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(50), unique=True, nullable=False)
+    daily_limit = db.Column(db.Integer, default=10)
+    description = db.Column(db.String(255))
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+class Admin(db.Model):
+    __tablename__ = 'admins'
+
+    id = db.Column(db.Integer, primary_key=True)
+    username = db.Column(db.String(50), unique=True, nullable=False)
+    password = db.Column(db.String(255), nullable=False)
+    email = db.Column(db.String(100))
+    totp_secret = db.Column(db.String(100))
+    is_2fa_enabled = db.Column(db.Boolean, default=False)
+    last_login_ip = db.Column(db.String(45))
+    last_login_at = db.Column(db.DateTime)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+class ParserAPI(db.Model):
+    __tablename__ = 'parser_apis'
+
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(100), nullable=False)
+    platform = db.Column(db.String(50), nullable=False)
+    api_url = db.Column(db.String(500), nullable=False)
+    api_key = db.Column(db.String(255))
+    weight = db.Column(db.Integer, default=1)
+    is_enabled = db.Column(db.Boolean, default=True)
+    health_status = db.Column(db.Boolean, default=True)
+    last_check_at = db.Column(db.DateTime)
+    fail_count = db.Column(db.Integer, default=0)
+    total_calls = db.Column(db.Integer, default=0)
+    success_calls = db.Column(db.Integer, default=0)
+    avg_response_time = db.Column(db.Integer, default=0)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+class SiteConfig(db.Model):
+    __tablename__ = 'site_config'
+
+    id = db.Column(db.Integer, primary_key=True)
+    config_key = db.Column(db.String(100), unique=True, nullable=False)
+    config_value = db.Column(db.Text)
+    config_type = db.Column(db.String(50), default='string')
+    description = db.Column(db.String(255))
+    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+class SMTPConfig(db.Model):
+    __tablename__ = 'smtp_config'
+
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(100), nullable=False)
+    host = db.Column(db.String(255), nullable=False)
+    port = db.Column(db.Integer, nullable=False)
+    username = db.Column(db.String(255), nullable=False)
+    password = db.Column(db.String(255), nullable=False)
+    from_email = db.Column(db.String(255), nullable=False)
+    from_name = db.Column(db.String(100))
+    use_tls = db.Column(db.Boolean, default=True)
+    is_enabled = db.Column(db.Boolean, default=True)
+    is_default = db.Column(db.Boolean, default=False)
+    weight = db.Column(db.Integer, default=1)
+    send_count = db.Column(db.Integer, default=0)
+    fail_count = db.Column(db.Integer, default=0)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+class EmailVerification(db.Model):
+    __tablename__ = 'email_verification'
+
+    id = db.Column(db.Integer, primary_key=True)
+    email = db.Column(db.String(100), nullable=False)
+    code = db.Column(db.String(10), nullable=False)
+    purpose = db.Column(db.String(50), nullable=False)
+    is_used = db.Column(db.Boolean, default=False)
+    expires_at = db.Column(db.DateTime, nullable=False)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+
+class ParseLog(db.Model):
+    __tablename__ = 'parse_logs'
+
+    id = db.Column(db.BigInteger, primary_key=True)
+    user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
+    ip_address = db.Column(db.String(45), nullable=False)
+    platform = db.Column(db.String(50), nullable=False)
+    video_url = db.Column(db.String(1000), nullable=False)
+    parser_api_id = db.Column(db.Integer, db.ForeignKey('parser_apis.id'))
+    status = db.Column(db.String(20), nullable=False)
+    error_message = db.Column(db.Text)
+    response_time = db.Column(db.Integer)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+
+class DailyParseStat(db.Model):
+    __tablename__ = 'daily_parse_stats'
+
+    id = db.Column(db.Integer, primary_key=True)
+    user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
+    ip_address = db.Column(db.String(45))
+    date = db.Column(db.Date, nullable=False)
+    parse_count = db.Column(db.Integer, default=0)
+    success_count = db.Column(db.Integer, default=0)
+    fail_count = db.Column(db.Integer, default=0)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+class HealthCheckConfig(db.Model):
+    __tablename__ = 'health_check_config'
+
+    id = db.Column(db.Integer, primary_key=True)
+    platform = db.Column(db.String(50), nullable=False)
+    test_url = db.Column(db.String(1000), nullable=False)
+    check_interval = db.Column(db.Integer, default=300)
+    is_enabled = db.Column(db.Boolean, default=True)
+    alert_email = db.Column(db.String(255))
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+class HealthCheckLog(db.Model):
+    __tablename__ = 'health_check_logs'
+
+    id = db.Column(db.BigInteger, primary_key=True)
+    parser_api_id = db.Column(db.Integer, db.ForeignKey('parser_apis.id'), nullable=False)
+    status = db.Column(db.String(20), nullable=False)
+    response_time = db.Column(db.Integer)
+    error_message = db.Column(db.Text)
+    checked_at = db.Column(db.DateTime, default=datetime.utcnow)