49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
import tempfile
|
||
from zipfile import ZipFile
|
||
from pathlib import Path
|
||
from ..utils.encoding import variant_base64_encode
|
||
from ..utils.crypto import encrypt_bytes
|
||
|
||
|
||
class LicenseType:
|
||
Professional: int = 1
|
||
Educational: int = 3
|
||
Personal: int = 4
|
||
|
||
|
||
def generate_license_file(
|
||
user_name: str,
|
||
version: str,
|
||
lic_type: int = LicenseType.Professional,
|
||
count: int = 1
|
||
) -> tuple[str, tempfile.TemporaryDirectory]:
|
||
"""
|
||
Генерация временного ZIP-файла с лицензией Custom.mxtpro.
|
||
Возвращает путь к файлу и объект временной директории для cleanup.
|
||
"""
|
||
# Разбор версии
|
||
try:
|
||
major_str, minor_str = version.split(".")
|
||
major_version: int = int(major_str)
|
||
minor_version: int = int(minor_str)
|
||
except Exception as e:
|
||
raise ValueError(f"Неверный формат версии: {version}") from e
|
||
|
||
# Формирование строки лицензии
|
||
license_str: str = (
|
||
f"{lic_type}#{user_name}|{major_version}{minor_version}"
|
||
f"#{count}#{major_version}3{minor_version}6{minor_version}#0#0#0#"
|
||
)
|
||
|
||
# Шифрование и кодирование
|
||
encrypted: bytes = encrypt_bytes(0x0787, license_str.encode("utf-8"))
|
||
encoded: str = variant_base64_encode(encrypted).decode("ascii")
|
||
|
||
# Создание временной директории и файла с фиксированным именем
|
||
temp_dir = tempfile.TemporaryDirectory()
|
||
filepath = Path(temp_dir.name) / "Custom.mxtpro"
|
||
with ZipFile(filepath, "w") as zf:
|
||
zf.writestr("Pro.key", encoded)
|
||
|
||
return str(filepath), temp_dir
|