upgrade: solve deprecate warnings of pydantic v2

Author: duprijil

dags/pipelines/notice_batch_processor_pipelines.py

@@ -22,7 +22,7 @@ def notices_batch_distillation_pipeline(notice_ids: List[str], mongodb_client: M
     for notice_id in notice_ids:
         notice = notice_repository.get(reference=notice_id)
         notice.set_distilled_rdf_manifestation(
-            distilled_rdf_manifestation=notice.rdf_manifestation.copy())
+            distilled_rdf_manifestation=notice.rdf_manifestation.model_copy())
         notices.append(notice)
     for cet_uri in CET_URIS:
         deduplicate_entities_by_cet_uri(notices=notices, cet_uri=cet_uri)

ted_sws/alignment_oracle/services/limes_configurator.py

@@ -28,7 +28,7 @@ def generate_xml_config_from_limes_config(limes_config_params: LimesConfigParams
     :param limes_config_params:
     :return:
     """
-    return TEMPLATES.get_template(LIMES_CONFIG_TEMPLATE).render(limes_config_params.dict())
+    return TEMPLATES.get_template(LIMES_CONFIG_TEMPLATE).render(limes_config_params.model_dump())
 
 
 def generate_default_limes_config_params(source_sparql_endpoint: str,

ted_sws/core/model/manifestation.py

@@ -10,7 +10,7 @@
 from enum import Enum
 from typing import List, Union, Optional, Dict
 
-from pydantic import Field
+from pydantic import Field, ConfigDict
 
 from ted_sws.core.model import PropertyBaseModel
 from ted_sws.core.model.validation_report_data import ReportNoticeData
@@ -44,11 +44,10 @@ class Manifestation(PropertyBaseModel):
         A manifestation that embodies a FRBR Work/Expression.
     """
 
-    class Config:
-        validate_assignment = True
-        orm_mode = True
+    model_config = ConfigDict(validate_assignment=True,
+                              from_attributes=True)
 
-    object_data: str = Field(..., allow_mutation=True)
+    object_data: str = Field(..., frozen=False)
 
     def __str__(self):
         STR_LEN = 150  # constant
@@ -226,8 +225,7 @@ class SPARQLQueryResult(PropertyBaseModel):
     message: Optional[str] = None
     identifier: Optional[str] = None
 
-    class Config:
-        use_enum_values = True
+    model_config = ConfigDict(use_enum_values=True)
 
 
 class SPARQLTestSuiteValidationReport(RDFValidationManifestation):

ted_sws/core/model/metadata.py

@@ -19,8 +19,6 @@ class Metadata(PropertyBaseModel):
         Unified interface for metadata
     """
 
-    class Config:
-        underscore_attrs_are_private = True
 
 
 class XMLMetadata(Metadata):

ted_sws/core/model/notice.py

@@ -19,7 +19,7 @@
 from functools import total_ordering
 from typing import Optional, List, Union
 
-from pydantic import Field, computed_field
+from pydantic import Field, computed_field, ConfigDict
 
 from ted_sws.core.model import PropertyBaseModel
 from ted_sws.core.model.lazy_object import LazyObjectABC, LazyObjectFieldsLoaderABC
@@ -120,10 +120,8 @@ class WorkExpression(PropertyBaseModel, abc.ABC):
             See: https://www.cosmicpython.com/book/chapter_11_external_events.html
     """
 
-    class Config:
-        underscore_attrs_are_private = True
-        validate_assignment = True
-        orm_mode = True
+    model_config = ConfigDict(validate_assignment=True,
+                              from_attributes=True)
 
     created_at: str = datetime.now().replace(microsecond=0).isoformat()
     version_number: int = 0
@@ -184,7 +182,7 @@ class Notice(LazyWorkExpression):
 
     """
     _status: NoticeStatus = NoticeStatus.RAW
-    ted_id: str = Field(..., allow_mutation=False)
+    ted_id: str = Field(..., frozen=True)
     _original_metadata: Optional[TEDMetadata] = None
     _xml_manifestation: Optional[XMLManifestation] = None
     _normalised_metadata: Optional[NormalisedMetadata] = None

ted_sws/core/model/supra_notice.py

@@ -10,6 +10,8 @@
 from datetime import datetime, date
 from typing import List, Optional
 
+from pydantic import ConfigDict
+
 from ted_sws.core.model import PropertyBaseModel
 from ted_sws.core.model.manifestation import Manifestation, ValidationSummaryReport
 
@@ -19,10 +21,8 @@ class SupraNotice(PropertyBaseModel, abc.ABC):
         This is an arbitrary aggregate over a list of notices.
     """
 
-    class Config:
-        underscore_attrs_are_private = True
-        validate_assignment = True
-        orm_mode = True
+    model_config = ConfigDict(validate_assignment=True,
+                              from_attributes=True)
 
     created_at: datetime = datetime.now().replace(microsecond=0)
 

ted_sws/core/model/transform.py

@@ -11,14 +11,13 @@
 from enum import Enum
 from typing import List, Optional, Union
 
-from pydantic import field_validator
+from pydantic import field_validator, ConfigDict
 
 from ted_sws.core.model import PropertyBaseModel
 
 
 class MappingSuiteComponent(PropertyBaseModel, abc.ABC):
-    class Config:
-        validate_assignment = True
+    model_config = ConfigDict(validate_assignment=True)
 
 
 class FileResource(MappingSuiteComponent):

ted_sws/data_manager/adapters/manifestation_repository.py

@@ -59,7 +59,7 @@ def _update_manifestation(self, reference: str, manifestation: Manifestation, up
         :return:
         """
         if manifestation is not None:
-            manifestation_dict = manifestation.dict()
+            manifestation_dict = manifestation.model_dump()
             manifestation_dict[AGGREGATE_REFERENCE_ID] = reference
             manifestation_dict[MANIFESTATION_TYPE_ID] = self._manifestation_type
             reference = self._build_reference(base_reference=reference)

ted_sws/data_manager/adapters/mapping_suite_repository.py

@@ -64,7 +64,7 @@ def _create_dict_from_mapping_suite(self, mapping_suite: MappingSuite) -> dict:
         :param mapping_suite:
         :return:
         """
-        mapping_suite_dict = mapping_suite.dict()
+        mapping_suite_dict = mapping_suite.model_dump()
         mapping_suite_dict[MONGODB_COLLECTION_ID] = mapping_suite.get_mongodb_id()
         mapping_suite_dict[MS_CREATED_AT_KEY] = datetime.fromisoformat(mapping_suite_dict[MS_CREATED_AT_KEY])
         inject_date_string_fields(data=mapping_suite_dict, date_field_name=MS_CREATED_AT_KEY)
@@ -215,7 +215,7 @@ def _write_package_metadata(self, mapping_suite: MappingSuite):
         package_path = self.repository_path / mapping_suite.identifier
         package_path.mkdir(parents=True, exist_ok=True)
         metadata_path = package_path / MS_METADATA_FILE_NAME
-        package_metadata = mapping_suite.dict()
+        package_metadata = mapping_suite.model_dump()
         [package_metadata.pop(key, None) for key in
          ["transformation_rule_set", "shacl_test_suites", "sparql_test_suites"]]
         with metadata_path.open("w", encoding="utf-8") as f:

ted_sws/data_manager/adapters/metadata_repository.py

@@ -37,7 +37,7 @@ def _update_metadata(self, reference: str, metadata: Metadata, upsert: bool = Fa
         :return:
         """
         if metadata is not None:
-            metadata_dict = metadata.dict()
+            metadata_dict = metadata.model_dump()
             metadata_dict[AGGREGATE_REFERENCE_ID] = reference
             metadata_dict[METADATA_TYPE_ID] = self._metadata_type
             reference = self._build_reference(base_reference=reference)