Add insert_many method to State model for bulk insertion with fingerprint generation (#260)

* Add insert_many method to State model for bulk insertion with fingerprint generation

- Introduced an `insert_many` class method in the State model to handle bulk state insertions while ensuring that fingerprints are generated for states that require them.
- Updated the `_generate_fingerprint` method to maintain existing functionality for individual state insertions.
- Enhanced the overall integrity of state management by ensuring consistent fingerprint generation during bulk operations.

* Refactor state.py to remove unused List import

- Removed the unused List import from the state.py file to clean up the code and improve readability. This change helps maintain a more efficient and organized codebase.

* Refactor fingerprint generation in insert_many method of State model

- Updated the insert_many method to always generate fingerprints for all states, removing the conditional check for does_unites. This change simplifies the logic and ensures consistent fingerprint generation for all state insertions.

Author: NiveditJain

state-manager/app/models/db/state.py

@@ -3,6 +3,7 @@
 from ..state_status_enum import StateStatusEnum
 from pydantic import Field
 from beanie import Insert, PydanticObjectId, Replace, Save, before_event
+from pymongo.results import InsertManyResult
 from typing import Any, Optional
 import hashlib
 import json
@@ -21,6 +22,7 @@ class State(BaseDatabaseModel):
     parents: dict[str, PydanticObjectId] = Field(default_factory=dict, description="Parents of the state")
     does_unites: bool = Field(default=False, description="Whether this state unites other states")
     state_fingerprint: str = Field(default="", description="Fingerprint of the state")
+    
     @before_event([Insert, Replace, Save])
     def _generate_fingerprint(self):
         if not self.does_unites:
@@ -42,6 +44,15 @@ def _generate_fingerprint(self):
             ensure_ascii=True,         # normalized non-ASCII escapes
         ).encode("utf-8")
         self.state_fingerprint = hashlib.sha256(payload).hexdigest()    
+    
+    @classmethod
+    async def insert_many(cls, documents: list["State"]) -> InsertManyResult:
+        """Override insert_many to ensure fingerprints are generated before insertion."""
+        # Generate fingerprints for states that need them
+        for state in documents:
+            state._generate_fingerprint()
+        
+        return await super().insert_many(documents) # type: ignore
 
     class Settings:
         indexes = [