fix entrypoint pointing to main

EnriqueSoria · Nov 1, 2022 · beb77e5 · beb77e5
1 parent 5b82495
commit beb77e5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,7 +12,7 @@ click = "^8.1.3"
 [tool.poetry.dev-dependencies]
 
 [tool.poetry.scripts]
-update_django_environ_sample = "update_django_environ_sample"
+update_django_environ_sample = "update_django_environ_sample:main"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]

diff --git a/update_django_environ_sample/__init__.py b/update_django_environ_sample/__init__.py
@@ -57,5 +57,9 @@ def generate_env_sample(*, environment_defaults_path: str, output: str, defaults
         f.write(text)
 
 
-if __name__ == "__main__":
+def main():
     generate_env_sample()
+
+
+if __name__ == "__main__":
+    main()