This works

cross_py_builder/agent_manager.py

@@ -13,8 +13,8 @@ import requests
 from packaging.version import Version
 from tabulate import tabulate
 
-from build_agent import build_agent_version
-from zeroconf_server import ZeroconfServer
+from .build_agent import build_agent_version
+from .zeroconf_server import ZeroconfServer
 
 DEFAULT_PORT = 9001
 TIMEOUT = 10
@@ -32,9 +32,9 @@ def find_server_ips():
         ZeroconfServer.stop()
 
     # get known hosts
-    with open("../known_hosts", "r") as file:
-        lines = file.readlines()
-    hostnames.extend(lines)
+    # with open("../known_hosts", "r") as file:
+    #     lines = file.readlines()
+    #     hostnames.extend(lines)
 
     return hostnames
 
@@ -293,7 +293,7 @@ def restart_agent(hostname):
 
 
 def main():
-    parser = argparse.ArgumentParser(description="Build agent manager for cross-py-builder")
+    parser = argparse.ArgumentParser(description="Build agent manager for cross_py_builder")
     parser.add_argument("--status", action="store_true", help="Get status of available servers")
     parser.add_argument("--build", type=str, help="Path to the project to build")
     parser.add_argument("--checkout", type=str, help="Url to Git repo for checkout")

cross_py_builder/build_agent.py

@@ -15,9 +15,9 @@ import glob
 import shutil
 import uuid
 import platform
-from zeroconf_server import ZeroconfServer
+from .zeroconf_server import ZeroconfServer
 
-APP_NAME = "cross-py-builder"
+APP_NAME = "cross_py_builder"
 build_agent_version = "0.1.34"
 app_port = 9001
 
@@ -394,8 +394,7 @@ def delete_cache():
     return jsonify(job_cache()), 200
 
 
-if __name__ == "__main__":
-
+def main():
     print(f"===== {APP_NAME} Build Agent (v{build_agent_version}) =====")
     ZeroconfServer.configure("_crosspybuilder._tcp.local.", socket.gethostname(), app_port)
     try:
@@ -405,3 +404,6 @@ if __name__ == "__main__":
         pass
     finally:
         ZeroconfServer.stop()
+
+if __name__ == "__main__":
+    main()

setup.py

@@ -1,14 +1,14 @@
 from setuptools import setup, find_packages
 
 setup(
-    name="cross-py-builder",
+    name="cross_py_builder",
     version="0.0.1",
     packages=find_packages(),
     install_requires=open("requirements.txt").read().splitlines(),
     entry_points={
         "console_scripts": [
-            "cross-py-agent=cross-py-builder.build_agent:main",
-            "cross-py-builder=cross-py-builder.agent_manager:main",
+            "cross-py-agent=cross_py_builder.build_agent:main",
+            "cross-py-builder=cross_py_builder.agent_manager:main",
         ],
     },
     author="Brett Williams",