From f772825ec56f95ed96e70d7b1c7e79b428b98165 Mon Sep 17 00:00:00 2001
From: hiranya911 <hiranya911@gmail.com>
Date: Mon, 8 Jan 2018 15:03:04 -0800
Subject: [PATCH 1/2] Shallow reads support in database API

---
 firebase_admin/db.py   | 11 +++++++++--
 integration/test_db.py |  5 +++++
 tests/test_db.py       | 16 ++++++++++++++++
 3 files changed, 30 insertions(+), 2 deletions(-)

diff --git a/firebase_admin/db.py b/firebase_admin/db.py
index 6c16e6fc3..9c13ed1e6 100644
--- a/firebase_admin/db.py
+++ b/firebase_admin/db.py
@@ -125,11 +125,14 @@ def child(self, path):
         full_path = self._pathurl + '/' + path
         return Reference(client=self._client, path=full_path)
 
-    def get(self, etag=False):
+    def get(self, etag=False, shallow=False):
         """Returns the value, and optionally the ETag, at the current location of the database.
 
         Args:
           etag: A boolean indicating whether the Etag value should be returned or not (optional).
+          shallow: A boolean indicating whther to execute a shallow read (optional). Shallow reads
+              does not fetch the child nodes of the current reference. Cannot be set to True
+              if ``etag`` is also set to True.
 
         Returns:
           object: If etag is False returns the decoded JSON value of the current database location.
@@ -137,14 +140,18 @@ def get(self, etag=False):
           associated with the current database location.
 
         Raises:
+          ValueError: If both ``etag`` and ``shallow`` are set to True.
           ApiCallError: If an error occurs while communicating with the remote database server.
         """
         if etag:
+            if shallow:
+                raise ValueError('etag and shallow cannot both be set to True.')
             headers, data = self._client.headers_and_body(
                 'get', self._add_suffix(), headers={'X-Firebase-ETag' : 'true'})
             return data, headers.get('ETag')
         else:
-            return self._client.body('get', self._add_suffix())
+            params = 'shallow=true' if shallow else None
+            return self._client.body('get', self._add_suffix(), params=params)
 
     def get_if_changed(self, etag):
         """Gets data in this location only if the specified ETag does not match.
diff --git a/integration/test_db.py b/integration/test_db.py
index 6ddc4439a..c3ba2e441 100644
--- a/integration/test_db.py
+++ b/integration/test_db.py
@@ -89,6 +89,11 @@ def test_get_value_and_etag(self, testref, testdata):
         assert testdata == value
         assert isinstance(etag, six.string_types)
 
+    def test_get_shallow(self, testref):
+        value = testref.get(shallow=True)
+        assert isinstance(value, dict)
+        assert value == {'dinosaurs': True, 'scores': True}
+
     def test_get_if_changed(self, testref, testdata):
         success, data, etag = testref.get_if_changed('wrong_etag')
         assert success is True
diff --git a/tests/test_db.py b/tests/test_db.py
index 79547c282..145480faa 100644
--- a/tests/test_db.py
+++ b/tests/test_db.py
@@ -155,6 +155,22 @@ def test_get_with_etag(self, data):
         assert recorder[0].headers['User-Agent'] == db._USER_AGENT
         assert recorder[0].headers['X-Firebase-ETag'] == 'true'
 
+    @pytest.mark.parametrize('data', valid_values)
+    def test_get_shallow(self, data):
+        ref = db.reference('/test')
+        recorder = self.instrument(ref, json.dumps(data))
+        assert ref.get(shallow=True) == data
+        assert len(recorder) == 1
+        assert recorder[0].method == 'GET'
+        assert recorder[0].url == 'https://test.firebaseio.com/test.json?shallow=true'
+        assert recorder[0].headers['Authorization'] == 'Bearer mock-token'
+        assert recorder[0].headers['User-Agent'] == db._USER_AGENT
+
+    def test_get_with_etag_and_shallow(self):
+        ref = db.reference('/test')
+        with pytest.raises(ValueError):
+            ref.get(etag=True, shallow=True)
+
     @pytest.mark.parametrize('data', valid_values)
     def test_get_if_changed(self, data):
         ref = db.reference('/test')

From e21eb0f884f0519ec04b879977cfa48b4ff187f0 Mon Sep 17 00:00:00 2001
From: hiranya911 <hiranya911@gmail.com>
Date: Mon, 8 Jan 2018 15:08:11 -0800
Subject: [PATCH 2/2] Updated docstring

---
 firebase_admin/db.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/firebase_admin/db.py b/firebase_admin/db.py
index 9c13ed1e6..9fce65044 100644
--- a/firebase_admin/db.py
+++ b/firebase_admin/db.py
@@ -130,9 +130,9 @@ def get(self, etag=False, shallow=False):
 
         Args:
           etag: A boolean indicating whether the Etag value should be returned or not (optional).
-          shallow: A boolean indicating whther to execute a shallow read (optional). Shallow reads
-              does not fetch the child nodes of the current reference. Cannot be set to True
-              if ``etag`` is also set to True.
+          shallow: A boolean indicating whether to execute a shallow read (optional). Shallow
+              reads do not retrieve the child nodes of the current database location. Cannot be
+              set to True if ``etag`` is also set to True.
 
         Returns:
           object: If etag is False returns the decoded JSON value of the current database location.