diff options
Diffstat (limited to 'examples/objectstore/views.py')
| -rw-r--r-- | examples/objectstore/views.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/examples/objectstore/views.py b/examples/objectstore/views.py index 2eaadd3c..4681fe70 100644 --- a/examples/objectstore/views.py +++ b/examples/objectstore/views.py @@ -8,6 +8,16 @@ import os import uuid OBJECT_STORE_DIR = os.path.join(settings.MEDIA_ROOT, 'objectstore') +MAX_FILES = 20 + + +def remove_oldest_files(dir, max_files): + """Remove the oldest files in a directory 'dir', leaving at most 'max_files' remaining. + We use this to limit the number of resources in the sandbox.""" + filepaths = [os.path.join(dir, file) for file in os.listdir(dir)] + ctime_sorted_paths = [item[0] for item in sorted([(path, os.path.getctime(path)) for path in filepaths], + key=operator.itemgetter(1), reverse=True)] + [os.remove(path) for path in ctime_sorted_paths[max_files:]] class ObjectStoreRoot(Resource): @@ -25,6 +35,7 @@ class ObjectStoreRoot(Resource): key = str(uuid.uuid1()) pathname = os.path.join(OBJECT_STORE_DIR, key) pickle.dump(content, open(pathname, 'wb')) + remove_oldest_files(OBJECT_STORE_DIR, MAX_FILES) return Response(status.HTTP_201_CREATED, content, {'Location': self.reverse(StoredObject, key=key)}) |