Move repo name to global var

sourcery-ai · Jul 26, 2022 · 123c991 · 123c991
1 parent 4167081
commit 123c991
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)