diff options
author | Victor Shyba <victor1984@riseup.net> | 2017-07-13 02:02:39 -0300 |
---|---|---|
committer | drebs <drebs@leap.se> | 2017-07-21 10:58:47 -0300 |
commit | 5d521e5a1c5fdd4cf926062dc26147dbcd110b05 (patch) | |
tree | 97a2e5e7bb20c3b01e89ca5ab79cc1cc124502e9 /src/leap/soledad/server/_blobs.py | |
parent | 64107009a869a6ddb6cea129e65735d9740e697b (diff) |
[feature] add possibility to count on blob manager
By using `only_count=True` parameter, instead of returning a full list,
server just count the amount of blobs.
-- Resolves: #8871
Diffstat (limited to 'src/leap/soledad/server/_blobs.py')
-rw-r--r-- | src/leap/soledad/server/_blobs.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/src/leap/soledad/server/_blobs.py b/src/leap/soledad/server/_blobs.py index 2a0e1d57..5abb0b02 100644 --- a/src/leap/soledad/server/_blobs.py +++ b/src/leap/soledad/server/_blobs.py @@ -141,6 +141,13 @@ class FilesystemBlobsBackend(object): def get_blob_size(user, blob_id, namespace=''): raise NotImplementedError + def count(self, user, request, namespace=''): + base_path = self._get_path(user, custom_preffix=namespace) + count = 0 + for _, _, filenames in os.walk(base_path): + count += len(filenames) + return json.dumps({"count": count}) + def list_blobs(self, user, request, namespace='', order_by=None): blob_ids = [] base_path = self._get_path(user, custom_preffix=namespace) @@ -227,7 +234,9 @@ class BlobsResource(resource.Resource): def render_GET(self, request): logger.info("http get: %s" % request.path) user, blob_id, namespace = self._validate(request) - if not blob_id: + if not blob_id and request.args.get('only_count', False): + return self._handler.count(user, request, namespace) + elif not blob_id: order = request.args.get('order_by', [None])[0] return self._handler.list_blobs(user, request, namespace, order_by=order) |