66 строки
2.5 KiB
Python
66 строки
2.5 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
# Copyright 2023 Google Inc.
|
||
|
#
|
||
|
# Licensed under the Apache License, Version 2.0 (the "License")
|
||
|
# you may not use this file except in compliance with the License.
|
||
|
# You may obtain a copy of the License at
|
||
|
#
|
||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
#
|
||
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
# See the License for the specific language governing permissions and
|
||
|
# limitations under the License.
|
||
|
|
||
|
from chromestatus_openapi.models import (
|
||
|
ComponentsUsersResponse,
|
||
|
OwnersAndSubscribersOfComponent,
|
||
|
ComponentsUser)
|
||
|
|
||
|
from google.cloud import ndb
|
||
|
|
||
|
from framework import basehandlers
|
||
|
from framework import permissions
|
||
|
from internals import user_models
|
||
|
|
||
|
class ComponentsUsersAPI(basehandlers.APIHandler):
|
||
|
"""The list of owners and subscribers for each component."""
|
||
|
|
||
|
@permissions.require_admin_site
|
||
|
def do_get(self, **kwargs) -> dict:
|
||
|
"""Returns a dict with 1) subscribers for each component and 2) each component."""
|
||
|
components: list[user_models.BlinkComponent] = user_models.BlinkComponent.query().order(
|
||
|
user_models.BlinkComponent.name).fetch(None)
|
||
|
possible_subscribers: list[user_models.FeatureOwner] = user_models.FeatureOwner.query().order(
|
||
|
user_models.FeatureOwner.name).fetch(None)
|
||
|
|
||
|
users = [
|
||
|
ComponentsUser(
|
||
|
id=fo.key.integer_id(), email=fo.email, name=fo.name)
|
||
|
for fo in possible_subscribers]
|
||
|
|
||
|
component_to_subscribers: dict[ndb.Key, list[int]] = {
|
||
|
c.key: [] for c in components}
|
||
|
component_to_owners: dict[ndb.Key, list[int]] = {
|
||
|
c.key: [] for c in components}
|
||
|
for ps in possible_subscribers:
|
||
|
for subed_component_key in ps.blink_components:
|
||
|
component_to_subscribers[subed_component_key].append(ps.key.integer_id())
|
||
|
for owned_component_key in ps.primary_blink_components:
|
||
|
component_to_owners[owned_component_key].append(ps.key.integer_id())
|
||
|
|
||
|
returned_components: list[OwnersAndSubscribersOfComponent] = []
|
||
|
for c in components:
|
||
|
returned_components.append(
|
||
|
OwnersAndSubscribersOfComponent(
|
||
|
id=c.key.integer_id(),
|
||
|
name=c.name,
|
||
|
subscriber_ids=component_to_subscribers[c.key],
|
||
|
owner_ids=component_to_owners[c.key]))
|
||
|
|
||
|
return ComponentsUsersResponse(
|
||
|
users=users,
|
||
|
components=returned_components[1:], # ditch generic "Blink" component
|
||
|
).to_dict()
|