generic entrypoint

andrew-ld · Mar 23, 2021 · 2bbca28 · 2bbca28
1 parent 6114a8c
commit 2bbca28
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
     python_requires=">=3.6",
     entry_points={
         "console_scripts": [
-            "smart_tv_telegram=smart_tv_telegram.__main__:arg_parser",
+            "smart_tv_telegram=smart_tv_telegram.__main__:entry_point",
         ],
     },
     include_package_data=True

diff --git a/smart_tv_telegram/__main__.py b/smart_tv_telegram/__main__.py
@@ -64,7 +64,7 @@ def arg_parser(devices: DeviceFinderCollection):
     main(args.config, devices)
 
 
-if __name__ == "__main__":
+def entry_point():
     _devices = DeviceFinderCollection()
     _devices.register_finder(UpnpDeviceFinder())
     _devices.register_finder(ChromecastDeviceFinder())
@@ -73,3 +73,7 @@ def arg_parser(devices: DeviceFinderCollection):
     _devices.register_finder(XbmcDeviceFinder())
 
     arg_parser(_devices)
+
+
+if __name__ == "__main__":
+    entry_point()