import datetime from typing import Any, Dict, List, Type, TypeVar, Union import attr from dateutil.parser import isoparse from ..models.ai_feedback import AiFeedback from ..models.api_call_status import ApiCallStatus from ..models.base64data import Base64Data from ..models.file_export_format import FileExportFormat from ..models.uuid import Uuid from ..types import UNSET, Unset RB = TypeVar("RB", bound="TextToCad") @attr.s(auto_attribs=True) class TextToCad: """A response from a text to CAD prompt.""" # noqa: E501 completed_at: Union[Unset, datetime.datetime] = UNSET created_at: Union[Unset, datetime.datetime] = UNSET error: Union[Unset, str] = UNSET feedback: Union[Unset, AiFeedback] = UNSET id: Union[Unset, str] = UNSET model_version: Union[Unset, str] = UNSET output_format: Union[Unset, FileExportFormat] = UNSET outputs: Union[Unset, Dict[str, Base64Data]] = UNSET prompt: Union[Unset, str] = UNSET started_at: Union[Unset, datetime.datetime] = UNSET status: Union[Unset, ApiCallStatus] = UNSET updated_at: Union[Unset, datetime.datetime] = UNSET user_id: Union[Unset, str] = UNSET additional_properties: Dict[str, Any] = attr.ib(init=False, factory=dict) def to_dict(self) -> Dict[str, Any]: completed_at: Union[Unset, str] = UNSET if not isinstance(self.completed_at, Unset): completed_at = self.completed_at.isoformat() created_at: Union[Unset, str] = UNSET if not isinstance(self.created_at, Unset): created_at = self.created_at.isoformat() error = self.error feedback: Union[Unset, AiFeedback] = UNSET if not isinstance(self.feedback, Unset): feedback = self.feedback id = self.id model_version = self.model_version output_format: Union[Unset, FileExportFormat] = UNSET if not isinstance(self.output_format, Unset): output_format = self.output_format outputs: Union[Unset, Dict[str, str]] = UNSET if not isinstance(self.outputs, Unset): new_dict: Dict[str, str] = {} for key, value in self.outputs.items(): new_dict[key] = value.get_encoded() outputs = new_dict prompt = self.prompt started_at: Union[Unset, str] = UNSET if not isinstance(self.started_at, Unset): started_at = self.started_at.isoformat() status: Union[Unset, ApiCallStatus] = UNSET if not isinstance(self.status, Unset): status = self.status updated_at: Union[Unset, str] = UNSET if not isinstance(self.updated_at, Unset): updated_at = self.updated_at.isoformat() user_id = self.user_id field_dict: Dict[str, Any] = {} field_dict.update(self.additional_properties) field_dict.update({}) if completed_at is not UNSET: field_dict["completed_at"] = completed_at if created_at is not UNSET: field_dict["created_at"] = created_at if error is not UNSET: field_dict["error"] = error if feedback is not UNSET: field_dict["feedback"] = feedback if id is not UNSET: field_dict["id"] = id if model_version is not UNSET: field_dict["model_version"] = model_version if output_format is not UNSET: field_dict["output_format"] = output_format if outputs is not UNSET: field_dict["outputs"] = outputs if prompt is not UNSET: field_dict["prompt"] = prompt if started_at is not UNSET: field_dict["started_at"] = started_at if status is not UNSET: field_dict["status"] = status if updated_at is not UNSET: field_dict["updated_at"] = updated_at if user_id is not UNSET: field_dict["user_id"] = user_id return field_dict @classmethod def from_dict(cls: Type[RB], src_dict: Dict[str, Any]) -> RB: d = src_dict.copy() _completed_at = d.pop("completed_at", UNSET) completed_at: Union[Unset, datetime.datetime] if isinstance(_completed_at, Unset): completed_at = UNSET else: completed_at = isoparse(_completed_at) _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] if isinstance(_created_at, Unset): created_at = UNSET else: created_at = isoparse(_created_at) error = d.pop("error", UNSET) _feedback = d.pop("feedback", UNSET) feedback: Union[Unset, AiFeedback] if isinstance(_feedback, Unset): feedback = UNSET if _feedback is None: feedback = UNSET else: feedback = _feedback _id = d.pop("id", UNSET) id: Union[Unset, Uuid] if isinstance(_id, Unset): id = UNSET if _id is None: id = UNSET else: id = _id model_version = d.pop("model_version", UNSET) _output_format = d.pop("output_format", UNSET) output_format: Union[Unset, FileExportFormat] if isinstance(_output_format, Unset): output_format = UNSET if _output_format is None: output_format = UNSET else: output_format = _output_format _outputs = d.pop("outputs", UNSET) if isinstance(_outputs, Unset): outputs = UNSET else: new_map: Dict[str, Base64Data] = {} for k, v in _outputs.items(): new_map[k] = Base64Data(bytes(v, "utf-8")) outputs = new_map # type: ignore prompt = d.pop("prompt", UNSET) _started_at = d.pop("started_at", UNSET) started_at: Union[Unset, datetime.datetime] if isinstance(_started_at, Unset): started_at = UNSET else: started_at = isoparse(_started_at) _status = d.pop("status", UNSET) status: Union[Unset, ApiCallStatus] if isinstance(_status, Unset): status = UNSET if _status is None: status = UNSET else: status = _status _updated_at = d.pop("updated_at", UNSET) updated_at: Union[Unset, datetime.datetime] if isinstance(_updated_at, Unset): updated_at = UNSET else: updated_at = isoparse(_updated_at) _user_id = d.pop("user_id", UNSET) user_id: Union[Unset, Uuid] if isinstance(_user_id, Unset): user_id = UNSET if _user_id is None: user_id = UNSET else: user_id = _user_id text_to_cad = cls( completed_at=completed_at, created_at=created_at, error=error, feedback=feedback, id=id, model_version=model_version, output_format=output_format, outputs=outputs, prompt=prompt, started_at=started_at, status=status, updated_at=updated_at, user_id=user_id, ) text_to_cad.additional_properties = d return text_to_cad @property def additional_keys(self) -> List[str]: return list(self.additional_properties.keys()) def __getitem__(self, key: str) -> Any: return self.additional_properties[key] def __setitem__(self, key: str, value: Any) -> None: self.additional_properties[key] = value def __delitem__(self, key: str) -> None: del self.additional_properties[key] def __contains__(self, key: str) -> bool: return key in self.additional_properties