initialize.py

from django.utils.crypto import get_random_string
import subprocess
from shutil import copyfile
import os
from project.utils import generate_prime_number


def request_input(question, options):
    print(f'{question} ({", ".join([option for option in options])})')
    while True:
        r = input()
        if r not in options:
            print('Digite uma resposta válida!')
        else:
            return r


def ask_user(name, title, required, default=None, next_questions={}):
    print()
    print(f'{title} {"(Opcional)" if not required else ""}')
    if list(next_questions.keys()) != []:
        print(f'Opções: {list(next_questions.keys())}')
    if default is not None:
        default = str(default)
        print(f'Não digite nada para escolher "{default}"')
    answer = input(f'{name}: ')
    if required and next_questions != {}:
        while answer not in list(next_questions.keys()):
            print(f'{answer} deve estar em {list(next_questions.keys())}!')
            answer = input(f'{name}: ')
        for question in next_questions[answer]:
            process_question(question)
    if required and answer == '':
        if default is not None:
            answer = default
        while answer == '':
            print(f'{name} é obrigatório!')
            answer = input(f'{name}: ')
    return answer


def process_question(question):
    name = question['name']
    title = question['title']
    default = question.get('default', None)
    ask = question.get('ask', True)
    required = question.get('required', False)
    next_questions = question.get('next', {})
    value = default
    if ask:
        value = ask_user(name, title, required, default, next_questions)
    env_string = f'\n# {title}\n{name}={value}\n'
    with open(".env-temp", "a") as file:
        file.write(env_string)


def process_questions(questions):
    with open(".env-temp", "w+") as file:
        file.write('# Environment Variables\n')
    for question in questions:
        process_question(question)


env = [
    {
        'name': 'DEBUG',
        'title': 'Ativar modo de debug',
        'default': True,
        'ask': False
    },
    {
        'name': 'SECRET_KEY',
        'title': 'Chave privada de encriptação usada pelo Django',
        'default': get_random_string(32),
        'ask': False,
    },
    {
        'name': 'SECRET_PRIME',
        'title': 'Número primo de 1024 bits usado para encriptação',
        'default': generate_prime_number(),
        'ask': False,
    },
    {
        'name': 'EMAIL_ACCOUNT',
        'title': 'Endereço de email para envio de logs de erro',
        'ask': True,
        'required': True
    },
    {
        'name': 'EMAIL_PASSWORD',
        'title': 'Senha do endereço de email',
        'ask': True,
        'required': True
    },
    {
        'name': 'ADMIN_ACCOUNT',
        'title': 'Conta de admin para receber logs de erro',
        'default': 'gustavomaronato@gmail.com',
        'ask': True,
        'required': True
    },
    {
        'name': 'ROOT_URL',
        'title': 'URL base do seu app',
        'default': 'localhost:8000',
        'ask': True,
        'required': True
    },
    {
        'name': 'GEOCODING_API_KEY',
        'title': 'Chave de API de Geocoding da Google',
        'ask': True,
        'required': True
    },
    {
        'name': 'SHOW_TOOLBAR_CALLBACK',
        'title': 'Mostrar toolbar de debug do Django',
        'default': True,
        'ask': True,
        'required': True
    },
    {
        'name': 'DATABASE_URL',
        'title': 'URI para seu banco de dados Postgres',
        'default': 'postgresql://localhost/',
        'ask': True,
        'required': True
    },
    {
        'name': 'REDIS_URL',
        'title': 'URI da sua instância Redis',
        'default': 'redis://localhost:6379/0',
        'ask': True,
        'required': True
    },
    {
        'name': 'MAINTENANCE_MODE',
        'title': 'Modo de manutenção do site',
        'default': False,
        'ask': False,
        'required': True
    },
    {
        'name': 'RECAPTCHA_PUBLIC_KEY',
        'title': 'Chave pública do reCaptcha',
        'default': False,
        'ask': True,
        'required': True
    },
    {
        'name': 'RECAPTCHA_PRIVATE_KEY',
        'title': 'Chave privada do reCaptcha',
        'default': False,
        'ask': True,
        'required': True
    },
    {
        'name': 'AWS_ACCESS_KEY_ID',
        'title': 'ID do seu usuário IAM na AWS. Usado para salvar logos de app',
        'ask': True,
        'required': True
    },
    {
        'name': 'AWS_SECRET_ACCESS_KEY',
        'title': 'Secret do seu usuário IAM na AWS. Usado para salvar logos de app',
        'ask': True,
        'required': True
    },
    {
        'name': 'AWS_STORAGE_BUCKET_NAME',
        'title': 'Bucket na AWS. Usado para salvar logos de app',
        'ask': True,
        'required': True
    },
    {
        'name': 'BLABLACAR_API_KEY',
        'title': 'API key do BlaBlaCar para pesquisas',
        'ask': True,
        'required': True
    }
]


def revert_changes():
    print('\n')
    print('Revertendo alterações...')
    os.remove('.env-temp')


def process_env():
    try:
        process_questions(env)
        r = request_input('Salvar alterações?', ['s', 'n'])
        if r == 's':
            print('Salvando...')
            copyfile('.env-temp', '.env')
            os.remove('.env-temp')
        else:
            revert_changes()
    except KeyboardInterrupt:
        revert_changes()


def process_db():
    r = request_input('Migrar banco de dados?', ['s', 'n'])
    if r == 's':
        print('Migrando banco de dados...')
        print(subprocess.check_output(['python', 'manage.py', 'migrate']).decode("utf-8"))


if __name__ == "__main__":
    ask_questions = True
    if os.path.isfile('.env'):
        print('Arquivo de variáveis de ambiente já existe')
        r = request_input('Deseja substituí-lo?', ['s', 'n'])
        if r == 'n':
            ask_questions = False
    if ask_questions:
        process_env()
    process_db()