v2.4

phi/cli/auth_server.py

@@ -3,7 +3,6 @@
 
 from phi.cli.settings import phi_cli_settings
 
-import socket
 
 class CliAuthRequestHandler(BaseHTTPRequestHandler):
  """Request Handler to accept the CLI auth token after the web based auth flow.
@@ -63,7 +62,7 @@ class CliAuthServer:
  Source: https://stackoverflow.com/a/38196725/10953921
  """
 
- def __init__(self, port=9191):
+ def __init__(self, port: int = 9191):
  import threading
 
  self._server = HTTPServer(("", port), CliAuthRequestHandler)
@@ -83,17 +82,25 @@ def shut_down(self):
  self._thread.close() # type: ignore
 
 
-def check_port(port):
+def check_port(port: int):
+ import socket
+
  with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
- return s.connect_ex(('localhost', port)) == 0
-
+ try:
+ return s.connect_ex(("localhost", port)) == 0
+ except Exception as e:
+ print(f"Error occurred: {e}")
+ return False
+
 
 def get_port_for_auth_server():
- if check_port(9191):
- return 9191
+ starting_port = 9191
+ for port in range(starting_port, starting_port + 100):
+ if not check_port(port):
+ return port
 
 
-def get_auth_token_from_web_flow(port) -> Optional[str]:
+def get_auth_token_from_web_flow(port: int) -> Optional[str]:
  """
  GET request: curl http:https://localhost:9191
  POST request: curl -d "foo=bar&bin=baz" http:https://localhost:9191