162 lines
6.8 KiB
Python
162 lines
6.8 KiB
Python
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)
|