tests work

Author: mesemus

invenio_records_resources/services/files/results.py

@@ -11,6 +11,7 @@
 
 from ..base import ServiceListResult
 from ..records.results import RecordItem
+from ...proxies import current_transfer_registry
 
 
 class FileItem(RecordItem):
@@ -92,8 +93,22 @@ def entries(self):
                     identity=self._identity,
                 ),
             )
+
+            # create links
             if self._links_item_tpl:
-                projection["links"] = self._links_item_tpl.expand(self._identity, entry)
+                links = self._links_item_tpl.expand(self._identity, entry)
+            else:
+                links = {}
+
+            # add transfer links
+            transfer = current_transfer_registry.get_transfer(file_record=entry)
+            for k, v in transfer.expand_links(self._identity, entry).items():
+                if v is not None:
+                    links[k] = v
+                else:
+                    links.pop(k, None)
+
+            projection["links"] = links
 
             yield projection
 

invenio_records_resources/services/files/schema.py

@@ -113,6 +113,9 @@ class Meta:
 
     @post_dump(pass_many=False, pass_original=True)
     def _dump_transfer_data(self, data, original_data, **kwargs):
+        """
+        Enriches the dumped data with the transfer data.
+        """
         transfer = current_transfer_registry.get_transfer(file_record=original_data)
         data |= transfer.transfer_data
         return data

invenio_records_resources/services/files/transfer/base.py

@@ -94,6 +94,10 @@ def transfer_data(self):
             'status': self.status,
         }
 
+    def expand_links(self, identity, file_record):
+        """Expand links."""
+        return {}
+
     # @abstractmethod
     # def read_file_content(self, record, file_metadata):
     #     """Read a file content."""

invenio_records_resources/services/files/transfer/registry.py

@@ -1,14 +1,15 @@
 from typing import Dict, Type
 
-from invenio_records_resources.services.files.transfer.base import BaseTransfer
+from .base import BaseTransfer
+from .types import LOCAL_TRANSFER_TYPE
 
 
 class TransferRegistry:
     """
     A registry for transfer providers.
     """
 
-    DEFAULT_TRANSFER_TYPE = "L"
+    DEFAULT_TRANSFER_TYPE = LOCAL_TRANSFER_TYPE
     """
     Default transfer type if no storage class is provided in file upload initiation.
     """
@@ -34,7 +35,6 @@ def get_transfer(self, *, transfer_type=None, file_record=None, **kwargs):
         if transfer_type is None:
             if file_record is not None and file_record.file is not None:
                 transfer_type = file_record.file.storage_class
-            else:
-                transfer_type = self.DEFAULT_TRANSFER_TYPE
 
-        return self._transfers[transfer_type](file_record=file_record, **kwargs)
+        return self._transfers[transfer_type or self.DEFAULT_TRANSFER_TYPE](
+            file_record=file_record, **kwargs)