setup.py - master - yeexzdev/frontiersystem

setup.py
53 строки2 KB
yeexzdev
initial commit
15 май 2026, 20:25
15 май 2026, 20:25bc97861
import os
from dotenv import load_dotenv, set_key
from src.config import C1_CONF, C2_CONF
ENV_PATH = ".env"

def setup():
    if not os.path.exists(ENV_PATH):
        with open(ENV_PATH, "w", encoding="utf-8") as f:
            f.write("")

    load_dotenv(ENV_PATH)
    print("/// Текущая конфигурация пинов: ///")
    print(f"Контур 1:\n{C1_CONF}\n")
    print(f"Контур 2:\n{C2_CONF}\n")
    print("\n/// Настройка системы РУБЕЖ ///")
    
    token = os.getenv("VK_TOKEN", None).strip()
    group = os.getenv("VK_GROUP_ID", None).strip()
    admin = os.getenv("VK_ADMIN_ID", None).strip()
    depts = os.getenv("DEPARTMENTS", None)

    new_token = input(f"Введите VK Token, сейчас установлено значение [{token}]: ").strip() or token
    new_group = input(f"Введите Group ID, сейчас установлено значение  [{group}]: ").strip() or group
    new_admin = input(f"Введите Admin ID, сейчас установлено значение  [{admin}]: ").strip() or admin

    print("\n/// Настройка прав доступа ///")
    raw_depts = input(f"Введите ВСЕ отделы через запятую [{depts}]: ").strip() or depts
    depts_list = [d.strip() for d in raw_depts.split(",")]

    print("\nСписок отделов:")
    for i, name in enumerate(depts_list, 1):
        print(f"{i}. {name}")

    indices = input("\nВведите номера отделов для Контура 2 (например '12'): ").strip()
    
    allowed_depts = []
    for char in indices:
        if char.isdigit():
            idx = int(char) - 1
            if 0 <= idx < len(depts_list):
                allowed_depts.append(depts_list[idx])

    set_key(ENV_PATH, "VK_TOKEN", str(new_token))
    set_key(ENV_PATH, "VK_GROUP_ID", str(new_group))
    set_key(ENV_PATH, "VK_ADMIN_ID", str(new_admin))
    set_key(ENV_PATH, "DEPARTMENTS", ", ".join(depts_list))
    set_key(ENV_PATH, "C2_ALLOWED_DEPTS", ", ".join(allowed_depts))

    print("\nНастройки успешно сохранены")
    print(f"Доступ в Контур 2: {', '.join(allowed_depts)}")

if __name__ == "__main__":
    setup()