files: set bucket quota

inveniosoftware · Sep 14, 2023 · 0146f20 · 0146f20
1 parent 192f9b1
commit 0146f20
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/invenio_records_resources/records/systemfields/files/manager.py b/invenio_records_resources/records/systemfields/files/manager.py
@@ -131,6 +131,14 @@ def set_bucket(self, bucket, overwrite=False):
         setattr(self.record, self._options["bucket_id_attr"], bucket.id)
         self._bucket = bucket
 
+    def set_quota(self, quota_size, max_file_size=None):
+        """Set bucket quota."""
+        if self.bucket is not None:
+            assert not self.bucket.locked
+            self.bucket.quota_size = quota_size
+            if max_file_size:
+                self.bucket.max_file_size = max_file_size
+
     def lock(self):
         """Lock the bucket."""
         self.bucket.locked = True