config: add max files count configs for files and media files

* closes zenodo/rdm-project#514
caltechlibrary · Dec 12, 2023 · be341b2 · be341b2
1 parent b5ffcdc
commit be341b2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/invenio_rdm_records/config.py b/invenio_rdm_records/config.py
@@ -535,3 +535,9 @@ def lock_edit_published_files(service, identity, record=None):
 # Feature flag to enable/disable user moderation
 RDM_USER_MODERATION_ENABLED = False
 """Flag to enable creation of user moderation requests on specific user actions."""
+
+RDM_RECORDS_MAX_FILES_COUNT = 100
+"""Max amount of files allowed to upload in the deposit form."""
+
+RDM_RECORDS_MAX_MEDIA_FILES_COUNT = 100
+"""Max amount of media files allowed to upload in the deposit form."""
diff --git a/invenio_rdm_records/services/config.py b/invenio_rdm_records/services/config.py
@@ -501,6 +501,8 @@ class RDMFileRecordServiceConfig(FileServiceConfig, ConfiguratorMixin):
         "RDM_PERMISSION_POLICY", default=RDMRecordPermissionPolicy
     )
 
+    max_files_count = FromConfig("RDM_RECORDS_MAX_FILES_COUNT", 100)
+
     file_links_list = {
         **FileServiceConfig.file_links_list,
         "archive": RecordLink(
@@ -538,6 +540,8 @@ class RDMMediaFileRecordServiceConfig(FileServiceConfig, ConfiguratorMixin):
     )
     permission_action_prefix = "media_"
 
+    max_files_count = FromConfig("RDM_RECORDS_MAX_MEDIA_FILES_COUNT", 100)
+
     file_links_list = {
         "self": RecordLink("{+api}/records/{id}/media-files"),
         "archive": RecordLink(
@@ -564,6 +568,8 @@ class RDMFileDraftServiceConfig(FileServiceConfig, ConfiguratorMixin):
         "RDM_PERMISSION_POLICY", default=RDMRecordPermissionPolicy
     )
 
+    max_files_count = FromConfig("RDM_RECORDS_MAX_FILES_COUNT", 100)
+
     file_links_list = {
         "self": RecordLink("{+api}/records/{id}/draft/files"),
         "archive": RecordLink(
@@ -603,6 +609,8 @@ class RDMMediaFileDraftServiceConfig(FileServiceConfig, ConfiguratorMixin):
         "RDM_PERMISSION_POLICY", default=RDMRecordPermissionPolicy
     )
 
+    max_files_count = FromConfig("RDM_RECORDS_MAX_MEDIA_FILES_COUNT", 100)
+
     file_links_list = {
         "self": RecordLink("{+api}/records/{id}/draft/media-files"),
         "archive": RecordLink(