updated indexes

Author: Dragos0000

ted_sws/data_manager/adapters/mapping_suite_repository.py

@@ -39,7 +39,6 @@ def __init__(self, mongodb_client: MongoClient, database_name: str = _database_n
         self._database_name = database_name
         notice_db = mongodb_client[self._database_name]
         self.collection = notice_db[self._collection_name]
-        self.collection.create_index([("identifier", ASCENDING)])
 
     def add(self, mapping_suite: MappingSuite):
         """
@@ -67,7 +66,7 @@ def get(self, reference) -> MappingSuite:
         :param reference:
         :return: MappingSuite
         """
-        result_dict = self.collection.find_one({"identifier": reference})
+        result_dict = self.collection.find_one({"_id": reference})
         return MappingSuite(**result_dict) if result_dict else None
 
     def list(self) -> Iterator[MappingSuite]:

ted_sws/data_manager/adapters/notice_repository.py

@@ -28,7 +28,8 @@ def __init__(self, mongodb_client: MongoClient, database_name: str = _database_n
         notice_db = mongodb_client[self._database_name]
         self.file_storage = gridfs.GridFS(notice_db)
         self.collection = notice_db[self._collection_name]
-        self.collection.create_index([("ted_id", ASCENDING)])
+        self.file_storage_collection = notice_db["fs.files"]
+        self.file_storage_collection.create_index([("notice_id", ASCENDING)])
 
     def get_file_content_from_grid_fs(self, file_id: str) -> str:
         """
@@ -177,7 +178,7 @@ def get(self, reference) -> Optional[Notice]:
         :param reference:
         :return: Notice
         """
-        result_dict = self.collection.find_one({"ted_id": reference})
+        result_dict = self.collection.find_one({"_id": reference})
         if result_dict is not None:
             notice = NoticeRepository._create_notice_from_repository_result(result_dict)
             notice = self.load_notice_fields_from_grid_fs(notice)