def __init__(self, host=None, port=None, aliases=None, timeout_connection=10): if host is None: host = self.DEFAULT_HOST if port is None: port = self.DEFAULT_PORT if aliases is None: aliases = HooqAliases() elif isinstance(aliases, basestring): aliases = HooqAliases.from_file(aliases) elif not isinstance(aliases, HooqAliases): raise TypeError("aliases must be None or str or an" " instance of HooqAliases") self.aliases = aliases self._socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) def connect(): """ try to connect """ try: self._socket.connect((host, port)) return True except socket.error, e: if e.errno != errno.ECONNREFUSED: raise return e
def __init__(self, host=None, port=None, aliases=None, timeout_connection=10): if host is None: host = self.DEFAULT_HOST if port is None: port = self.DEFAULT_PORT if aliases is None: aliases = HooqAliases() elif isinstance(aliases, basestring): aliases = HooqAliases.from_file(aliases) elif not isinstance(aliases, HooqAliases): raise TypeError("aliases must be None or str or an" " instance of HooqAliases") self.aliases = aliases def connect(): """ try to connect """ try: self._socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self._socket.connect((host, port)) return True except socket.error as e: if e.errno != errno.ECONNREFUSED: raise return e wait_for(connect, timeout_connection, 0.2) self._socket.settimeout(timeout_connection) self._fsocket = self._socket.makefile(mode="rwb")
def create_aliases(self): """ Create and returns and aliases object. """ if not self.aliases: return None return HooqAliases.from_file(self.aliases, self.global_options.funq_gkit_file, self.global_options.funq_gkit)
def _parse(self, data, gkit_data=None, gkit='default'): f = NamedTemporaryFile(delete=False) path = f.name if gkit_data: fgkit = NamedTemporaryFile(delete=False) pathgkit = fgkit.name fgkit.write(gkit_data.encode('utf-8')) fgkit.close() else: pathgkit = None try: f.write(data.encode('utf-8')) f.close() aliases = HooqAliases.from_file(path, pathgkit, gkit) finally: os.unlink(path) if pathgkit: os.unlink(pathgkit) return aliases
def _parse(self, data, gkit_data=None, gkit='default'): f = NamedTemporaryFile(delete=False) path = f.name if gkit_data: fgkit = NamedTemporaryFile(delete=False) pathgkit = fgkit.name fgkit.write(gkit_data) fgkit.close() else: pathgkit = None try: f.write(data) f.close() aliases = HooqAliases.from_file(path, pathgkit, gkit) finally: os.unlink(path) if pathgkit: os.unlink(pathgkit) return aliases