script.module.pyrrent2http/lib/pyrrent2http/util.py

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')