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 __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 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 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
def setup(self):
self.aliases = HooqAliases()