toolkit/obfuscate.py

# -*- coding: utf-8 -*-
"""Лёгкая обфускация СВОЕГО кода (№9) — усложняет чтение/копирование.

Запуск:
  python obfuscate.py <файл.py> [backend]

Честно: это не криптозащита, а усложнение чтения (переименование, сжатие, уборка
комментариев). Для реальной защиты используй PyArmor (pip install pyarmor) или
PyInstaller (сборка в .exe). Модель даёт только базовую обфускацию.
"""
import sys

from brain import pick, generate


def main():
    args = sys.argv[1:]
    if not args:
        print("usage: obfuscate.py <файл.py> [backend]")
        sys.exit(2)
    path = args[0]
    try:
        code = open(path, encoding="utf-8").read()
    except Exception as e:
        print(f"не прочитан: {e}")
        sys.exit(1)
    backend = pick()
    prompt = (f"Обфусцируй этот Python-код для защиты от лёгкого копирования: переименуй "
              f"переменные и функции в короткие непонятные имена, убери комментарии и "
              f"докстринги, уплотни строки. Поведение сохрани без изменений. Верни "
              f"ТОЛЬКО обфусцированный код, без пояснений и без ```.\n\n{code}")
    print(generate(backend, [{"role": "user", "content": prompt}], 2048))


if __name__ == "__main__":
    main()