44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
|
import os
|
||
|
|
import requests
|
||
|
|
|
||
|
|
|
||
|
|
class MBSWrapper(object):
|
||
|
|
def __init__(self, api_url):
|
||
|
|
"""
|
||
|
|
:param string api_url: e.g. https://example.com/module-build-service/2
|
||
|
|
"""
|
||
|
|
self.api_url = api_url
|
||
|
|
|
||
|
|
def _get(self, resource, params=None):
|
||
|
|
"""Get specified resource.
|
||
|
|
|
||
|
|
:param string resource: e.g. module-builds, final-modulemd
|
||
|
|
:param dict data:
|
||
|
|
"""
|
||
|
|
url = os.path.join(self.api_url, resource)
|
||
|
|
try:
|
||
|
|
resp = requests.get(url, params=params)
|
||
|
|
except Exception as e:
|
||
|
|
raise Exception(
|
||
|
|
"Failed to query URL %s with params %s - %s" % (url, params, str(e))
|
||
|
|
)
|
||
|
|
resp.raise_for_status()
|
||
|
|
return resp
|
||
|
|
|
||
|
|
def module_builds(self, filters=None):
|
||
|
|
return self._get("module-builds", filters).json()
|
||
|
|
|
||
|
|
def get_module_build_by_nsvc(self, nsvc):
|
||
|
|
nsvc_list = nsvc.split(":")
|
||
|
|
if len(nsvc_list) != 4:
|
||
|
|
raise ValueError("Invalid N:S:V:C - %s" % nsvc)
|
||
|
|
filters = dict(zip(["name", "stream", "version", "context"], nsvc_list))
|
||
|
|
resp = self.module_builds(filters)
|
||
|
|
if resp["items"]:
|
||
|
|
return resp["items"][0]
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def final_modulemd(self, module_build_id):
|
||
|
|
return self._get("final-modulemd/%s" % module_build_id).json()
|