chore: formatting

Author: carlinmack

invenio_rdm_records/proxies.py

@@ -43,4 +43,4 @@
 current_rdm_records_storage_service = LocalProxy(
     lambda: current_app.extensions["invenio-rdm-records"].storage_service
 )
-"""Helper proxy to get the current RDM-Records storage service."""
+"""Helper proxy to get the current RDM-Records storage service."""

invenio_rdm_records/services/storage/service.py

@@ -5,14 +5,19 @@
 # Invenio-RDM-Records is free software; you can redistribute it and/or modify
 # it under the terms of the MIT License; see LICENSE file for more details.
 
+"""Storage Service."""
+
 import logging
+
 from flask import current_app
 from invenio_access.permissions import system_identity
 from invenio_search.engine import dsl
+
 from invenio_rdm_records.records.models import RDMRecordQuota
 
 logger = logging.getLogger(__name__)
 
+
 class StorageService:
     """Service providing per-user storage quota information."""
 
@@ -28,7 +33,10 @@ def default_quota(self):
     @property
     def max_additional_quota(self):
         """Get the maximum additional quota allowed per user."""
-        return current_app.config.get("RDM_FILES_DEFAULT_MAX_ADDITIONAL_QUOTA_SIZE") or 150 * 10**9
+        return (
+            current_app.config.get("RDM_FILES_DEFAULT_MAX_ADDITIONAL_QUOTA_SIZE")
+            or 150 * 10**9
+        )
 
     def _search_user_resources(self, user, drafts=False):
         """Fetch user records or drafts."""
@@ -47,7 +55,9 @@ def _search_user_resources(self, user, drafts=False):
 
     def _resolve_records_and_quotas(self, items, draft=False):
         """Resolve search hits and fetch quotas."""
-        cls = self.records_service.draft_cls if draft else self.records_service.record_cls
+        cls = (
+            self.records_service.draft_cls if draft else self.records_service.record_cls
+        )
 
         records = []
         parent_ids = set()
@@ -90,14 +100,16 @@ def _compute_usage(self, records, quotas):
             total_extra += extra_quota
             total_used += additional_used
 
-            results.append({
-                "item": item,
-                "record": record,
-                "quota": quota,
-                "used_bytes": used_bytes,
-                "extra_quota": extra_quota,
-                "additional_used": additional_used,
-            })
+            results.append(
+                {
+                    "item": item,
+                    "record": record,
+                    "quota": quota,
+                    "used_bytes": used_bytes,
+                    "extra_quota": extra_quota,
+                    "additional_used": additional_used,
+                }
+            )
 
         return results, total_extra, total_used
 
@@ -115,8 +127,7 @@ def get_user_storage_usage(self, user, include_drafts=True):
         draft_data, extra_d, used_d = [], 0, 0
         if include_drafts:
             draft_data, extra_d, used_d = self._process_resources(
-                self._search_user_resources(user, drafts=True),
-                draft=True
+                self._search_user_resources(user, drafts=True), draft=True
             )
 
         return {

invenio_rdm_records/views.py

@@ -15,6 +15,7 @@
 from flask import Blueprint, abort, current_app, render_template
 from flask_login import current_user, login_required
 from invenio_records_resources.services.files.transfer import constants
+
 from .proxies import current_rdm_records_storage_service
 
 blueprint = Blueprint("invenio_rdm_records_ext", __name__)
@@ -120,6 +121,7 @@ def file_transfer_type():
 
     return {"transfer_types": file_transfer_type_constants}
 
+
 def _format_storage(data):
     """Format storage data for UI."""
     BYTES_TO_GB = 1e9
@@ -129,15 +131,19 @@ def _format_storage(data):
         item = e["item"]
         record = e["record"]
 
-        rows.append({
-            "title": item.get("metadata", {}).get("title", "Empty title"),
-            "url": item["links"]["self_html"],
-            "additional_quota": round(e["extra_quota"] / BYTES_TO_GB, 1),
-            "used": round(e["used_bytes"] / BYTES_TO_GB, 1),
-            "total": round((data["default_quota"] + e["extra_quota"]) / BYTES_TO_GB, 1),
-            "date": item.get("metadata", {}).get("publication_date", ""),
-            "status": "Draft" if not record.is_published else "Published",
-        })
+        rows.append(
+            {
+                "title": item.get("metadata", {}).get("title", "Empty title"),
+                "url": item["links"]["self_html"],
+                "additional_quota": round(e["extra_quota"] / BYTES_TO_GB, 1),
+                "used": round(e["used_bytes"] / BYTES_TO_GB, 1),
+                "total": round(
+                    (data["default_quota"] + e["extra_quota"]) / BYTES_TO_GB, 1
+                ),
+                "date": item.get("metadata", {}).get("publication_date", ""),
+                "status": "Draft" if not record.is_published else "Published",
+            }
+        )
 
     return {
         "default_quota": round(data["default_quota"] / BYTES_TO_GB, 1),
@@ -150,6 +156,7 @@ def _format_storage(data):
         "records": rows,
     }
 
+
 @blueprint.route("/account/settings/quota/", endpoint="storage_settings")
 @login_required
 def storage_settings():
@@ -164,4 +171,4 @@ def storage_settings():
     return render_template(
         "invenio_rdm_records/settings/storage.html",
         storage=storage,
-    )
+    )