Add py.typed marker (#6)

* Add py.typed marker, improve typing, make release process automatically set package version

* Remove unneeded bound.

.github/workflows/publish.yml

@@ -29,6 +29,15 @@ jobs:
           python -m pip install --upgrade pip
           pip install poetry
 
+      - name: Set version from GITHUB_REF
+        run: >-
+          # GITHUB_REF will be `refs/tags/v1.2.3` for a tag named "v1.2.3"
+          # we strip the `refs/tags/` part to get the version number
+          # we also strip "v" from the version number, if it exists, getting "1.2.3"
+          export VERSION=$(echo ${{ github.ref }} | sed -e 's,.*/\(.*\),\1,' -e 's/^v//')
+          echo Setting version to $VERSION
+          poetry version $VERSION
+
       - name: Build wheels and source tarball
         run: >-
           poetry build

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python-main"
-version = "1.0.2"
+version = "0.0.0"
 homepage = "https://github.com/flipbit03/main"
 description = "Decorator which runs the tagged function if the current module is being run as a script. No more `if __name__ == \"__main__\"` madness."
 authors = ["Cadu <[email protected]>"]
@@ -14,6 +14,10 @@ classifiers=[
     'Natural Language :: English',
 ]
 
+packages = [
+    { include = "python_main"},
+]
+
 [tool.poetry.dependencies]
 python = "^3.9"
 

python_main/__init__.py

@@ -1,10 +1,16 @@
-from typing import Callable, Optional
+from typing import Callable, TypeVar
 
 __RAN_AS_SCRIPT_MODULE = "__main__"
 __CALLABLE_MODULE_PROP = "__module__"
 
+__MAIN_RETURN_TYPE = TypeVar("__MAIN_RETURN_TYPE")
 
-def main(f: Callable[[], Optional[int]]) -> Callable:
+
+def main(f: Callable[[], __MAIN_RETURN_TYPE]) -> Callable[[], __MAIN_RETURN_TYPE]:
     if getattr(f, __CALLABLE_MODULE_PROP) == __RAN_AS_SCRIPT_MODULE:
         f()
     return f
+
+
+# Only export the main function
+__all__ = ["main"]