build.py

# This python script will take care of anything
# required during building the images. It is
# used by Makefile.

import random
import re
import sys


def update_env_file(key, value):
    # Update the generated secret key
    # in the .env file.

    with open('.env', 'r') as file_handle:
        file_string = file_handle.read()
    file_string = re.sub(fr'{key}=.*', fr'{key}={value}', file_string)
    if file_string[-1] != '\n':
        file_string += '\n'
    if f'{key}' not in file_string:
        file_string += f'{key}={value}'
    with open('.env', 'w') as file_handle:
        file_handle.write(file_string)


def get_secret_key(allow_special_chars=True):
    chars = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVXYZ0123456789'
    if allow_special_chars:
        chars += '#^[]-_*%&=+/'
    keygen = ''.join([random.SystemRandom().choice(chars) for _ in range(50)])
    print(keygen)
    return keygen


if __name__ == '__main__':
    arguments = sys.argv[1:]
    if 'get-secret-key' in arguments:
        get_secret_key()
    if 'change-secret-key' in arguments:
        keygen = get_secret_key()
        update_env_file('DJANGO_SECRET_KEY', keygen)
    if 'default-secret-key' in arguments:
        update_env_file('DJANGO_SECRET_KEY', 'default_secret_key')
    if 'change-database-credentials' in arguments:
        keygen1 = get_secret_key(allow_special_chars=False)
        keygen2 = get_secret_key()
        update_env_file('DB_USER', keygen1)
        update_env_file('DB_PASS', keygen2)