mozilla / relman-auto-nag / #5558

import urllib

from dataclasses import dataclass

from typing import Any, Iterable, Mapping, Optional

from bugbot import gcp

from bugbot.bzcleaner import BzCleaner

@dataclass

class FeatureUrls:

    feature: str

    spec_url: Optional[list[str]]

    feature_url: str

    sp_url: Optional[str]

def url_keys(urls: Iterable[str]) -> Mapping[tuple[str, str], str]:

    rv = {}

    for url in urls:

        try:

            parsed = urllib.parse.urlparse(url)

            if parsed.hostname is None:

                continue

            rv[(parsed.hostname, parsed.path)] = url

        except ValueError:

            pass

    return rv

class WebPlatformFeatures(BzCleaner):

    def __init__(self) -> None:

        super().__init__()

        self.feature_bugs: Mapping[int, FeatureUrls] = {}

    def description(self) -> str:

        return "Update See Also for web-features bugs"

    def filter_no_nag_keyword(self) -> bool:

        return False

    def has_default_products(self) -> bool:

        return False

    def columns(self) -> list[str]:

        return ["id", "summary", "added"]

    def handle_bug(

        features_key = bug["id"]

        bug_id = str(bug["id"])

        changes = {}

        if bug_id not in data:

            data[bug_id] = {}

        data[bug_id]["added"] = []

        if features_key in self.feature_bugs:

            see_also_keys = url_keys(bug["see_also"])

            feature_urls = self.feature_bugs[features_key]

            expected_urls = [feature_urls.feature_url]

            if feature_urls.sp_url is not None:

                expected_urls.append(feature_urls.sp_url)

            if feature_urls.spec_url is not None:

                expected_urls.extend(feature_urls.spec_url)

            expected_keys = url_keys(expected_urls)

            add_urls = [

            if add_urls:

                changes["see_also"] = bug["see_also"] + add_urls

                data[bug_id]["added"] = add_urls

        if changes:

            self.autofix_changes[bug_id] = changes

            return bug

        return None

    def get_bz_params(self, date) -> dict[str, str | int | list[str] | list[int]]:

        fields = ["id", "see_also"]

        self.feature_bugs = self.get_feature_bugs()

        print(self.feature_bugs)

        return {"include_fields": fields, "id": list(self.feature_bugs.keys())}

    def get_feature_bugs(self) -> Mapping[int, FeatureUrls]:

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

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

        query = f"""

        return {

if __name__ == "__main__":

    WebPlatformFeatures().run()