# coding: utf-8

"""
    Jobs API 2.1

    The Jobs API allows you to create, edit, and delete jobs. You should never hard code secrets or store them in plain text. Use the [Secrets API](https://docs.microsoft.com/azure/databricks/dev-tools/api/latest/secrets) to manage secrets in the [Databricks CLI](https://docs.microsoft.com/azure/databricks/dev-tools/cli/index). Use the [Secrets utility](https://docs.microsoft.com/azure/databricks/dev-tools/databricks-utils#dbutils-secrets) to reference secrets in notebooks and jobs.  # noqa: E501

    The version of the OpenAPI document: 2.1
    Generated by: https://openapi-generator.tech
"""


from __future__ import annotations

import json
import pprint
import re  # noqa: F401
from inspect import getfullargspec
from typing import List, Optional

from pydantic import BaseModel, Field, StrictInt, StrictStr, validator

from databricks_jobs.models.run_state import RunState


class RepairHistoryItem(BaseModel):
    """NOTE: This class is auto generated by OpenAPI Generator.
    Ref: https://openapi-generator.tech

    Do not edit the class manually.
    """

    type: Optional[StrictStr] = Field(
        None,
        description="The repair history item type. Indicates whether a run is the original run or a repair run.",
    )
    start_time: Optional[StrictInt] = Field(
        None, description="The start time of the (repaired) run."
    )
    end_time: Optional[StrictInt] = Field(
        None, description="The end time of the (repaired) run."
    )
    state: Optional[RunState] = None
    id: Optional[StrictInt] = Field(
        None,
        description="The ID of the repair. Only returned for the items that represent a repair in `repair_history`.",
    )
    task_run_ids: Optional[List[StrictInt]] = Field(
        None,
        description="The run IDs of the task runs that ran as part of this repair history item.",
    )
    __properties = ["type", "start_time", "end_time", "state", "id", "task_run_ids"]

    @validator("type")
    def type_validate_enum(cls, v):
        if v is None:
            return v

        if v not in ("ORIGINAL", "REPAIR"):
            raise ValueError("must validate the enum values ('ORIGINAL', 'REPAIR')")
        return v

    class Config:
        allow_population_by_field_name = True
        validate_assignment = True

    def to_str(self) -> str:
        """Returns the string representation of the model using alias"""
        return pprint.pformat(self.dict(by_alias=True))

    def to_json(self) -> str:
        """Returns the JSON representation of the model using alias"""
        return json.dumps(self.to_dict())

    @classmethod
    def from_json(cls, json_str: str) -> RepairHistoryItem:
        """Create an instance of RepairHistoryItem from a JSON string"""
        return cls.from_dict(json.loads(json_str))

    def to_dict(self):
        """Returns the dictionary representation of the model using alias"""
        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
        # override the default output from pydantic by calling `to_dict()` of state
        if self.state:
            _dict["state"] = self.state.to_dict()
        return _dict

    @classmethod
    def from_dict(cls, obj: dict) -> RepairHistoryItem:
        """Create an instance of RepairHistoryItem from a dict"""
        if obj is None:
            return None

        if type(obj) is not dict:
            return RepairHistoryItem.parse_obj(obj)

        _obj = RepairHistoryItem.parse_obj(
            {
                "type": obj.get("type"),
                "start_time": obj.get("start_time"),
                "end_time": obj.get("end_time"),
                "state": RunState.from_dict(obj.get("state"))
                if obj.get("state") is not None
                else None,
                "id": obj.get("id"),
                "task_run_ids": obj.get("task_run_ids"),
            }
        )
        return _obj
