57 lines
1.3 KiB
Python
57 lines
1.3 KiB
Python
import sys
|
|
import socket
|
|
import chardet
|
|
|
|
|
|
class Struct(dict):
|
|
def __getattr__(self, attr):
|
|
return self[attr]
|
|
def __setattr__(self, attr, value):
|
|
self[attr] = value
|
|
|
|
|
|
def localize_path(path):
|
|
path = path.decode(chardet.detect(path)['encoding'])
|
|
if not sys.platform.startswith('win'):
|
|
path = path.encode(True and sys.getfilesystemencoding() or 'utf-8')
|
|
return path
|
|
|
|
def can_bind(host, port):
|
|
"""
|
|
Checks we can bind to specified host and port
|
|
|
|
:param host: Host
|
|
:param port: Port
|
|
:return: True if bind succeed
|
|
"""
|
|
try:
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
s.bind((host, port))
|
|
s.close()
|
|
except socket.error:
|
|
return False
|
|
return True
|
|
|
|
|
|
def find_free_port(host):
|
|
"""
|
|
Finds free TCP port that can be used for binding
|
|
|
|
:param host: Host
|
|
:return: Free port
|
|
"""
|
|
try:
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
s.bind((host, 0))
|
|
port = s.getsockname()[1]
|
|
s.close()
|
|
except socket.error:
|
|
return False
|
|
return port
|
|
|
|
|
|
def ensure_fs_encoding(string):
|
|
if isinstance(string, str):
|
|
string = string.decode('utf-8')
|
|
return string.encode(sys.getfilesystemencoding() or 'utf-8')
|