pyproger/generate_dotenv.py

36 lines
1.4 KiB
Python

import os
import uuid
from dotenv import load_dotenv
dotenv_path = os.path.join(os.path.dirname(__file__), "pyproger", ".env")
if os.path.exists(dotenv_path):
load_dotenv(dotenv_path)
with open(dotenv_path, "a") as f:
print("Генерирую SECRET_KEY...")
if os.getenv("SECRET_KEY") is None:
f.writelines(f"SECRET_KEY={uuid.uuid4().hex}\n")
print("_Ok_")
else:
print("SECRET_KEY уже установлен, прорускаю")
print('Генерирую "Соль"...')
if os.getenv("SECURITY_PASSWORD_SALT") is None:
f.writelines(f"SECURITY_PASSWORD_SALT={uuid.uuid4().hex}\n")
print("_Ok_")
else:
print("SECURITY_PASSWORD_SALT уже установлен, пропускаю")
if os.getenv("SQLALCHEMY_DATABASE_URI") is None:
print("Настроки подключения к базе данных Posgresql:")
login = input("Введите логин пользователя бд: ")
passwd = input("Пароль: ")
db = input("Название бд (по умолчанию pyproger):") or "pyproger"
ip = input("Адрес бд (по умолчанию localhost)") or "localhost"
port = input("Порт подключения: (по умолчанию 5432)") or "5432"
f.writelines(
f"SQLALCHEMY_DATABASE_URI=postgresql+psycopg2://{login}:{passwd}@{ip}:{port}/{db}"
)