Handling mongo indexes in firm

api/rpk-api/firm/contract/models.py

@@ -7,7 +7,8 @@ from beanie import PydanticObjectId
 from pydantic import BaseModel, Field, ConfigDict
 from pydantic.json_schema import SkipJsonSchema
 
-from firm.core.models import CrudDocument, RichtextSingleline, RichtextMultiline, DictionaryEntry, ForeignKey
+from firm.core.models import CrudDocument, RichtextSingleline, RichtextMultiline, DictionaryEntry, ForeignKey, \
+    CrudDocumentConfig
 from firm.core.filter import Filter, FilterSchema
 from firm.entity.models import Entity
 
@@ -98,15 +99,6 @@ class ContractDraft(CrudDocument):
     status: ContractDraftStatus = Field(default=ContractDraftStatus.in_progress, title="Statut")
     todo: List[str] = Field(default=[], title="Reste à faire")
 
-    class Settings(CrudDocument.Settings):
-        fulltext_search = ['name', 'title']
-
-        bson_encoders = {
-            datetime.date: lambda dt: dt if hasattr(dt, 'hour')
-            else datetime.datetime(year=dt.year, month=dt.month, day=dt.day,
-                                   hour=0, minute=0, second=0)
-        }
-
     async def check_is_ready(self, db):
         if self.status == ContractDraftStatus.published:
             return
@@ -150,6 +142,11 @@ class Contract(CrudDocument):
     Contrat publié. Les contrats ne peuvent pas être modifiés.
     Ils peuvent seulement être signés par les parties et imprimés par l'avocat
     """
+    model_config = ConfigDict(title='Contrat')
+    document_config = CrudDocumentConfig(
+        indexes=["parties.signature_uuid"],
+    )
+
     name: str = Field(title="Nom")
     title: str = Field(title="Titre")
     parties: List[Party] = Field(title="Parties")
@@ -170,15 +167,6 @@ class Contract(CrudDocument):
         contract_label = f"{contract_label} - {self.date.strftime('%m/%d/%Y')}"
         return contract_label
 
-    class Settings(CrudDocument.Settings):
-        fulltext_search = ['name', 'title']
-
-        bson_encoders = {
-            datetime.date: lambda dt: dt if hasattr(dt, 'hour')
-            else datetime.datetime(year=dt.year, month=dt.month, day=dt.day,
-                                   hour=0, minute=0, second=0)
-        }
-
     @classmethod
     async def find_by_signature_id(cls, db, signature_id: UUID):
         request = {'parties': {"$elemMatch": {"signature_uuid": str(signature_id) }}}