⬆️ Upgrade libraries (pydantic v2)

ci/helpers/requirements.txt

@@ -1,17 +1,19 @@
 # This file was autogenerated by uv via the following command:
-#    uv pip compile requirements.in
+#    uv pip compile requirements.in -o requirements.txt
 aiohttp==3.9.5
+    # via
+    #   -c ../../requirements/constraints.txt
+    #   -r requirements.in
 aiosignal==1.3.1
     # via aiohttp
+annotated-types==0.7.0
+    # via pydantic
 anyio==4.3.0
     # via starlette
-async-timeout==4.0.3
-    # via aiohttp
 attrs==23.2.0
     # via aiohttp
-exceptiongroup==1.2.1
-    # via anyio
-fastapi==0.99.1
+fastapi==0.115.0
+    # via -r requirements.in
 frozenlist==1.4.1
     # via
     #   aiohttp
@@ -24,16 +26,22 @@ multidict==6.0.5
     # via
     #   aiohttp
     #   yarl
-pydantic==1.10.15
-    # via fastapi
+pydantic==2.9.2
+    # via
+    #   -c ../../requirements/constraints.txt
+    #   fastapi
+pydantic-core==2.23.4
+    # via pydantic
 sniffio==1.3.1
     # via anyio
-starlette==0.27.0
-    # via fastapi
+starlette==0.38.6
+    # via
+    #   -c ../../requirements/constraints.txt
+    #   fastapi
 typing-extensions==4.11.0
     # via
-    #   anyio
     #   fastapi
     #   pydantic
+    #   pydantic-core
 yarl==1.9.4
     # via aiohttp

packages/aws-library/requirements/_base.txt

@@ -32,6 +32,8 @@ aiormq==6.8.0
     # via aio-pika
 aiosignal==1.3.1
     # via aiohttp
+annotated-types==0.7.0
+    # via pydantic
 anyio==4.4.0
     # via
     #   fast-depends
@@ -105,7 +107,7 @@ orjson==3.10.7
     #   -r requirements/../../../packages/service-library/requirements/../../../packages/models-library/requirements/_base.in
 pamqp==3.3.0
     # via aiormq
-pydantic==1.10.17
+pydantic==2.9.1
     # via
     #   -c requirements/../../../packages/models-library/requirements/../../../requirements/constraints.txt
     #   -c requirements/../../../packages/service-library/requirements/../../../packages/models-library/requirements/../../../requirements/constraints.txt
@@ -120,6 +122,20 @@ pydantic==1.10.17
     #   -r requirements/../../../packages/settings-library/requirements/_base.in
     #   -r requirements/_base.in
     #   fast-depends
+    #   pydantic-extra-types
+    #   pydantic-settings
+pydantic-core==2.23.3
+    # via pydantic
+pydantic-extra-types==2.9.0
+    # via
+    #   -r requirements/../../../packages/models-library/requirements/_base.in
+    #   -r requirements/../../../packages/service-library/requirements/../../../packages/models-library/requirements/_base.in
+pydantic-settings==2.5.2
+    # via
+    #   -r requirements/../../../packages/models-library/requirements/_base.in
+    #   -r requirements/../../../packages/service-library/requirements/../../../packages/models-library/requirements/_base.in
+    #   -r requirements/../../../packages/service-library/requirements/../../../packages/settings-library/requirements/_base.in
+    #   -r requirements/../../../packages/settings-library/requirements/_base.in
 pygments==2.18.0
     # via rich
 pyinstrument==4.7.2
@@ -128,6 +144,8 @@ python-dateutil==2.9.0.post0
     # via
     #   arrow
     #   botocore
+python-dotenv==1.0.1
+    # via pydantic-settings
 pyyaml==6.0.2
     # via
     #   -c requirements/../../../packages/models-library/requirements/../../../requirements/constraints.txt
@@ -198,6 +216,7 @@ typing-extensions==4.12.2
     #   aiodebug
     #   faststream
     #   pydantic
+    #   pydantic-core
     #   typer
     #   types-aiobotocore
     #   types-aiobotocore-ec2

packages/aws-library/requirements/_test.txt

@@ -1,3 +1,7 @@
+annotated-types==0.7.0
+    # via
+    #   -c requirements/_base.txt
+    #   pydantic
 antlr4-python3-runtime==4.13.2
     # via moto
 appdirs==1.4.4
@@ -151,11 +155,15 @@ py-partiql-parser==0.5.5
     # via moto
 pycparser==2.22
     # via cffi
-pydantic==1.10.17
+pydantic==2.9.1
     # via
     #   -c requirements/../../../requirements/constraints.txt
     #   -c requirements/_base.txt
     #   aws-sam-translator
+pydantic-core==2.23.3
+    # via
+    #   -c requirements/_base.txt
+    #   pydantic
 pyparsing==3.1.2
     # via moto
 pytest==8.3.2
@@ -193,7 +201,9 @@ python-dateutil==2.9.0.post0
     #   faker
     #   moto
 python-dotenv==1.0.1
-    # via -r requirements/_test.in
+    # via
+    #   -c requirements/_base.txt
+    #   -r requirements/_test.in
 pyyaml==6.0.2
     # via
     #   -c requirements/../../../requirements/constraints.txt
@@ -268,6 +278,7 @@ typing-extensions==4.12.2
     #   flexparser
     #   pint
     #   pydantic
+    #   pydantic-core
     #   types-aioboto3
     #   types-aiobotocore
 urllib3==2.2.2

packages/aws-library/requirements/ci.txt

@@ -11,8 +11,9 @@
 --requirement _test.txt
 
 # installs this repo's packages
+simcore-common-library @ ../common-library
+simcore-models-library @ ../models-library/
 pytest-simcore @ ../pytest-simcore
-simcore-models-library @ ../models-library
 simcore-service-library @ ../service-library/
 simcore-settings-library @ ../settings-library/
 

packages/aws-library/requirements/dev.txt

@@ -12,8 +12,9 @@
 --requirement _tools.txt
 
 # installs this repo's packages
---editable ../pytest-simcore
+--editable ../common-library/
 --editable ../models-library/
+--editable ../pytest-simcore/
 --editable ../service-library/
 --editable ../settings-library/
 

packages/aws-library/src/aws_library/ec2/__init__.py

@@ -1,6 +1,10 @@
 from ._client import SimcoreEC2API
 from ._errors import EC2AccessError, EC2NotConnectedError, EC2RuntimeError
 from ._models import (
+    AWS_TAG_KEY_MAX_LENGTH,
+    AWS_TAG_KEY_MIN_LENGTH,
+    AWS_TAG_VALUE_MAX_LENGTH,
+    AWS_TAG_VALUE_MIN_LENGTH,
     AWSTagKey,
     AWSTagValue,
     EC2InstanceBootSpecific,
@@ -14,6 +18,10 @@
 __all__: tuple[str, ...] = (
     "AWSTagKey",
     "AWSTagValue",
+    "AWS_TAG_KEY_MIN_LENGTH",
+    "AWS_TAG_KEY_MAX_LENGTH",
+    "AWS_TAG_VALUE_MIN_LENGTH",
+    "AWS_TAG_VALUE_MAX_LENGTH",
     "EC2AccessError",
     "EC2InstanceBootSpecific",
     "EC2InstanceConfig",

packages/aws-library/src/aws_library/ec2/_errors.py

@@ -1,12 +1,9 @@
 # pylint: disable=too-many-ancestors
-from typing import Any
-
-from models_library.errors_classes import OsparcErrorMixin
+from common_library.errors_classes import OsparcErrorMixin
 
 
 class EC2BaseError(OsparcErrorMixin, Exception):
-    def __init__(self, **ctx: Any) -> None:
-        super().__init__(**ctx)
+    pass
 
 
 class EC2RuntimeError(EC2BaseError, RuntimeError):

packages/aws-library/src/aws_library/ec2/_models.py

@@ -2,18 +2,19 @@
 import re
 import tempfile
 from dataclasses import dataclass
-from typing import Any, ClassVar, TypeAlias
+from typing import Annotated, Final, TypeAlias
 
 import sh  # type: ignore[import-untyped]
 from models_library.docker import DockerGenericTag
 from pydantic import (
     BaseModel,
     ByteSize,
-    ConstrainedStr,
+    ConfigDict,
     Field,
     NonNegativeFloat,
     NonNegativeInt,
-    validator,
+    StringConstraints,
+    field_validator,
 )
 from types_aiobotocore_ec2.literals import InstanceStateNameType, InstanceTypeType
 
@@ -33,26 +34,26 @@ def __gt__(self, other: "Resources") -> bool:
         return self.cpus > other.cpus or self.ram > other.ram
 
     def __add__(self, other: "Resources") -> "Resources":
-        return Resources.construct(
+        return Resources.model_construct(
             **{
                 key: a + b
                 for (key, a), b in zip(
-                    self.dict().items(), other.dict().values(), strict=True
+                    self.model_dump().items(), other.model_dump().values(), strict=True
                 )
             }
         )
 
     def __sub__(self, other: "Resources") -> "Resources":
-        return Resources.construct(
+        return Resources.model_construct(
             **{
                 key: a - b
                 for (key, a), b in zip(
-                    self.dict().items(), other.dict().values(), strict=True
+                    self.model_dump().items(), other.model_dump().values(), strict=True
                 )
             }
         )
 
-    @validator("cpus", pre=True)
+    @field_validator("cpus", mode="before")
     @classmethod
     def _floor_cpus_to_0(cls, v: float) -> float:
         return max(v, 0)
@@ -67,19 +68,31 @@ class EC2InstanceType:
 InstancePrivateDNSName: TypeAlias = str
 
 
-class AWSTagKey(ConstrainedStr):
+AWS_TAG_KEY_MIN_LENGTH: Final[int] = 1
+AWS_TAG_KEY_MAX_LENGTH: Final[int] = 128
+AWSTagKey: TypeAlias = Annotated[
     # see [https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/Using_Tags.html#tag-restrictions]
-    regex = re.compile(r"^(?!(_index|\.{1,2})$)[a-zA-Z0-9\+\-=\._:@]+$")
-    min_length = 1
-    max_length = 128
-
-
-class AWSTagValue(ConstrainedStr):
+    str,
+    StringConstraints(
+        min_length=AWS_TAG_KEY_MIN_LENGTH,
+        max_length=AWS_TAG_KEY_MAX_LENGTH,
+        pattern=re.compile(r"^(?!(_index|\.{1,2})$)[a-zA-Z0-9\+\-=\._:@]+$"),
+    ),
+]
+
+
+AWS_TAG_VALUE_MIN_LENGTH: Final[int] = 0
+AWS_TAG_VALUE_MAX_LENGTH: Final[int] = 256
+AWSTagValue: TypeAlias = Annotated[
     # see [https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/Using_Tags.html#tag-restrictions]
     # quotes []{} were added as it allows to json encode. it seems to be accepted as a value
-    regex = re.compile(r"^[a-zA-Z0-9\s\+\-=\.,_:/@\"\'\[\]\{\}]*$")
-    min_length = 0
-    max_length = 256
+    str,
+    StringConstraints(
+        min_length=0,
+        max_length=256,
+        pattern=r"^[a-zA-Z0-9\s\+\-=\.,_:/@\"\'\[\]\{\}]*$",
+    ),
+]
 
 
 EC2Tags: TypeAlias = dict[AWSTagKey, AWSTagValue]
@@ -148,8 +161,23 @@ class EC2InstanceBootSpecific(BaseModel):
         default=0, description="number of buffer EC2s to keep (defaults to 0)"
     )
 
-    class Config:
-        schema_extra: ClassVar[dict[str, Any]] = {
+    @field_validator("custom_boot_scripts")
+    @classmethod
+    def validate_bash_calls(cls, v):
+        try:
+            with tempfile.NamedTemporaryFile(mode="wt", delete=True) as temp_file:
+                temp_file.writelines(v)
+                temp_file.flush()
+                # NOTE: this will not capture runtime errors, but at least some syntax errors such as invalid quotes
+                sh.bash("-n", temp_file.name)
+        except sh.ErrorReturnCode as exc:
+            msg = f"Invalid bash call in custom_boot_scripts: {v}, Error: {exc.stderr}"
+            raise ValueError(msg) from exc
+
+        return v
+
+    model_config = ConfigDict(
+        json_schema_extra={
             "examples": [
                 {
                     # just AMI
@@ -205,18 +233,4 @@ class Config:
                 },
             ]
         }
-
-    @validator("custom_boot_scripts")
-    @classmethod
-    def validate_bash_calls(cls, v):
-        try:
-            with tempfile.NamedTemporaryFile(mode="wt", delete=True) as temp_file:
-                temp_file.writelines(v)
-                temp_file.flush()
-                # NOTE: this will not capture runtime errors, but at least some syntax errors such as invalid quotes
-                sh.bash("-n", temp_file.name)
-        except sh.ErrorReturnCode as exc:
-            msg = f"Invalid bash call in custom_boot_scripts: {v}, Error: {exc.stderr}"
-            raise ValueError(msg) from exc
-
-        return v
+    )