wan-22-comfy/generate_keys.py

#!/usr/bin/env python3
"""
Генерация пары ключей для HMAC авторизации.
Запусти один раз, сохрани ключи на клиенте и сервере.

Использование:
    python generate_keys.py
"""

import secrets
import json
import os


def generate_keys():
    """Генерирует пару ключей: client_id (публичный) и secret_key (секретный)."""
    client_id = f"client_{secrets.token_hex(8)}"
    secret_key = secrets.token_hex(32)

    keys = {
        "client_id": client_id,
        "secret_key": secret_key
    }

    print("=" * 60)
    print("🔐 Сгенерированы ключи для HMAC авторизации")
    print("=" * 60)
    print(f"\n  Client ID (публичный):  {client_id}")
    print(f"  Secret Key (секретный): {secret_key}")
    print()

    # Сохраняем в файл
    keys_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), "keys.json")
    with open(keys_file, "w") as f:
        json.dump(keys, f, indent=2)

    print(f"✅ Ключи сохранены в: {keys_file}")
    print()
    print("⚠️  ВАЖНО:")
    print("   1. Скопируй keys.json на СЕРВЕР в папку custom_comfyui/")
    print("   2. Скопируй keys.json на КЛИЕНТ (твой ПК)")
    print("   3. НЕ коммить keys.json в git!")
    print("=" * 60)


if __name__ == "__main__":
    generate_keys()