From 88f427aa0ef61d69288683980800fb76c55aa5e8 Mon Sep 17 00:00:00 2001
From: Abdullah Adeel <easyautosol@gmail.com>
Date: Tue, 11 Jan 2022 02:32:48 +0500
Subject: [PATCH] remove unused db folders with

---
 hooks/post_gen_project.py | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/hooks/post_gen_project.py b/hooks/post_gen_project.py
index e57430adc..96baba041 100644
--- a/hooks/post_gen_project.py
+++ b/hooks/post_gen_project.py
@@ -143,6 +143,32 @@ def remove_dotgithub_folder():
     shutil.rmtree(".github")
 
 
+def remove_postgres_env_files():
+    local_postgres_envs_path = os.path.join(".envs", ".local", ".postgres")
+    production_postgres_envs_path = os.path.join(".envs", ".production", ".postgres")
+
+    os.remove(local_postgres_envs_path)
+    os.remove(production_postgres_envs_path)
+
+
+def remove_mysql_env_files():
+    local_mysql_envs_path = os.path.join(".envs", ".local", ".mysql")
+    production_mysql_envs_path = os.path.join(".envs", ".production", ".mysql")
+
+    os.remove(local_mysql_envs_path)
+    os.remove(production_mysql_envs_path)
+
+
+def remove_postgres_docker_folder():
+    postgres_compose_path = os.path.join("compose", "production", "postgres")
+    shutil.rmtree(postgres_compose_path)
+
+
+def remove_mysql_docker_folder():
+    mysql_compose_path = os.path.join("compose", "production", "mysql")
+    shutil.rmtree(mysql_compose_path)
+
+
 def generate_random_string(
     length, using_digits=False, using_ascii_letters=False, using_punctuation=False
 ):
@@ -391,6 +417,10 @@ def main():
         remove_pycharm_files()
 
     if "{{ cookiecutter.use_docker }}".lower() == "y":
+        if "{{ cookiecutter.database_engine }}".lower() == "postgresql":
+            remove_mysql_docker_folder()
+        elif "{{ cookiecutter.database_engine }}".lower() == "mysql":
+            remove_postgres_docker_folder()
         remove_utility_files()
     else:
         remove_docker_files()
@@ -406,6 +436,11 @@ def main():
     elif "{{ cookiecutter.use_compressor }}".lower() == "n":
         remove_heroku_build_hooks()
 
+    if "{{ cookiecutter.database_engine }}".lower() == "postgresql":
+        remove_postgres_env_files()
+    elif "{{ cookiecutter.database_engine }}".lower() == "mysql":
+        remove_mysql_env_files()
+
     if (
         "{{ cookiecutter.use_docker }}".lower() == "n"
         and "{{ cookiecutter.use_heroku }}".lower() == "n"