cookiecutter/cookiecutter-django
1
1
Fork 0
mirror of https://github.com/cookiecutter/cookiecutter-django.git synced 2025-08-18 19:04:51 +03:00
Code Issues Packages Projects Releases Wiki Activity

setup .envs according to database

Browse Source
This commit is contained in:
Abdullah Adeel 2022-01-10 01:46:56 +05:00
parent 46ab64b4b9
commit 1594cc6924
1 changed files with 58 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
hooks/post_gen_project.py
View File

@ -217,25 +217,51 @@ def generate_random_user():
return generate_random_string(length=32, using_ascii_letters=True) return generate_random_string(length=32, using_ascii_letters=True)
def generate_postgres_user(debug=False): def generate_database_user(debug=False):
return DEBUG_VALUE if debug else generate_random_user() return DEBUG_VALUE if debug else generate_random_user()
def set_postgres_user(file_path, value): def set_database_user(file_path: str, value: str, database_engine: str):
postgres_user = set_flag(file_path, "!!!SET POSTGRES_USER!!!", value=value) database_user = set_flag(
return postgres_user
def set_postgres_password(file_path, value=None):
postgres_password = set_flag(
file_path, file_path,
"!!!SET POSTGRES_PASSWORD!!!", f"!!!SET {database_engine.upper()}_USER!!!",
value=value)
return database_user
def set_database_password(file_path: str, database_engine: str, value: str | None=None):
database_password = set_flag(
file_path,
f"!!!SET {database_engine.upper()}_USER!!!",
value=value, value=value,
length=64, length=64,
using_digits=True, using_digits=True,
using_ascii_letters=True, using_ascii_letters=True,
) )
return postgres_password return database_password
def get_database_env_path(env: str, database_engine: str):
local_postgres_envs_path = os.path.join(".envs", ".local", ".postgres")
production_postgres_envs_path = os.path.join(".envs", ".production", ".postgres")
local_mysql_envs_path = os.path.join(".envs", ".local", ".mysql")
production_mysql_envs_path = os.path.join(".envs", ".production", ".mysql")
is_mysql = database_engine == "mysql"
is_postgres = database_engine == "postgresql"
if env == "local":
if is_mysql:
return local_mysql_envs_path
if is_postgres:
return local_postgres_envs_path
if env == "prod":
if is_mysql:
return production_mysql_envs_path
if is_postgres:
return production_postgres_envs_path
return None
def set_celery_flower_user(file_path, value): def set_celery_flower_user(file_path, value):
@ -263,22 +289,35 @@ def append_to_gitignore_file(ignored_line):
gitignore_file.write("\n") gitignore_file.write("\n")
def set_flags_in_envs(postgres_user, celery_flower_user, debug=False): def set_flags_in_envs(database_user, celery_flower_user, debug=False):
local_django_envs_path = os.path.join(".envs", ".local", ".django") local_django_envs_path = os.path.join(".envs", ".local", ".django")
production_django_envs_path = os.path.join(".envs", ".production", ".django") production_django_envs_path = os.path.join(".envs", ".production", ".django")
local_postgres_envs_path = os.path.join(".envs", ".local", ".postgres")
production_postgres_envs_path = os.path.join(".envs", ".production", ".postgres") selected_database = "{{ cookiecutter.database_engine }}"
set_django_secret_key(production_django_envs_path) set_django_secret_key(production_django_envs_path)
set_django_admin_url(production_django_envs_path) set_django_admin_url(production_django_envs_path)
set_postgres_user(local_postgres_envs_path, value=postgres_user) set_database_user(
set_postgres_password( get_database_env_path(env="local", database_engine=selected_database),
local_postgres_envs_path, value=DEBUG_VALUE if debug else None value=database_user,
database_engine=selected_database
) )
set_postgres_user(production_postgres_envs_path, value=postgres_user) set_database_password(
set_postgres_password( get_database_env_path(env="local", database_engine=selected_database),
production_postgres_envs_path, value=DEBUG_VALUE if debug else None database_engine=selected_database,
value=DEBUG_VALUE if debug else None
)
set_database_user(
get_database_env_path(env="prod", database_engine=selected_database),
value=database_user,
database_engine=selected_database
)
set_database_password(
get_database_env_path(env="prod", database_engine=selected_database),
database_engine=selected_database,
value=DEBUG_VALUE if debug else None
) )
set_celery_flower_user(local_django_envs_path, value=celery_flower_user) set_celery_flower_user(local_django_envs_path, value=celery_flower_user)