fractal-server 2.14.4a0__py3-none-any.whl → 2.14.6__py3-none-any.whl

fractal_server/app/schemas/v2/dumps.py

@@ -7,8 +7,6 @@ These models are used in at least two situations:
 1. In the "*_dump" attributes of Job models;
 2. In the history items, to trim their size.
 """
-from typing import Optional
-
 from pydantic import BaseModel
 from pydantic import ConfigDict
 from pydantic import Field
@@ -29,10 +27,10 @@ class TaskDumpV2(BaseModel):
     name: str
     type: TaskTypeType
 
-    command_non_parallel: Optional[str] = None
-    command_parallel: Optional[str] = None
-    source: Optional[str] = None
-    version: Optional[str] = None
+    command_non_parallel: str | None = None
+    command_parallel: str | None = None
+    source: str | None = None
+    version: str | None = None
 
     input_types: dict[str, bool]
     output_types: dict[str, bool]
@@ -47,12 +45,12 @@ class WorkflowTaskDumpV2(BaseModel):
 
     id: int
     workflow_id: int
-    order: Optional[int] = None
+    order: int | None = None
 
     type_filters: dict[str, bool]
 
-    task_id: Optional[int] = None
-    task: Optional[TaskDumpV2] = None
+    task_id: int | None = None
+    task: TaskDumpV2 | None = None
 
 
 class WorkflowDumpV2(BaseModel):
@@ -81,11 +79,11 @@ class TaskGroupDumpV2(BaseModel):
     id: int
     origin: TaskGroupV2OriginEnum
     pkg_name: str
-    version: Optional[str] = None
-    python_version: Optional[str] = None
-    pip_extras: Optional[str] = None
+    version: str | None = None
+    python_version: str | None = None
+    pip_extras: str | None = None
     pinned_package_versions: dict[str, str] = Field(default_factory=dict)
 
-    path: Optional[str] = None
-    venv_path: Optional[str] = None
-    wheel_path: Optional[str] = None
+    path: str | None = None
+    venv_path: str | None = None
+    wheel_path: str | None = None

fractal_server/app/schemas/v2/history.py

@@ -1,7 +1,6 @@
 from datetime import datetime
-from enum import Enum
+from enum import StrEnum
 from typing import Any
-from typing import Optional
 
 from pydantic import AwareDatetime
 from pydantic import BaseModel
@@ -10,7 +9,7 @@ from pydantic import field_serializer
 from ....images import SingleImage
 
 
-class HistoryUnitStatus(str, Enum):
+class HistoryUnitStatus(StrEnum):
     """
     Available status for images
 
@@ -25,7 +24,7 @@ class HistoryUnitStatus(str, Enum):
     FAILED = "failed"
 
 
-class HistoryUnitStatusQuery(str, Enum):
+class HistoryUnitStatusQuery(StrEnum):
 
     SUBMITTED = "submitted"
     DONE = "done"
@@ -36,7 +35,7 @@ class HistoryUnitStatusQuery(str, Enum):
 
 class HistoryUnitRead(BaseModel):
     id: int
-    logfile: Optional[str] = None
+    logfile: str | None = None
     status: HistoryUnitStatus
     zarr_urls: list[str]
 
@@ -44,7 +43,7 @@ class HistoryUnitRead(BaseModel):
 class HistoryRunRead(BaseModel):
     id: int
     dataset_id: int
-    workflowtask_id: Optional[int] = None
+    workflowtask_id: int | None = None
     job_id: int
     workflowtask_dump: dict[str, Any]
     task_group_dump: dict[str, Any]
@@ -77,4 +76,4 @@ class ImageLogsRequest(BaseModel):
 
 
 class SingleImageWithStatus(SingleImage):
-    status: Optional[HistoryUnitStatus] = None
+    status: HistoryUnitStatus | None = None

fractal_server/app/schemas/v2/job.py

@@ -1,6 +1,5 @@
 from datetime import datetime
-from enum import Enum
-from typing import Optional
+from enum import StrEnum
 
 from pydantic import BaseModel
 from pydantic import ConfigDict
@@ -19,7 +18,7 @@ from fractal_server.types import NonEmptyStr
 from fractal_server.types import TypeFilters
 
 
-class JobStatusTypeV2(str, Enum):
+class JobStatusTypeV2(StrEnum):
     """
     Define the available job statuses
 
@@ -43,9 +42,9 @@ class JobStatusTypeV2(str, Enum):
 class JobCreateV2(BaseModel):
     model_config = ConfigDict(extra="forbid")
 
-    first_task_index: Optional[NonNegativeInt] = None
-    last_task_index: Optional[NonNegativeInt] = None
-    slurm_account: Optional[StrictStr] = None
+    first_task_index: NonNegativeInt | None = None
+    last_task_index: NonNegativeInt | None = None
+    slurm_account: StrictStr | None = None
     worker_init: NonEmptyStr = None
 
     attribute_filters: AttributeFilters = Field(default_factory=dict)
@@ -68,23 +67,23 @@ class JobCreateV2(BaseModel):
 
 class JobReadV2(BaseModel):
     id: int
-    project_id: Optional[int] = None
+    project_id: int | None = None
     project_dump: ProjectDumpV2
     user_email: str
-    slurm_account: Optional[str] = None
-    workflow_id: Optional[int] = None
+    slurm_account: str | None = None
+    workflow_id: int | None = None
     workflow_dump: WorkflowDumpV2
-    dataset_id: Optional[int] = None
+    dataset_id: int | None = None
     dataset_dump: DatasetDumpV2
     start_timestamp: AwareDatetime
-    end_timestamp: Optional[AwareDatetime] = None
+    end_timestamp: AwareDatetime | None = None
     status: str
-    log: Optional[str] = None
-    working_dir: Optional[str] = None
-    working_dir_user: Optional[str] = None
-    first_task_index: Optional[int] = None
-    last_task_index: Optional[int] = None
-    worker_init: Optional[str] = None
+    log: str | None = None
+    working_dir: str | None = None
+    working_dir_user: str | None = None
+    first_task_index: int | None = None
+    last_task_index: int | None = None
+    worker_init: str | None = None
     attribute_filters: AttributeFilters
     type_filters: dict[str, bool]
 
@@ -93,7 +92,7 @@ class JobReadV2(BaseModel):
         return v.isoformat()
 
     @field_serializer("end_timestamp")
-    def serialize_datetime_end(v: Optional[datetime]) -> Optional[str]:
+    def serialize_datetime_end(v: datetime | None) -> str | None:
         if v is None:
             return None
         else:

fractal_server/app/schemas/v2/manifest.py

@@ -1,5 +1,4 @@
 from typing import Literal
-from typing import Optional
 
 from pydantic import BaseModel
 from pydantic import Field
@@ -41,22 +40,22 @@ class TaskManifestV2(BaseModel):
     """
 
     name: str
-    executable_non_parallel: Optional[str] = None
-    executable_parallel: Optional[str] = None
+    executable_non_parallel: str | None = None
+    executable_parallel: str | None = None
     input_types: dict[str, bool] = Field(default_factory=dict)
     output_types: dict[str, bool] = Field(default_factory=dict)
     meta_non_parallel: DictStrAny = Field(default_factory=dict)
     meta_parallel: DictStrAny = Field(default_factory=dict)
-    args_schema_non_parallel: Optional[DictStrAny] = None
-    args_schema_parallel: Optional[DictStrAny] = None
-    docs_info: Optional[str] = None
-    docs_link: Optional[HttpUrlStr] = None
+    args_schema_non_parallel: DictStrAny | None = None
+    args_schema_parallel: DictStrAny | None = None
+    docs_info: str | None = None
+    docs_link: HttpUrlStr | None = None
 
-    category: Optional[str] = None
-    modality: Optional[str] = None
+    category: str | None = None
+    modality: str | None = None
     tags: list[str] = Field(default_factory=list)
 
-    type: Optional[
+    type: None | (
         Literal[
             "compound",
             "converter_compound",
@@ -64,7 +63,7 @@ class TaskManifestV2(BaseModel):
             "converter_non_parallel",
             "parallel",
         ]
-    ] = None
+    ) = None
 
     @model_validator(mode="after")
     def validate_executable_args_meta(self):
@@ -140,8 +139,8 @@ class ManifestV2(BaseModel):
     manifest_version: Literal["2"]
     task_list: list[TaskManifestV2]
     has_args_schemas: bool = False
-    args_schema_version: Optional[str] = None
-    authors: Optional[NonEmptyStr] = None
+    args_schema_version: str | None = None
+    authors: NonEmptyStr | None = None
 
     @model_validator(mode="after")
     def _check_args_schemas_are_present(self):

fractal_server/app/schemas/v2/status_legacy.py

@@ -1,10 +1,10 @@
-from enum import Enum
+from enum import StrEnum
 
 from pydantic import BaseModel
 from pydantic import Field
 
 
-class WorkflowTaskStatusTypeV2(str, Enum):
+class WorkflowTaskStatusTypeV2(StrEnum):
     """
     Define the available values for the status of a `WorkflowTask`.
 

fractal_server/app/schemas/v2/task.py

@@ -1,6 +1,5 @@
 from typing import Any
 from typing import Literal
-from typing import Optional
 
 from pydantic import BaseModel
 from pydantic import ConfigDict
@@ -35,24 +34,24 @@ class TaskCreateV2(BaseModel):
     command_non_parallel: NonEmptyStr = None
     command_parallel: NonEmptyStr = None
 
-    meta_non_parallel: Optional[DictStrAny] = None
-    meta_parallel: Optional[DictStrAny] = None
+    meta_non_parallel: DictStrAny | None = None
+    meta_parallel: DictStrAny | None = None
     version: NonEmptyStr = None
-    args_schema_non_parallel: Optional[DictStrAny] = None
-    args_schema_parallel: Optional[DictStrAny] = None
+    args_schema_non_parallel: DictStrAny | None = None
+    args_schema_parallel: DictStrAny | None = None
     args_schema_version: NonEmptyStr = None
-    docs_info: Optional[str] = None
-    docs_link: Optional[HttpUrlStr] = None
+    docs_info: str | None = None
+    docs_link: HttpUrlStr | None = None
 
     input_types: TypeFilters = Field(default={})
     output_types: TypeFilters = Field(default={})
 
-    category: Optional[NonEmptyStr] = None
-    modality: Optional[NonEmptyStr] = None
+    category: NonEmptyStr | None = None
+    modality: NonEmptyStr | None = None
     tags: ListUniqueNonEmptyString = Field(default_factory=list)
-    authors: Optional[NonEmptyStr] = None
+    authors: NonEmptyStr | None = None
 
-    type: Optional[TaskTypeType] = None
+    type: TaskTypeType | None = None
 
     @model_validator(mode="after")
     def validate_commands(self):
@@ -92,26 +91,26 @@ class TaskReadV2(BaseModel):
     id: int
     name: str
     type: TaskTypeType
-    source: Optional[str] = None
-    version: Optional[str] = None
+    source: str | None = None
+    version: str | None = None
 
-    command_non_parallel: Optional[str] = None
-    command_parallel: Optional[str] = None
+    command_non_parallel: str | None = None
+    command_parallel: str | None = None
     meta_parallel: dict[str, Any]
     meta_non_parallel: dict[str, Any]
-    args_schema_non_parallel: Optional[dict[str, Any]] = None
-    args_schema_parallel: Optional[dict[str, Any]] = None
-    args_schema_version: Optional[str] = None
-    docs_info: Optional[str] = None
-    docs_link: Optional[str] = None
+    args_schema_non_parallel: dict[str, Any] | None = None
+    args_schema_parallel: dict[str, Any] | None = None
+    args_schema_version: str | None = None
+    docs_info: str | None = None
+    docs_link: str | None = None
     input_types: dict[str, bool]
     output_types: dict[str, bool]
 
-    taskgroupv2_id: Optional[int] = None
+    taskgroupv2_id: int | None = None
 
-    category: Optional[str] = None
-    modality: Optional[str] = None
-    authors: Optional[str] = None
+    category: str | None = None
+    modality: str | None = None
+    authors: str | None = None
     tags: list[str]
 
 
@@ -123,17 +122,17 @@ class TaskUpdateV2(BaseModel):
     input_types: TypeFilters = None
     output_types: TypeFilters = None
 
-    category: Optional[NonEmptyStr] = None
-    modality: Optional[NonEmptyStr] = None
-    authors: Optional[NonEmptyStr] = None
-    tags: Optional[ListUniqueNonEmptyString] = None
+    category: NonEmptyStr | None = None
+    modality: NonEmptyStr | None = None
+    authors: NonEmptyStr | None = None
+    tags: ListUniqueNonEmptyString | None = None
 
 
 class TaskImportV2(BaseModel):
     model_config = ConfigDict(extra="forbid")
 
     pkg_name: NonEmptyStr
-    version: Optional[NonEmptyStr] = None
+    version: NonEmptyStr | None = None
     name: NonEmptyStr
 
 
@@ -143,5 +142,5 @@ class TaskImportV2Legacy(BaseModel):
 
 class TaskExportV2(BaseModel):
     pkg_name: NonEmptyStr
-    version: Optional[NonEmptyStr] = None
+    version: NonEmptyStr | None = None
     name: NonEmptyStr

fractal_server/app/schemas/v2/task_collection.py

@@ -1,5 +1,4 @@
 from typing import Literal
-from typing import Optional
 
 from pydantic import BaseModel
 from pydantic import ConfigDict
@@ -47,11 +46,11 @@ class TaskCollectPipV2(BaseModel):
     """
 
     model_config = ConfigDict(extra="forbid")
-    package: Optional[NonEmptyStr] = None
-    package_version: Optional[NonEmptyStr] = None
-    package_extras: Optional[NonEmptyStr] = None
-    python_version: Optional[Literal["3.9", "3.10", "3.11", "3.12"]] = None
-    pinned_package_versions: Optional[DictStrStr] = None
+    package: NonEmptyStr | None = None
+    package_version: NonEmptyStr | None = None
+    package_extras: NonEmptyStr | None = None
+    python_version: Literal["3.9", "3.10", "3.11", "3.12"] | None = None
+    pinned_package_versions: DictStrStr | None = None
 
     @field_validator(
         "package", "package_version", "package_extras", mode="after"
@@ -94,9 +93,9 @@ class TaskCollectCustomV2(BaseModel):
     manifest: ManifestV2
     python_interpreter: AbsolutePathStr
     label: NonEmptyStr
-    package_root: Optional[AbsolutePathStr] = None
-    package_name: Optional[NonEmptyStr] = None
-    version: Optional[NonEmptyStr] = None
+    package_root: AbsolutePathStr | None = None
+    package_name: NonEmptyStr | None = None
+    version: NonEmptyStr | None = None
 
     @field_validator("package_name", mode="after")
     @classmethod

fractal_server/app/schemas/v2/task_group.py

@@ -1,6 +1,5 @@
 from datetime import datetime
-from enum import Enum
-from typing import Optional
+from enum import StrEnum
 
 from pydantic import BaseModel
 from pydantic import ConfigDict
@@ -14,20 +13,20 @@ from fractal_server.types import DictStrStr
 from fractal_server.types import NonEmptyStr
 
 
-class TaskGroupV2OriginEnum(str, Enum):
+class TaskGroupV2OriginEnum(StrEnum):
     PYPI = "pypi"
     WHEELFILE = "wheel-file"
     OTHER = "other"
 
 
-class TaskGroupActivityStatusV2(str, Enum):
+class TaskGroupActivityStatusV2(StrEnum):
     PENDING = "pending"
     ONGOING = "ongoing"
     FAILED = "failed"
     OK = "OK"
 
 
-class TaskGroupActivityActionV2(str, Enum):
+class TaskGroupActivityActionV2(StrEnum):
     COLLECT = "collect"
     DEACTIVATE = "deactivate"
     REACTIVATE = "reactivate"
@@ -36,17 +35,17 @@ class TaskGroupActivityActionV2(str, Enum):
 class TaskGroupCreateV2(BaseModel):
     model_config = ConfigDict(extra="forbid")
     user_id: int
-    user_group_id: Optional[int] = None
+    user_group_id: int | None = None
     active: bool = True
     origin: TaskGroupV2OriginEnum
     pkg_name: str
-    version: Optional[str] = None
+    version: str | None = None
     python_version: NonEmptyStr = None
     path: AbsolutePathStr = None
     venv_path: AbsolutePathStr = None
     wheel_path: AbsolutePathStr = None
     pip_extras: NonEmptyStr = None
-    pip_freeze: Optional[str] = None
+    pip_freeze: str | None = None
     pinned_package_versions: DictStrStr = Field(default_factory=dict)
 
 
@@ -66,21 +65,21 @@ class TaskGroupReadV2(BaseModel):
     task_list: list[TaskReadV2]
 
     user_id: int
-    user_group_id: Optional[int] = None
+    user_group_id: int | None = None
 
     origin: TaskGroupV2OriginEnum
     pkg_name: str
-    version: Optional[str] = None
-    python_version: Optional[str] = None
-    path: Optional[str] = None
-    venv_path: Optional[str] = None
-    wheel_path: Optional[str] = None
-    pip_freeze: Optional[str] = None
-    pip_extras: Optional[str] = None
+    version: str | None = None
+    python_version: str | None = None
+    path: str | None = None
+    venv_path: str | None = None
+    wheel_path: str | None = None
+    pip_freeze: str | None = None
+    pip_extras: str | None = None
     pinned_package_versions: dict[str, str] = Field(default_factory=dict)
 
-    venv_size_in_kB: Optional[int] = None
-    venv_file_number: Optional[int] = None
+    venv_size_in_kB: int | None = None
+    venv_file_number: int | None = None
 
     active: bool
     timestamp_created: AwareDatetime
@@ -93,27 +92,27 @@ class TaskGroupReadV2(BaseModel):
 
 class TaskGroupUpdateV2(BaseModel):
     model_config = ConfigDict(extra="forbid")
-    user_group_id: Optional[int] = None
+    user_group_id: int | None = None
 
 
 class TaskGroupActivityV2Read(BaseModel):
     id: int
     user_id: int
-    taskgroupv2_id: Optional[int] = None
+    taskgroupv2_id: int | None = None
     timestamp_started: AwareDatetime
-    timestamp_ended: Optional[AwareDatetime] = None
+    timestamp_ended: AwareDatetime | None = None
     pkg_name: str
     version: str
     status: TaskGroupActivityStatusV2
     action: TaskGroupActivityActionV2
-    log: Optional[str] = None
+    log: str | None = None
 
     @field_serializer("timestamp_started")
     def serialize_datetime_start(v: datetime) -> str:
         return v.isoformat()
 
     @field_serializer("timestamp_ended")
-    def serialize_datetime_end(v: Optional[datetime]) -> Optional[str]:
+    def serialize_datetime_end(v: datetime | None) -> str | None:
         if v is None:
             return None
         else:

fractal_server/app/schemas/v2/workflow.py

@@ -1,5 +1,4 @@
 from datetime import datetime
-from typing import Optional
 
 from pydantic import BaseModel
 from pydantic import ConfigDict
@@ -47,7 +46,7 @@ class WorkflowUpdateV2(BaseModel):
     model_config = ConfigDict(extra="forbid")
 
     name: NonEmptyStr = None
-    reordered_workflowtask_ids: Optional[ListUniqueNonNegativeInt] = None
+    reordered_workflowtask_ids: ListUniqueNonNegativeInt | None = None
 
 
 class WorkflowImportV2(BaseModel):

fractal_server/app/schemas/v2/workflowtask.py

@@ -1,6 +1,4 @@
 from typing import Any
-from typing import Optional
-from typing import Union
 
 from pydantic import BaseModel
 from pydantic import ConfigDict
@@ -20,30 +18,30 @@ from fractal_server.types import WorkflowTaskArgument
 class WorkflowTaskCreateV2(BaseModel):
     model_config = ConfigDict(extra="forbid")
 
-    meta_non_parallel: Optional[DictStrAny] = None
-    meta_parallel: Optional[DictStrAny] = None
-    args_non_parallel: Optional[WorkflowTaskArgument] = None
-    args_parallel: Optional[WorkflowTaskArgument] = None
+    meta_non_parallel: DictStrAny | None = None
+    meta_parallel: DictStrAny | None = None
+    args_non_parallel: WorkflowTaskArgument | None = None
+    args_parallel: WorkflowTaskArgument | None = None
     type_filters: TypeFilters = Field(default_factory=dict)
 
 
 class WorkflowTaskReplaceV2(BaseModel):
     """Used by 'replace-task' endpoint"""
 
-    args_non_parallel: Optional[dict[str, Any]] = None
-    args_parallel: Optional[dict[str, Any]] = None
+    args_non_parallel: dict[str, Any] | None = None
+    args_parallel: dict[str, Any] | None = None
 
 
 class WorkflowTaskReadV2(BaseModel):
     id: int
 
     workflow_id: int
-    order: Optional[int] = None
-    meta_non_parallel: Optional[dict[str, Any]] = None
-    meta_parallel: Optional[dict[str, Any]] = None
+    order: int | None = None
+    meta_non_parallel: dict[str, Any] | None = None
+    meta_parallel: dict[str, Any] | None = None
 
-    args_non_parallel: Optional[dict[str, Any]] = None
-    args_parallel: Optional[dict[str, Any]] = None
+    args_non_parallel: dict[str, Any] | None = None
+    args_parallel: dict[str, Any] | None = None
 
     type_filters: dict[str, bool]
 
@@ -53,30 +51,30 @@ class WorkflowTaskReadV2(BaseModel):
 
 
 class WorkflowTaskReadV2WithWarning(WorkflowTaskReadV2):
-    warning: Optional[str] = None
+    warning: str | None = None
 
 
 class WorkflowTaskUpdateV2(BaseModel):
     model_config = ConfigDict(extra="forbid")
 
-    meta_non_parallel: Optional[DictStrAny] = None
-    meta_parallel: Optional[DictStrAny] = None
-    args_non_parallel: Optional[WorkflowTaskArgument] = None
-    args_parallel: Optional[WorkflowTaskArgument] = None
+    meta_non_parallel: DictStrAny | None = None
+    meta_parallel: DictStrAny | None = None
+    args_non_parallel: WorkflowTaskArgument | None = None
+    args_parallel: WorkflowTaskArgument | None = None
     type_filters: TypeFilters = None
 
 
 class WorkflowTaskImportV2(BaseModel):
     model_config = ConfigDict(extra="forbid")
 
-    meta_non_parallel: Optional[DictStrAny] = None
-    meta_parallel: Optional[DictStrAny] = None
-    args_non_parallel: Optional[DictStrAny] = None
-    args_parallel: Optional[DictStrAny] = None
-    type_filters: Optional[TypeFilters] = None
-    input_filters: Optional[dict[str, Any]] = None
+    meta_non_parallel: DictStrAny | None = None
+    meta_parallel: DictStrAny | None = None
+    args_non_parallel: DictStrAny | None = None
+    args_parallel: DictStrAny | None = None
+    type_filters: TypeFilters | None = None
+    input_filters: dict[str, Any] | None = None
 
-    task: Union[TaskImportV2, TaskImportV2Legacy]
+    task: TaskImportV2 | TaskImportV2Legacy
 
     @model_validator(mode="before")
     @classmethod
@@ -108,10 +106,10 @@ class WorkflowTaskImportV2(BaseModel):
 
 
 class WorkflowTaskExportV2(BaseModel):
-    meta_non_parallel: Optional[dict[str, Any]] = None
-    meta_parallel: Optional[dict[str, Any]] = None
-    args_non_parallel: Optional[dict[str, Any]] = None
-    args_parallel: Optional[dict[str, Any]] = None
+    meta_non_parallel: dict[str, Any] | None = None
+    meta_parallel: dict[str, Any] | None = None
+    args_non_parallel: dict[str, Any] | None = None
+    args_parallel: dict[str, Any] | None = None
     type_filters: dict[str, bool] = Field(default_factory=dict)
 
     task: TaskExportV2