121 lines
3.8 KiB
Python
121 lines
3.8 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
许可证生成工具类
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from app import db
|
|||
|
|
from app.models import License, Product, Package
|
|||
|
|
from datetime import datetime, timedelta
|
|||
|
|
import secrets
|
|||
|
|
import string
|
|||
|
|
|
|||
|
|
|
|||
|
|
class LicenseGenerator:
|
|||
|
|
"""许可证生成器"""
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
"""初始化"""
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def generate_license_key(self, length=32, prefix=''):
|
|||
|
|
"""
|
|||
|
|
生成卡密（格式：XXXX-XXXX-XXXX-XXXX）
|
|||
|
|
:param length: 卡密长度
|
|||
|
|
:param prefix: 卡密前缀
|
|||
|
|
:return: 生成的卡密
|
|||
|
|
"""
|
|||
|
|
# 生成随机字符串
|
|||
|
|
chars = string.ascii_uppercase + string.digits
|
|||
|
|
# 生成32位字符（4组，每组8位）
|
|||
|
|
random_chars = ''.join(secrets.choice(chars) for _ in range(32 - len(prefix)))
|
|||
|
|
|
|||
|
|
# 格式化为XXXX-XXXX-XXXX-XXXX格式
|
|||
|
|
formatted_key = '-'.join([
|
|||
|
|
random_chars[i:i+8] for i in range(0, len(random_chars), 8)
|
|||
|
|
])
|
|||
|
|
|
|||
|
|
# 组合前缀和格式化后的密钥
|
|||
|
|
license_key = prefix + formatted_key if prefix else formatted_key
|
|||
|
|
|
|||
|
|
# 确保唯一性
|
|||
|
|
while License.query.filter_by(license_key=license_key).first():
|
|||
|
|
random_chars = ''.join(secrets.choice(chars) for _ in range(32 - len(prefix)))
|
|||
|
|
formatted_key = '-'.join([
|
|||
|
|
random_chars[i:i+8] for i in range(0, len(random_chars), 8)
|
|||
|
|
])
|
|||
|
|
license_key = prefix + formatted_key if prefix else formatted_key
|
|||
|
|
|
|||
|
|
return license_key
|
|||
|
|
|
|||
|
|
def generate_license(self, product_id, package_id, contact_person, phone, quantity=1, license_type=1):
|
|||
|
|
"""
|
|||
|
|
生成许可证
|
|||
|
|
:param product_id: 产品ID
|
|||
|
|
:param package_id: 套餐ID
|
|||
|
|
:param contact_person: 联系人
|
|||
|
|
:param phone: 手机号
|
|||
|
|
:param quantity: 数量
|
|||
|
|
:param license_type: 许可证类型 (0=试用, 1=正式)
|
|||
|
|
:return: 生成的许可证密钥
|
|||
|
|
"""
|
|||
|
|
# 查询产品和套餐
|
|||
|
|
product = Product.query.filter_by(product_id=product_id).first()
|
|||
|
|
if not product:
|
|||
|
|
raise ValueError(f"产品不存在: {product_id}")
|
|||
|
|
|
|||
|
|
package = Package.query.filter_by(package_id=package_id).first()
|
|||
|
|
if not package:
|
|||
|
|
raise ValueError(f"套餐不存在: {package_id}")
|
|||
|
|
|
|||
|
|
# 计算有效期
|
|||
|
|
if package.duration == -1: # 永久卡
|
|||
|
|
expire_time = None
|
|||
|
|
else:
|
|||
|
|
expire_time = datetime.utcnow() + timedelta(days=package.duration)
|
|||
|
|
|
|||
|
|
# 生成许可证密钥
|
|||
|
|
license_key = self.generate_license_key()
|
|||
|
|
|
|||
|
|
# 创建许可证记录
|
|||
|
|
license_obj = License(
|
|||
|
|
license_key=license_key,
|
|||
|
|
product_id=product_id,
|
|||
|
|
package_id=package_id,
|
|||
|
|
contact_person=contact_person,
|
|||
|
|
phone=phone,
|
|||
|
|
license_type=license_type,
|
|||
|
|
expire_time=expire_time,
|
|||
|
|
max_devices=package.max_devices,
|
|||
|
|
status=1 # 启用
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 保存到数据库
|
|||
|
|
db.session.add(license_obj)
|
|||
|
|
db.session.commit()
|
|||
|
|
|
|||
|
|
return license_key
|
|||
|
|
|
|||
|
|
def generate_batch(self, product_id, package_id, contact_person, phone, count=1, license_type=1):
|
|||
|
|
"""
|
|||
|
|
批量生成许可证
|
|||
|
|
:param product_id: 产品ID
|
|||
|
|
:param package_id: 套餐ID
|
|||
|
|
:param contact_person: 联系人
|
|||
|
|
:param phone: 手机号
|
|||
|
|
:param count: 数量
|
|||
|
|
:param license_type: 许可证类型
|
|||
|
|
:return: 生成的许可证密钥列表
|
|||
|
|
"""
|
|||
|
|
license_keys = []
|
|||
|
|
for _ in range(count):
|
|||
|
|
license_key = self.generate_license(
|
|||
|
|
product_id=product_id,
|
|||
|
|
package_id=package_id,
|
|||
|
|
contact_person=contact_person,
|
|||
|
|
phone=phone,
|
|||
|
|
quantity=1,
|
|||
|
|
license_type=license_type
|
|||
|
|
)
|
|||
|
|
license_keys.append(license_key)
|
|||
|
|
|
|||
|
|
return license_keys
|