255 lines
9.5 KiB
Python
255 lines
9.5 KiB
Python
import mysql.connector
|
||
from mysql.connector import Error
|
||
import uuid
|
||
from datetime import datetime, timedelta
|
||
import hashlib
|
||
import os
|
||
|
||
|
||
class LicenseDatabase:
|
||
def __init__(self, host=None, database=None, user=None, password=None):
|
||
self.host = host or os.environ.get('DB_HOST', 'taiyiagi.xyz')
|
||
self.database = database or os.environ.get('DB_NAME', 'filesend')
|
||
self.user = user or os.environ.get('DB_USER', 'taiyi')
|
||
self.password = password or os.environ.get('DB_PASSWORD', 'taiyi1224')
|
||
self.connection = None
|
||
|
||
def connect(self):
|
||
"""连接到数据库"""
|
||
try:
|
||
self.connection = mysql.connector.connect(
|
||
host=self.host,
|
||
database=self.database,
|
||
user=self.user,
|
||
password=self.password
|
||
)
|
||
if self.connection.is_connected():
|
||
return True
|
||
return False
|
||
except Error as e:
|
||
print(f"数据库连接错误: {e}")
|
||
return False
|
||
|
||
def create_tables(self):
|
||
"""创建必要的数据库表"""
|
||
if not self.connection or not self.connection.is_connected():
|
||
if not self.connect():
|
||
return False
|
||
|
||
try:
|
||
cursor = self.connection.cursor()
|
||
|
||
# 创建卡密表
|
||
cursor.execute('''
|
||
CREATE TABLE IF NOT EXISTS license_keys (
|
||
id INT AUTO_INCREMENT PRIMARY KEY,
|
||
key_code VARCHAR(50) NOT NULL UNIQUE,
|
||
machine_code VARCHAR(100) DEFAULT NULL,
|
||
start_time DATETIME DEFAULT NULL,
|
||
end_time DATETIME NOT NULL,
|
||
status ENUM('unused', 'active', 'expired', 'banned') DEFAULT 'unused',
|
||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||
)
|
||
''')
|
||
|
||
self.connection.commit()
|
||
cursor.close()
|
||
return True
|
||
except Error as e:
|
||
print(f"创建表错误: {e}")
|
||
return False
|
||
|
||
def generate_key(self, days_valid):
|
||
"""生成一个新的卡密并保存到数据库"""
|
||
if not self.connection or not self.connection.is_connected():
|
||
if not self.connect():
|
||
return None
|
||
|
||
try:
|
||
# 生成UUID作为基础，然后进行哈希处理
|
||
key_uuid = uuid.uuid4().hex
|
||
hash_obj = hashlib.sha256(key_uuid.encode())
|
||
# 取前20个字符作为卡密
|
||
key_code = hash_obj.hexdigest()[:20].upper()
|
||
|
||
# 格式化卡密，每5个字符一组
|
||
formatted_key = '-'.join([key_code[i:i + 5] for i in range(0, len(key_code), 5)])
|
||
|
||
# 计算过期时间
|
||
end_time = datetime.now() + timedelta(days=days_valid)
|
||
|
||
cursor = self.connection.cursor()
|
||
query = """
|
||
INSERT INTO license_keys (key_code, end_time)
|
||
VALUES (%s, %s)
|
||
"""
|
||
cursor.execute(query, (formatted_key, end_time))
|
||
self.connection.commit()
|
||
cursor.close()
|
||
|
||
return formatted_key
|
||
except Error as e:
|
||
print(f"生成卡密错误: {e}")
|
||
return None
|
||
|
||
def validate_key(self, key_code, machine_code):
|
||
"""验证卡密是否有效，并绑定机器码 - 严格一机一码"""
|
||
if not self.connection or not self.connection.is_connected():
|
||
if not self.connect():
|
||
return False, "数据库连接失败"
|
||
|
||
try:
|
||
cursor = self.connection.cursor(dictionary=True)
|
||
|
||
# 查询卡密信息
|
||
query = "SELECT * FROM license_keys WHERE key_code = %s"
|
||
cursor.execute(query, (key_code,))
|
||
key_info = cursor.fetchone()
|
||
|
||
if not key_info:
|
||
cursor.close()
|
||
return False, "无效的激活码"
|
||
|
||
# 检查卡密状态
|
||
if key_info['status'] == 'banned':
|
||
cursor.close()
|
||
return False, "激活码已被封禁"
|
||
|
||
if key_info['status'] == 'expired':
|
||
cursor.close()
|
||
return False, "激活码已过期"
|
||
|
||
# 一机一码严格检查：每个激活码只能在一台机器上使用
|
||
if key_info['status'] == 'active':
|
||
if key_info['machine_code'] != machine_code:
|
||
# 这个码已经用过了，不能再次使用
|
||
cursor.close()
|
||
return False, f"此激活码已在设备{key_info['machine_code'][:8]}...上使用，一个激活码只能在一台设备上使用一次"
|
||
else:
|
||
# 已经激活过这台机器，验证是否过期
|
||
if datetime.now() > key_info['end_time']:
|
||
update_query = "UPDATE license_keys SET status = 'expired' WHERE key_code = %s"
|
||
cursor.execute(update_query, (key_code,))
|
||
self.connection.commit()
|
||
cursor.close()
|
||
return False, "激活码已过期"
|
||
else:
|
||
cursor.close()
|
||
return True, "此设备已激活，继续使用"
|
||
|
||
# 首次激活：验证通过后绑定到机器
|
||
if key_info['status'] == 'unused':
|
||
update_query = """
|
||
UPDATE license_keys
|
||
SET status = 'active', machine_code = %s, start_time = %s
|
||
WHERE key_code = %s AND status = 'unused'
|
||
"""
|
||
cursor.execute(update_query, (machine_code, datetime.now(), key_code))
|
||
rows_affected = cursor.rowcount
|
||
self.connection.commit()
|
||
|
||
if rows_affected == 0:
|
||
# 可能已经被其他并发操作激活
|
||
cursor.close()
|
||
return False, "激活码已被使用，请使用新的激活码"
|
||
|
||
# 再次检查是否过期（防止并发问题）
|
||
final_check_query = "SELECT * FROM license_keys WHERE key_code = %s"
|
||
cursor.execute(final_check_query, (key_code,))
|
||
final_info = cursor.fetchone()
|
||
|
||
if final_info and datetime.now() > final_info['end_time']:
|
||
update_query = "UPDATE license_keys SET status = 'expired' WHERE key_code = %s"
|
||
cursor.execute(update_query, (key_code,))
|
||
self.connection.commit()
|
||
cursor.close()
|
||
return False, "激活码已过期"
|
||
|
||
cursor.close()
|
||
return True, "激活成功"
|
||
|
||
except Error as e:
|
||
print(f"验证激活码错误: {e}")
|
||
return False, f"验证过程出错: {str(e)}"
|
||
|
||
def get_all_keys(self):
|
||
"""获取所有卡密信息"""
|
||
if not self.connection or not self.connection.is_connected():
|
||
if not self.connect():
|
||
return []
|
||
|
||
try:
|
||
cursor = self.connection.cursor(dictionary=True)
|
||
query = "SELECT * FROM license_keys ORDER BY created_at DESC"
|
||
cursor.execute(query)
|
||
keys = cursor.fetchall()
|
||
cursor.close()
|
||
return keys
|
||
except Error as e:
|
||
print(f"获取卡密列表错误: {e}")
|
||
return []
|
||
|
||
def update_key_status(self, key_code, status):
|
||
"""更新卡密状态"""
|
||
if not self.connection or not self.connection.is_connected():
|
||
if not self.connect():
|
||
return False
|
||
|
||
try:
|
||
cursor = self.connection.cursor()
|
||
query = "UPDATE license_keys SET status = %s WHERE key_code = %s"
|
||
cursor.execute(query, (status, key_code))
|
||
self.connection.commit()
|
||
cursor.close()
|
||
return True
|
||
except Error as e:
|
||
print(f"更新卡密状态错误: {e}")
|
||
return False
|
||
|
||
def release_key(self, key_code):
|
||
"""释放已使用的激活码 - 将其重置为未使用状态，清空机器码"""
|
||
if not self.connection or not self.connection.is_connected():
|
||
if not self.connect():
|
||
return False, "数据库连接失败"
|
||
|
||
try:
|
||
cursor = self.connection.cursor(dictionary=True)
|
||
|
||
# 检查卡密是否存在且处于已激活状态
|
||
check_query = "SELECT * FROM license_keys WHERE key_code = %s"
|
||
cursor.execute(check_query, (key_code,))
|
||
key_info = cursor.fetchone()
|
||
|
||
if not key_info:
|
||
cursor.close()
|
||
return False, "激活码不存在"
|
||
|
||
if key_info['status'] != 'active':
|
||
cursor.close()
|
||
return False, f"激活码处于 {key_info['status']} 状态，只能释放已使用的激活码"
|
||
|
||
# 释放激活码：重置为未使用状态，清空机器码和开始时间
|
||
release_query = """
|
||
UPDATE license_keys
|
||
SET status = 'unused', machine_code = NULL, start_time = NULL
|
||
WHERE key_code = %s
|
||
"""
|
||
cursor.execute(release_query, (key_code,))
|
||
rows_affected = cursor.rowcount
|
||
self.connection.commit()
|
||
cursor.close()
|
||
|
||
if rows_affected > 0:
|
||
return True, "激活码已释放，可以重新使用"
|
||
else:
|
||
return False, "释放激活码失败"
|
||
|
||
except Error as e:
|
||
print(f"释放激活码错误: {e}")
|
||
return False, f"释放过程出错: {str(e)}"
|
||
|
||
def close(self):
|
||
"""关闭数据库连接"""
|
||
if self.connection and self.connection.is_connected():
|
||
self.connection.close()
|