Source code for annofabapi.pydantic_models.task_generate_rule

"""


No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)

The version of the OpenAPI document: 1.0.0
Generated by OpenAPI Generator (https://openapi-generator.tech)

Do not edit the class manually.
"""

from __future__ import annotations

import json
import pprint
from typing import Any, Dict, Set, Union

from pydantic import BaseModel, ConfigDict, ValidationError, field_validator
from typing_extensions import Self

from annofabapi.pydantic_models.task_generate_rule_by_count import TaskGenerateRuleByCount
from annofabapi.pydantic_models.task_generate_rule_by_directory import TaskGenerateRuleByDirectory
from annofabapi.pydantic_models.task_generate_rule_by_input_data_csv import TaskGenerateRuleByInputDataCsv

TASKGENERATERULE_ONE_OF_SCHEMAS = ["TaskGenerateRuleByCount", "TaskGenerateRuleByDirectory", "TaskGenerateRuleByInputDataCsv"]


[docs] class TaskGenerateRule(BaseModel): """ タスク生成のルール * `TaskGenerateRuleByCount`: 1つのタスクに割り当てる入力データの個数を指定してタスクを生成します。 * `TaskGenerateRuleByDirectory`: 入力データ名をファイルパスに見立てて、ディレクトリ単位でタスクを生成します。 * `TaskGenerateRuleByInputDataCsv`: 各タスクへの入力データへの割り当てを記入したCSVをアップロードした、一時データ保存先パスを指定してタスクを生成します。 """ # data type: TaskGenerateRuleByCount oneof_schema_1_validator: TaskGenerateRuleByCount | None = None # data type: TaskGenerateRuleByDirectory oneof_schema_2_validator: TaskGenerateRuleByDirectory | None = None # data type: TaskGenerateRuleByInputDataCsv oneof_schema_3_validator: TaskGenerateRuleByInputDataCsv | None = None actual_instance: Union[TaskGenerateRuleByCount, TaskGenerateRuleByDirectory, TaskGenerateRuleByInputDataCsv] | None = None one_of_schemas: Set[str] = {"TaskGenerateRuleByCount", "TaskGenerateRuleByDirectory", "TaskGenerateRuleByInputDataCsv"} model_config = ConfigDict( validate_assignment=True, protected_namespaces=(), ) discriminator_value_class_map: Dict[str, str] = {} def __init__(self, *args, **kwargs) -> None: if args: if len(args) > 1: raise ValueError("If a position argument is used, only 1 is allowed to set `actual_instance`") if kwargs: raise ValueError("If a position argument is used, keyword arguments cannot be used.") super().__init__(actual_instance=args[0]) else: super().__init__(**kwargs)
[docs] @field_validator("actual_instance") def actual_instance_must_validate_oneof(cls, v): instance = TaskGenerateRule.model_construct() error_messages = [] match = 0 # validate data type: TaskGenerateRuleByCount if not isinstance(v, TaskGenerateRuleByCount): error_messages.append(f"Error! Input type `{type(v)}` is not `TaskGenerateRuleByCount`") else: match += 1 # validate data type: TaskGenerateRuleByDirectory if not isinstance(v, TaskGenerateRuleByDirectory): error_messages.append(f"Error! Input type `{type(v)}` is not `TaskGenerateRuleByDirectory`") else: match += 1 # validate data type: TaskGenerateRuleByInputDataCsv if not isinstance(v, TaskGenerateRuleByInputDataCsv): error_messages.append(f"Error! Input type `{type(v)}` is not `TaskGenerateRuleByInputDataCsv`") else: match += 1 if match > 1: # more than 1 match raise ValueError( "Multiple matches found when setting `actual_instance` in TaskGenerateRule with oneOf schemas: TaskGenerateRuleByCount, TaskGenerateRuleByDirectory, TaskGenerateRuleByInputDataCsv. Details: " + ", ".join(error_messages) ) elif match == 0: # no match raise ValueError( "No match found when setting `actual_instance` in TaskGenerateRule with oneOf schemas: TaskGenerateRuleByCount, TaskGenerateRuleByDirectory, TaskGenerateRuleByInputDataCsv. Details: " + ", ".join(error_messages) ) else: return v
[docs] @classmethod def from_dict(cls, obj: Union[str, Dict[str, Any]]) -> Self: return cls.from_json(json.dumps(obj))
[docs] @classmethod def from_json(cls, json_str: str) -> Self: """Returns the object represented by the json string""" instance = cls.model_construct() error_messages = [] match = 0 # deserialize data into TaskGenerateRuleByCount try: instance.actual_instance = TaskGenerateRuleByCount.from_json(json_str) match += 1 except (ValidationError, ValueError) as e: error_messages.append(str(e)) # deserialize data into TaskGenerateRuleByDirectory try: instance.actual_instance = TaskGenerateRuleByDirectory.from_json(json_str) match += 1 except (ValidationError, ValueError) as e: error_messages.append(str(e)) # deserialize data into TaskGenerateRuleByInputDataCsv try: instance.actual_instance = TaskGenerateRuleByInputDataCsv.from_json(json_str) match += 1 except (ValidationError, ValueError) as e: error_messages.append(str(e)) if match > 1: # more than 1 match raise ValueError( "Multiple matches found when deserializing the JSON string into TaskGenerateRule with oneOf schemas: TaskGenerateRuleByCount, TaskGenerateRuleByDirectory, TaskGenerateRuleByInputDataCsv. Details: " + ", ".join(error_messages) ) elif match == 0: # no match raise ValueError( "No match found when deserializing the JSON string into TaskGenerateRule with oneOf schemas: TaskGenerateRuleByCount, TaskGenerateRuleByDirectory, TaskGenerateRuleByInputDataCsv. Details: " + ", ".join(error_messages) ) else: return instance
[docs] def to_json(self) -> str: """Returns the JSON representation of the actual instance""" if self.actual_instance is None: return "null" if hasattr(self.actual_instance, "to_json") and callable(self.actual_instance.to_json): return self.actual_instance.to_json() else: return json.dumps(self.actual_instance)
[docs] def to_dict(self) -> Union[Dict[str, Any], TaskGenerateRuleByCount, TaskGenerateRuleByDirectory, TaskGenerateRuleByInputDataCsv] | None: """Returns the dict representation of the actual instance""" if self.actual_instance is None: return None if hasattr(self.actual_instance, "to_dict") and callable(self.actual_instance.to_dict): return self.actual_instance.to_dict() else: # primitive type return self.actual_instance
[docs] def to_str(self) -> str: """Returns the string representation of the actual instance""" return pprint.pformat(self.model_dump())