explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-28 10:14:07 +03:00
Code Issues Packages Projects Releases Wiki Activity

Pass in the host when checking port availability

Browse Source
This commit is contained in:
Paul O'Leary McCann 2022-12-26 20:14:32 +09:00
parent c362acb5ce
commit c0b2472211
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
spacy/util.py
View File

@ -1739,17 +1739,18 @@ def all_equal(iterable):
return next(g, True) and not next(g, False)
def is_port_in_use(port):
def is_port_in_use(port, host="localhost"):
"""Check if localhost:port is in use."""
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
try:
s.bind(("localhost", port))
s.bind((host, port))
return False
except socket.error:
return True
finally:
s.close()
def find_available_port(start, host, auto_select_port=False):
"""Given a starting port and a host, handle finding a port.
@ -1757,22 +1758,19 @@ def find_available_port(start, host, auto_select_port=False):
If `auto_select_port` is True, the next free higher port will be used.
"""
if not is_port_in_use(start):
if not is_port_in_use(start, host):
return start
port = start
if not auto_select_port:
raise ValueError(Errors.E1049.format(port=port))
while is_port_in_use(port) and port < 65535:
while is_port_in_use(port, host) and port < 65535:
port += 1
if port == 65535 and is_port_in_use(port):
if port == 65535 and is_port_in_use(port, host):
raise ValueError(Errors.E1048.format(host=host))
# if we get here, the port changed
warnings.warn(
Warnings.W124.format(host=host, port=start, serve_port=port)
)
warnings.warn(Warnings.W124.format(host=host, port=start, serve_port=port))
return port