From ed7609f7153f297b674019df3a58f053f68f25c2 Mon Sep 17 00:00:00 2001 From: "Nikita P. Shupeyko" Date: Thu, 28 Jun 2018 16:43:34 +0300 Subject: [PATCH] Remove utility/ when use_docker=y --- hooks/post_gen_project.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/hooks/post_gen_project.py b/hooks/post_gen_project.py index b75ec3d90..669edb0ef 100644 --- a/hooks/post_gen_project.py +++ b/hooks/post_gen_project.py @@ -64,6 +64,10 @@ def remove_docker_files(): os.remove(file_name) +def remove_utility_files(): + shutil.rmtree("utility") + + def remove_heroku_files(): file_names = ["Procfile", "runtime.txt", "requirements.txt"] for file_name in file_names: @@ -278,7 +282,9 @@ def main(): if "{{ cookiecutter.use_pycharm }}".lower() == "n": remove_pycharm_files() - if "{{ cookiecutter.use_docker }}".lower() == "n": + if "{{ cookiecutter.use_docker }}".lower() == "y": + remove_utility_files() + else: remove_docker_files() if "{{ cookiecutter.use_heroku }}".lower() == "n":