From 123c9912d3ddaf0446c2299daca6ff081b927656 Mon Sep 17 00:00:00 2001 From: Josiah Baldwin Date: Mon, 25 Jul 2022 19:51:33 -0700 Subject: [PATCH] Move repo name to global var --- hooks/post_gen_project.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/hooks/post_gen_project.py b/hooks/post_gen_project.py index f92535a..1a363d8 100644 --- a/hooks/post_gen_project.py +++ b/hooks/post_gen_project.py @@ -1,6 +1,7 @@ import os import sys +repo_name = '{{cookiecutter.repo_name }}' def set_python_version(): python_version = str(sys.version_info.major) + "." + str(sys.version_info.minor) @@ -15,7 +16,7 @@ def set_python_version(): def remove_main_if_lib(): is_lib = '{{ cookiecutter.binary }}' - main_file_path = os.path.join('{{ cookiecutter.repo_name }}', '__main__.py') + main_file_path = os.path.join(repo_name, '__main__.py') if not (is_lib == "y" or is_lib == "Y"): os.remove(main_file_path)