Greenwave query class and configuration update

Signed-off-by: Valerij Maljulin <vmaljuli@redhat.com>

module_build_service/builder/KojiContentGenerator.py

@@ -295,14 +295,9 @@ class KojiContentGenerator(object):
         return rpms
 
     def _get_build(self):
-        ret = {}
-        ret[u"name"] = self.module.name
+        ret = self.module.nvr
         if self.devel:
             ret["name"] += "-devel"
-        ret[u"version"] = self.module.stream.replace("-", "_")
-        # Append the context to the version to make NVRs of modules unique in the event of
-        # module stream expansion
-        ret[u"release"] = "{0}.{1}".format(self.module.version, self.module.context)
         ret[u"source"] = self.module.scmurl
         ret[u"start_time"] = calendar.timegm(self.module.time_submitted.utctimetuple())
         ret[u"end_time"] = calendar.timegm(self.module.time_completed.utctimetuple())

module_build_service/config.py

@@ -603,6 +603,22 @@ class Config(object):
             "corresponding suffix added to formatted stream version. "
             'For example, {r"regexp": 0.1, ...}',
         },
+        "greenwave_url": {
+            "type": str,
+            "default": "",
+            "desc": "The URL of the server where Greenwave is running (should include "
+                    "the root of the API)"
+        },
+        "greenwave_subject_type": {
+            "type": str,
+            "default": "",
+            "desc": "Subject type for Greenwave requests"
+        },
+        "greenwave_timeout": {
+            "type": int,
+            "default": 60,
+            "desc": "Greenwave response timeout"
+        }
     }
 
     def __init__(self, conf_section_obj):

module_build_service/errors.py

@@ -55,6 +55,10 @@ class StreamAmbigous(ValueError):
     pass
 
 
+class GreenwaveError(RuntimeError):
+    pass
+
+
 def json_error(status, error, message):
     response = jsonify({"status": status, "error": error, "message": message})
     response.status_code = status

module_build_service/models.py

@@ -34,6 +34,7 @@ from collections import OrderedDict
 from datetime import datetime
 
 import sqlalchemy
+import kobo.rpmlib
 from flask import has_app_context
 from sqlalchemy import func, and_
 from sqlalchemy.orm import lazyload
@@ -601,6 +602,18 @@ class ModuleBuild(MBSBase):
         )
         return [build.id for build in query.all()]
 
+    @property
+    def nvr(self):
+        return {
+            u"name": self.name,
+            u"version": self.stream.replace("-", "_"),
+            u"release": "{0}.{1}".format(self.version, self.context)
+        }
+
+    @property
+    def nvr_string(self):
+        return kobo.rpmlib.make_nvr(self.nvr)
+
     @classmethod
     def create(
         cls,

module_build_service/utils/greenwave.py

@@ -0,0 +1,116 @@
+# -*- coding: utf-8 -*-
+# Copyright (c) 2019  Red Hat, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+# Written by Valerij Maljulin <vmaljuli@redhat.com>
+
+
+import requests
+import json
+from module_build_service import log, conf
+from module_build_service.errors import GreenwaveError
+
+
+class Greenwave(object):
+    def __init__(self):
+        """
+        Initialize greenwave instance with config
+        """
+        self.url = conf.greenwave_url
+        if not self.url:
+            raise GreenwaveError("No Greenwave URL set")
+        self._decision_context = conf.greenwave_decision_context
+        if not self.decision_context:
+            raise GreenwaveError("No Greenwave decision context set")
+        self._subj_type = conf.greenwave_subject_type
+        self._gw_timeout = conf.greenwave_timeout
+
+    def query_decision(self, build, prod_version):
+        """
+        Query decision to greenwave
+        :param build: build object
+        :type build: module_build_service.models.ModuleBuild
+        :param prod_version: The product version string used for querying WaiverDB
+        :type prod_version: str
+        :return: response
+        :rtype: dict
+        """
+        payload = {
+            "decision_context": self.decision_context,
+            "product_version": prod_version,
+            "subject_type": self.subject_type,
+            "subject_identifier": build.nvr_string
+        }
+        url = "{0}/decision".format(self.url)
+        headers = {"Content-Type": "application/json"}
+        try:
+            response = requests.post(
+                url=url, headers=headers, data=json.dumps(payload), timeout=self.timeout)
+        except requests.exceptions.Timeout:
+            raise GreenwaveError("Greenwave request timed out")
+        except Exception as exc:
+            log.exception(str(exc))
+            raise GreenwaveError("Greenwave request error")
+
+        try:
+            resp_json = response.json()
+        except ValueError:
+            log.debug("Greenwave response content (status {0}): {1}".format(
+                response.status_code, response.text))
+            raise GreenwaveError("Greenwave returned invalid JSON.")
+
+        log.debug('Query to Greenwave result: status=%d, content="%s"',
+                  (response.status_code, resp_json))
+
+        if response.status_code == 200:
+            return resp_json
+
+        try:
+            err_msg = resp_json["message"]
+        except KeyError:
+            err_msg = response.text
+        raise GreenwaveError("Greenwave returned {0} status code. Message: {1}".format(
+            response.status_code, err_msg))
+
+    @property
+    def url(self):
+        return self._url
+
+    @url.setter
+    def url(self, value):
+        value = value.rstrip("/")
+        if value:
+            self._url = value
+
+    @property
+    def decision_context(self):
+        return self._decision_context
+
+    @property
+    def subject_type(self):
+        return self._subj_type
+
+    @property
+    def timeout(self):
+        return self._gw_timeout
+
+    @timeout.setter
+    def timeout(self, value):
+        self._gw_timeout = value