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)