Beispiel #1
0
    def _get_status(self):
        """Gets the `IsEnabledResponse` for the capabilities listed.

    Returns:
      `IsEnabledResponse` for the specified capabilities.

    Raises:
      UnknownCapabilityError: If an unknown capability was requested.
    """
        req = IsEnabledRequest()
        req.set_package(self._package)
        for capability in self._capabilities:
            req.add_capability(capability)
        for method in self._methods:
            req.add_call(method)

        resp = capability_service_pb.IsEnabledResponse()
        if (self._package == 'datastore_v3' and 'write' in self._capabilities
                or self._methods != 0):
            self._stub_map.MakeSyncCall('capability_service', 'IsEnabled', req,
                                        resp)
        else:
            resp = IsEnabledResponse()
            resp.set_summary_status(IsEnabledResponse.ENABLED)
            for capability in self._capabilities:
                resp.add_config(package=self._package,
                                capability=capability,
                                status=CapabilityConfig.ENABLED)

        return resp
Beispiel #2
0
  def _get_status(self):
    """Get an IsEnabledResponse for the capabilities listed.

    Returns:
      IsEnabledResponse for the specified capabilities.

    Raises:
      UnknownCapabilityError: If an unknown capability was requested.
    """
    req = IsEnabledRequest()
    req.set_package(self._package)
    for capability in self._capabilities:
      req.add_capability(capability)
    for method in self._methods:
      req.add_call(method)

    resp = capability_service_pb.IsEnabledResponse()
    self._stub_map.MakeSyncCall('capability_service', 'IsEnabled', req, resp)

    if resp.summary_status() == IsEnabledResponse.UNKNOWN:
      raise UnknownCapabilityError()

    return resp