From beb77e5d713317914dbdf557eb16ad946d1fc55b Mon Sep 17 00:00:00 2001
From: Enrique Soria <me@enriquesoria.com>
Date: Tue, 1 Nov 2022 21:15:44 +0100
Subject: [PATCH] fix entrypoint pointing to main

---
 pyproject.toml                           | 2 +-
 update_django_environ_sample/__init__.py | 6 +++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 5bce68f..3d23f01 100644
--- 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
index fad315e..ada4a0f 100644
--- 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()