models: fix record file indices

inveniosoftware · Feb 5, 2024 · c169db5 · c169db5
1 parent 2ae2597
commit c169db5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/invenio_records_resources/records/models.py b/invenio_records_resources/records/models.py
@@ -34,6 +34,7 @@ def record_id(cls):
             UUIDType,
             db.ForeignKey(cls.__record_model_cls__.id, ondelete="RESTRICT"),
             nullable=False,
+            index=True,
         )
 
     @declared_attr
@@ -60,9 +61,10 @@ def object_version(cls):
     def __table_args__(cls):
         """Table args."""
         return (
+            # To make sure we don't have duplicate keys for record files
             db.Index(
-                f"uidx_{cls.__tablename__}_id_key",
-                "id",
+                f"uidx_{cls.__tablename__}_record_id_key",
+                "record_id",
                 "key",
                 unique=True,
             ),