mozilla / relman-auto-nag / #5571

from dataclasses import dataclass

from typing import Any, Mapping, Optional

from bugbot import gcp

from bugbot.bzcleaner import BzCleaner

@dataclass(frozen=True)

class MetricType:

    field: str

    whiteboard_entry: str

metrics = [

class WebcompatSightline(BzCleaner):

    def __init__(self):

        super().__init__()

        self.metric_bugs = {}

    def description(self) -> str:

        return "Bugs with the [webcompat:<metric name>] whiteboard tag updated"

    def filter_no_nag_keyword(self) -> bool:

        return False

    def has_default_products(self) -> bool:

        return False

    def handle_bug(

        bug_id = str(bug["id"])

        whiteboard = bug["whiteboard"]

        bug_metrics = self.metric_bugs[bug["id"]]

        for metric, include in bug_metrics.items():

            if include and metric.whiteboard_entry not in whiteboard:

                whiteboard += metric.whiteboard_entry

            elif not include and metric.whiteboard_entry in whiteboard:

                whiteboard = whiteboard.replace(metric.whiteboard_entry, "")

        if whiteboard != bug["whiteboard"]:

            self.autofix_changes[bug_id] = {"whiteboard": whiteboard}

            return bug

        return None

    def get_bz_params(self, date) -> dict[str, Any]:

        fields = ["id", "summary", "whiteboard"]

        self.metric_bugs = self.get_metric_bugs()

        return {

    def get_metric_bugs(self) -> Mapping[int, Mapping[MetricType, bool]]:

        project = "moz-fx-dev-dschubert-wckb"

        dataset = "webcompat_knowledge_base"

        fields = []

        conditions = []

        for metric in metrics:

            fields.append(metric.field)

            conditions.append(

        client = gcp.get_bigquery_client(project, ["cloud-platform", "drive"])

        query = f"""

        results = {}

        for row in client.query(query).result():

            result = {metric: row[metric.field] for metric in metrics}

            results[row.number] = result

        return results

if __name__ == "__main__":

    WebcompatSightline().run()