oarepo-runtime 1.5.35__py3-none-any.whl → 1.5.37__py3-none-any.whl

oarepo_runtime/services/files/components.py

@@ -0,0 +1,55 @@
+import mimetypes
+import os
+from invenio_records_resources.services.files.components import FileServiceComponent
+from marshmallow.exceptions import ValidationError
+
+class AllowedFileTypesComponent(FileServiceComponent):
+
+    def guess_content_type(self, filename: str | None) -> str | None:
+        if filename:
+            return mimetypes.guess_type(filename)[0] or "application/octet-stream"
+        return None
+
+    @property
+    def allowed_mimetypes(self):
+        """Returns files attribute (field) key."""
+        return getattr(self.service.config, "allowed_mimetypes", [])
+
+    def guess_extension(self, file, mimetype):
+        """File extension."""
+        # The ``ext`` property is used to in search to aggregate file types, and we want e.g. both ``.jpeg`` and
+        # ``.jpg`` to be aggregated under ``.jpg``
+        ext_guessed = mimetypes.guess_extension(mimetype)
+
+        # Check if a valid extension is guessed and it's not the default mimetype
+        if (
+                ext_guessed is not None
+                and mimetype != "application/octet-stream"
+        ):
+            return ext_guessed[1:]
+
+        # Support non-standard file extensions that cannot be guessed
+        _, ext = os.path.splitext(file)
+        if ext and len(ext) <= 5:
+            return ext[1:].lower()
+
+        if ext_guessed:
+            return ext_guessed[1:]
+
+    @property
+    def allowed_extensions(self):
+        """Returns files attribute (field) key."""
+        return getattr(self.service.config, "allowed_extensions", [])
+
+    def init_files(self, identity, id_, data, uow=None):
+        """Initialize the file upload for the record."""
+        list_files = list(data.files.entries)
+
+        for file in list_files:
+            allowed_type = self.guess_content_type(file)
+            allowed_ext = self.guess_extension(file, allowed_type)
+            if len(self.allowed_mimetypes) > 0 and allowed_type not in self.allowed_mimetypes:
+                raise ValidationError(f"Mimetype not supported, supported mimetypes: {self.allowed_mimetypes}")
+            elif len(self.allowed_extensions) > 0 and allowed_ext not in self.allowed_extensions:
+                raise ValidationError(f"Extension not supported, supported extensions: {self.allowed_extensions}")
+

oarepo_runtime/services/schema/ui.py

@@ -130,3 +130,4 @@ class InvenioUISchema(ma.Schema):
     updated = LocalizedDateTime(dump_only=True)
     links = ma.fields.Raw(dump_only=True)
     revision_id = ma.fields.Integer(dump_only=True)
+    expanded = ma.fields.Raw(dump_only=True)

{oarepo_runtime-1.5.35.dist-info → oarepo_runtime-1.5.37.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oarepo-runtime
-Version: 1.5.35
+Version: 1.5.37
 Summary: A set of runtime extensions of Invenio repository
 Description-Content-Type: text/markdown
 License-File: LICENSE

{oarepo_runtime-1.5.35.dist-info → oarepo_runtime-1.5.37.dist-info}/RECORD

@@ -89,6 +89,7 @@ oarepo_runtime/services/facets/max_facet.py,sha256=TZ4KMKKVJHzyU1KgNne4V7IMQPu1A
 oarepo_runtime/services/facets/nested_facet.py,sha256=y0xgjx37HsSj2xW7URxNemYTksD8hpPs7kOEfIBw22k,971
 oarepo_runtime/services/facets/params.py,sha256=5z8I4wKdp-7RJK4x6cSKE92fDJI8OkUz5JR5UKWAeYg,3564
 oarepo_runtime/services/files/__init__.py,sha256=K8MStrEQf_BUhvzhwPTF93Hkhwrd1dtv35LDo7iZeTM,268
+oarepo_runtime/services/files/components.py,sha256=yQpZ0rCaIUkECPZMZ4aH2m1NIRlEWPgiPPV_6yUoPbY,2272
 oarepo_runtime/services/files/service.py,sha256=8DH0Pefr9kilM2JnOb-UYsnqerE8Z1Mu4p6DOJ4j_ZU,608
 oarepo_runtime/services/relations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 oarepo_runtime/services/relations/components.py,sha256=3g0VdnGUM-2yYt50fPi-OADReBGJb4h05vmYHfh-QFs,592
@@ -102,7 +103,7 @@ oarepo_runtime/services/schema/i18n_validation.py,sha256=fyMTi2Rw-KiHv7c7HN61zGx
 oarepo_runtime/services/schema/marshmallow.py,sha256=LmcSxvbZ9jIhkNHCqqxt1SA2UNijoDmIzqli1MkoTrE,1153
 oarepo_runtime/services/schema/oneofschema.py,sha256=X_pXzrkYcLGGAtGN1qltrz45OzD_atrJHkHkp3L01xg,6660
 oarepo_runtime/services/schema/polymorphic.py,sha256=f9yC7MGVynAFGM0fXIq0NbqGJxI65Xpi8GaM2ZM4c6Q,561
-oarepo_runtime/services/schema/ui.py,sha256=qDR-nukJmeLwDZCdqcBg3CbseQ_A_BdxXyt_R4Qc_2c,3732
+oarepo_runtime/services/schema/ui.py,sha256=xQgW-zLyZoHldGw47uVtXQj-5LexVNKTholyq4MiBZo,3777
 oarepo_runtime/services/schema/validation.py,sha256=fahqKGDdIYWux5ZeoljrEe8VD2fDZR9VpfvYmTYAmpw,1050
 oarepo_runtime/translations/default_translations.py,sha256=060GBlA1ghWxfeumo6NqxCCZDb-6OezOuF6pr-_GEOQ,104
 oarepo_runtime/translations/jinjax_messages.jinja,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -115,9 +116,9 @@ oarepo_runtime/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 oarepo_runtime/utils/functools.py,sha256=gKS9YZtlIYcDvdNA9cmYO00yjiXBYV1jg8VpcRUyQyg,1324
 oarepo_runtime/utils/path.py,sha256=V1NVyk3m12_YLbj7QHYvUpE1wScO78bYsX1LOLeXDkI,3108
 tests/pkg_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-oarepo_runtime-1.5.35.dist-info/LICENSE,sha256=h2uWz0OaB3EN-J1ImdGJZzc7yvfQjvHVYdUhQ-H7ypY,1064
-oarepo_runtime-1.5.35.dist-info/METADATA,sha256=ov1D4zGmQwPC2LNFD5obBpcphrCt9GsW3dfByS_XNYc,4680
-oarepo_runtime-1.5.35.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-oarepo_runtime-1.5.35.dist-info/entry_points.txt,sha256=QrlXAKuPDVBinaSh_v3yO9_Nb9ZNmJCJ0VFcCW-z0Jg,327
-oarepo_runtime-1.5.35.dist-info/top_level.txt,sha256=bHhlkT1_RQC4IkfTQCqA3iN4KCB6cSFQlsXpQMSP-bE,21
-oarepo_runtime-1.5.35.dist-info/RECORD,,
+oarepo_runtime-1.5.37.dist-info/LICENSE,sha256=h2uWz0OaB3EN-J1ImdGJZzc7yvfQjvHVYdUhQ-H7ypY,1064
+oarepo_runtime-1.5.37.dist-info/METADATA,sha256=9DcMboRlkhaXmUfLa6t8nUSTjoCJS26u7e2OjxXb7jI,4680
+oarepo_runtime-1.5.37.dist-info/WHEEL,sha256=mguMlWGMX-VHnMpKOjjQidIo1ssRlCFu4a4mBpz1s2M,91
+oarepo_runtime-1.5.37.dist-info/entry_points.txt,sha256=QrlXAKuPDVBinaSh_v3yO9_Nb9ZNmJCJ0VFcCW-z0Jg,327
+oarepo_runtime-1.5.37.dist-info/top_level.txt,sha256=bHhlkT1_RQC4IkfTQCqA3iN4KCB6cSFQlsXpQMSP-bE,21
+oarepo_runtime-1.5.37.dist-info/RECORD,,

{oarepo_runtime-1.5.35.dist-info → oarepo_runtime-1.5.37.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (70.1.1)
 Root-Is-Purelib: true
 Tag: py3-none-any