Added delete option to database storage.

venv/Lib/site-packages/firebase_admin/storage.py

@@ -0,0 +1,82 @@
+# 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
+try:
+    from google.cloud import storage
+except ImportError:
+    raise ImportError('Failed to import the Cloud Storage library for Python. Make sure '
+                      'to install the "google-cloud-storage" module.')
+
+from firebase_admin import _utils
+
+
+_STORAGE_ATTRIBUTE = '_storage'
+
+def bucket(name=None, app=None):
+    """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.
+
+    Args:
+      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."""
+
+    def __init__(self, credentials, project, default_bucket):
+        self._client = storage.Client(credentials=credentials, project=project)
+        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(
+                'Invalid storage bucket name: "{0}". Bucket name must be a non-empty '
+                'string.'.format(bucket_name))
+        return self._client.bucket(bucket_name)