fix tests not working due to incorrect imports

1 files changed, 16 insertions, 0 deletions

diff --git a/src/main.py b/src/main.py
new file mode 100644
index 0000000..d705374
--- /dev/null
+++ b/src/main.py
@@ -0,0 +1,16 @@
+from discover import discover
+import os
+from logger import get_logger
+import asyncio
+
+logger = get_logger("Main")
+
+async def main(debug_flag=None):
+    check_debug = debug_flag if debug_flag is not None else os.getenv("DEBUG", "FALSE").upper() == "TRUE"
+    if check_debug:
+        logger.info("Debug mode is enabled")
+    devices = await discover(debug=check_debug)
+    print(devices)
+
+if __name__ == "__main__":
+    asyncio.run(main())