iframe-proxy | Sunbelt Computer Software

87 lines (70 loc) · 3.43 KB
# Copyright 2017 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.
"""Firebase Cloud Storage module.
This module contains utilities for accessing Google Cloud Storage buckets associated with
Firebase apps. This requires the ``google-cloud-storage`` Python module.
# pylint: disable=import-error,no-name-in-module
    from google.cloud import storage
except ImportError as exception:
    raise ImportError('Failed to import the Cloud Storage library for Python. Make sure '
                      'to install the "google-cloud-storage" module.') from exception
from firebase_admin import _utils
_STORAGE_ATTRIBUTE = '_storage'
def bucket(name=None, app=None) -> storage.Bucket:
    """Returns a handle to a Google Cloud Storage bucket.
    If the name argument is not provided, uses the 'storageBucket' option specified when
    initializing the App. If that is also not available raises an error. This function
    does not make any RPC calls.
      name: Name of a cloud storage bucket (optional).
      app: An App instance (optional).
    Returns:
      google.cloud.storage.Bucket: A handle to the specified bucket.
    Raises:
      ValueError: If a bucket name is not specified either via options or method arguments,
          or if the specified bucket name is not a valid string.
    client = _utils.get_app_service(app, _STORAGE_ATTRIBUTE, _StorageClient.from_app)
    return client.bucket(name)
class _StorageClient:
    """Holds a Google Cloud Storage client instance."""
    STORAGE_HEADERS = {
        'x-goog-api-client': _utils.get_metrics_header(),
    def __init__(self, credentials, project, default_bucket):
        self._client = storage.Client(
            credentials=credentials, project=project, extra_headers=self.STORAGE_HEADERS)
        self._default_bucket = default_bucket
    @classmethod
    def from_app(cls, app):
        credentials = app.credential.get_credential()
        default_bucket = app.options.get('storageBucket')
        # Specifying project ID is not required, but providing it when available
        # significantly speeds up the initialization of the storage client.
        return _StorageClient(credentials, app.project_id, default_bucket)
    def bucket(self, name=None):
        """Returns a handle to the specified Cloud Storage Bucket."""
        bucket_name = name if name is not None else self._default_bucket
        if bucket_name is None:
            raise ValueError(
                'Storage bucket name not specified. Specify the bucket name via the '
                '"storageBucket" option when initializing the App, or specify the bucket '
                'name explicitly when calling the storage.bucket() function.')
        if not bucket_name or not isinstance(bucket_name, str):
            raise ValueError(
                f'Invalid storage bucket name: "{bucket_name}". Bucket name must be a non-empty '
                'string.')
        return self._client.bucket(bucket_name)
Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

storage.py

Latest commit

History

storage.py

File metadata and controls
