class AppResponseServiceDefLoader(ServiceDefLoadHook):
"""This class serves as the custom hook for service definition manager
for AppResponse devices.
"""
SERVICE_ID = '/api/{name}/{version}'
def __init__(self, connection):
"""Initialize AppResponse object.
:param connection: Connection object to the AppResponse appliance.
should be an instance of
:py:class:`Connection<steelscript.common.connection.Connection`
"""
self.connection = connection
self.ss_dir = SteelScriptDir('AppResponse', 'files')
def get_fnames(self, name, version):
"""Return both the base filename and absolute file name of the
Service Def file.
:param str name: name of the service
:param str version: version string
:return: base file name and absolute file name
"""
rel_fname = name + '-' + version + '.yml'
abs_fname = os.path.join(self.ss_dir.basedir, rel_fname)
return rel_fname, abs_fname
def find_by_id(self, id_):
"""Return ServiceDef object of the given service.
:param str id_: uri of the service definition
"""
_, name, version = id_.rsplit('/', 2)
rel_fname, abs_fname = self.get_fnames(name, version)
if self.ss_dir.isfile(rel_fname):
return ServiceDef.create_from_file(abs_fname)
resp = self.connection.request(method='GET', path=id_)
# Write the yaml file
with open(abs_fname, 'w+') as f:
yaml.safe_dump(resp.json(), f)
return ServiceDef.create_from_file(abs_fname)
def find_by_name(self, name, version, provider):
"""Return ServiceDef object of the given service and version."""
assert (provider == 'riverbed')
rel_fname, abs_fname = self.get_fnames(name, version)
if self.ss_dir.isfile(rel_fname):
# Found cache file
return ServiceDef.create_from_file(abs_fname)
service_id = self.SERVICE_ID.format(name=name, version=version)
return self.find_by_id(service_id)
