wds-client 0.6.0__py3-none-any.whl → 0.8.0__py3-none-any.whl

wds_client/models/job_v1.py

@@ -3,292 +3,112 @@
 """
     Workspace Data Service
 
-    This page lists current APIs. As of v0.2, all APIs are subject to change without notice.   # noqa: E501
+    This page lists current APIs.  All v0.2 APIs are subject to change without notice.  Changelog at [https://github.com/DataBiosphere/terra-workspace-data-service/releases](https://github.com/DataBiosphere/terra-workspace-data-service/releases) 
 
     The version of the OpenAPI document: v0.2
-    Generated by: https://openapi-generator.tech
-"""
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
 
 
+from __future__ import annotations
 import pprint
 import re  # noqa: F401
+import json
 
-import six
-
-from wds_client.configuration import Configuration
-
-
-class JobV1(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
-    """
+from datetime import datetime
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
 
+class JobV1(BaseModel):
     """
-    Attributes:
-      openapi_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
-    """
-    openapi_types = {
-        'job_id': 'str',
-        'job_type': 'str',
-        'instance_id': 'str',
-        'status': 'str',
-        'created': 'datetime',
-        'updated': 'datetime',
-        'error_message': 'str'
-    }
-
-    attribute_map = {
-        'job_id': 'jobId',
-        'job_type': 'jobType',
-        'instance_id': 'instanceId',
-        'status': 'status',
-        'created': 'created',
-        'updated': 'updated',
-        'error_message': 'errorMessage'
-    }
-
-    def __init__(self, job_id=None, job_type=None, instance_id=None, status=None, created=None, updated=None, error_message=None, local_vars_configuration=None):  # noqa: E501
-        """JobV1 - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration()
-        self.local_vars_configuration = local_vars_configuration
-
-        self._job_id = None
-        self._job_type = None
-        self._instance_id = None
-        self._status = None
-        self._created = None
-        self._updated = None
-        self._error_message = None
-        self.discriminator = None
-
-        self.job_id = job_id
-        self.job_type = job_type
-        self.instance_id = instance_id
-        self.status = status
-        self.created = created
-        self.updated = updated
-        if error_message is not None:
-            self.error_message = error_message
-
-    @property
-    def job_id(self):
-        """Gets the job_id of this JobV1.  # noqa: E501
-
-
-        :return: The job_id of this JobV1.  # noqa: E501
-        :rtype: str
-        """
-        return self._job_id
-
-    @job_id.setter
-    def job_id(self, job_id):
-        """Sets the job_id of this JobV1.
-
-
-        :param job_id: The job_id of this JobV1.  # noqa: E501
-        :type: str
+    JobV1
+    """ # noqa: E501
+    job_id: StrictStr = Field(alias="jobId")
+    job_type: StrictStr = Field(alias="jobType")
+    instance_id: StrictStr = Field(alias="instanceId")
+    status: StrictStr
+    created: datetime
+    updated: datetime
+    error_message: Optional[StrictStr] = Field(default=None, alias="errorMessage")
+    __properties: ClassVar[List[str]] = ["jobId", "jobType", "instanceId", "status", "created", "updated", "errorMessage"]
+
+    @field_validator('job_type')
+    def job_type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['DATA_IMPORT', 'UNKNOWN']):
+            raise ValueError("must be one of enum values ('DATA_IMPORT', 'UNKNOWN')")
+        return value
+
+    @field_validator('status')
+    def status_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['CREATED', 'QUEUED', 'RUNNING', 'SUCCEEDED', 'ERROR', 'CANCELLED', 'UNKNOWN']):
+            raise ValueError("must be one of enum values ('CREATED', 'QUEUED', 'RUNNING', 'SUCCEEDED', 'ERROR', 'CANCELLED', 'UNKNOWN')")
+        return value
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of JobV1 from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
         """
-        if self.local_vars_configuration.client_side_validation and job_id is None:  # noqa: E501
-            raise ValueError("Invalid value for `job_id`, must not be `None`")  # noqa: E501
-
-        self._job_id = job_id
-
-    @property
-    def job_type(self):
-        """Gets the job_type of this JobV1.  # noqa: E501
-
-
-        :return: The job_type of this JobV1.  # noqa: E501
-        :rtype: str
-        """
-        return self._job_type
-
-    @job_type.setter
-    def job_type(self, job_type):
-        """Sets the job_type of this JobV1.
-
-
-        :param job_type: The job_type of this JobV1.  # noqa: E501
-        :type: str
-        """
-        if self.local_vars_configuration.client_side_validation and job_type is None:  # noqa: E501
-            raise ValueError("Invalid value for `job_type`, must not be `None`")  # noqa: E501
-        allowed_values = ["DATA_IMPORT", "UNKNOWN"]  # noqa: E501
-        if self.local_vars_configuration.client_side_validation and job_type not in allowed_values:  # noqa: E501
-            raise ValueError(
-                "Invalid value for `job_type` ({0}), must be one of {1}"  # noqa: E501
-                .format(job_type, allowed_values)
-            )
-
-        self._job_type = job_type
-
-    @property
-    def instance_id(self):
-        """Gets the instance_id of this JobV1.  # noqa: E501
-
-
-        :return: The instance_id of this JobV1.  # noqa: E501
-        :rtype: str
-        """
-        return self._instance_id
-
-    @instance_id.setter
-    def instance_id(self, instance_id):
-        """Sets the instance_id of this JobV1.
-
-
-        :param instance_id: The instance_id of this JobV1.  # noqa: E501
-        :type: str
-        """
-        if self.local_vars_configuration.client_side_validation and instance_id is None:  # noqa: E501
-            raise ValueError("Invalid value for `instance_id`, must not be `None`")  # noqa: E501
-
-        self._instance_id = instance_id
-
-    @property
-    def status(self):
-        """Gets the status of this JobV1.  # noqa: E501
-
-
-        :return: The status of this JobV1.  # noqa: E501
-        :rtype: str
-        """
-        return self._status
-
-    @status.setter
-    def status(self, status):
-        """Sets the status of this JobV1.
-
-
-        :param status: The status of this JobV1.  # noqa: E501
-        :type: str
-        """
-        if self.local_vars_configuration.client_side_validation and status is None:  # noqa: E501
-            raise ValueError("Invalid value for `status`, must not be `None`")  # noqa: E501
-        allowed_values = ["CREATED", "QUEUED", "RUNNING", "SUCCEEDED", "ERROR", "CANCELLED", "UNKNOWN"]  # noqa: E501
-        if self.local_vars_configuration.client_side_validation and status not in allowed_values:  # noqa: E501
-            raise ValueError(
-                "Invalid value for `status` ({0}), must be one of {1}"  # noqa: E501
-                .format(status, allowed_values)
-            )
-
-        self._status = status
-
-    @property
-    def created(self):
-        """Gets the created of this JobV1.  # noqa: E501
-
-
-        :return: The created of this JobV1.  # noqa: E501
-        :rtype: datetime
-        """
-        return self._created
-
-    @created.setter
-    def created(self, created):
-        """Sets the created of this JobV1.
-
-
-        :param created: The created of this JobV1.  # noqa: E501
-        :type: datetime
-        """
-        if self.local_vars_configuration.client_side_validation and created is None:  # noqa: E501
-            raise ValueError("Invalid value for `created`, must not be `None`")  # noqa: E501
-
-        self._created = created
-
-    @property
-    def updated(self):
-        """Gets the updated of this JobV1.  # noqa: E501
-
-
-        :return: The updated of this JobV1.  # noqa: E501
-        :rtype: datetime
-        """
-        return self._updated
-
-    @updated.setter
-    def updated(self, updated):
-        """Sets the updated of this JobV1.
-
-
-        :param updated: The updated of this JobV1.  # noqa: E501
-        :type: datetime
-        """
-        if self.local_vars_configuration.client_side_validation and updated is None:  # noqa: E501
-            raise ValueError("Invalid value for `updated`, must not be `None`")  # noqa: E501
-
-        self._updated = updated
-
-    @property
-    def error_message(self):
-        """Gets the error_message of this JobV1.  # noqa: E501
-
-
-        :return: The error_message of this JobV1.  # noqa: E501
-        :rtype: str
-        """
-        return self._error_message
-
-    @error_message.setter
-    def error_message(self, error_message):
-        """Sets the error_message of this JobV1.
-
-
-        :param error_message: The error_message of this JobV1.  # noqa: E501
-        :type: str
-        """
-
-        self._error_message = error_message
-
-    def to_dict(self):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
-                    value
-                ))
-            elif hasattr(value, "to_dict"):
-                result[attr] = value.to_dict()
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], item[1].to_dict())
-                    if hasattr(item[1], "to_dict") else item,
-                    value.items()
-                ))
-            else:
-                result[attr] = value
-
-        return result
-
-    def to_str(self):
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self):
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other):
-        """Returns true if both objects are equal"""
-        if not isinstance(other, JobV1):
-            return False
-
-        return self.to_dict() == other.to_dict()
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of JobV1 from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "jobId": obj.get("jobId"),
+            "jobType": obj.get("jobType"),
+            "instanceId": obj.get("instanceId"),
+            "status": obj.get("status"),
+            "created": obj.get("created"),
+            "updated": obj.get("updated"),
+            "errorMessage": obj.get("errorMessage")
+        })
+        return _obj
 
-    def __ne__(self, other):
-        """Returns true if both objects are not equal"""
-        if not isinstance(other, JobV1):
-            return True
 
-        return self.to_dict() != other.to_dict()

wds_client/models/record_query_response.py

@@ -3,177 +3,101 @@
 """
     Workspace Data Service
 
-    This page lists current APIs. As of v0.2, all APIs are subject to change without notice.   # noqa: E501
+    This page lists current APIs.  All v0.2 APIs are subject to change without notice.  Changelog at [https://github.com/DataBiosphere/terra-workspace-data-service/releases](https://github.com/DataBiosphere/terra-workspace-data-service/releases) 
 
     The version of the OpenAPI document: v0.2
-    Generated by: https://openapi-generator.tech
-"""
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
 
 
+from __future__ import annotations
 import pprint
 import re  # noqa: F401
+import json
 
-import six
-
-from wds_client.configuration import Configuration
+from pydantic import BaseModel, ConfigDict, Field, StrictInt
+from typing import Any, ClassVar, Dict, List
+from wds_client.models.record_response import RecordResponse
+from wds_client.models.search_request import SearchRequest
+from typing import Optional, Set
+from typing_extensions import Self
 
-
-class RecordQueryResponse(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
+class RecordQueryResponse(BaseModel):
     """
-
-    """
-    Attributes:
-      openapi_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
-    """
-    openapi_types = {
-        'search_request': 'SearchRequest',
-        'total_records': 'int',
-        'records': 'list[RecordResponse]'
-    }
-
-    attribute_map = {
-        'search_request': 'searchRequest',
-        'total_records': 'totalRecords',
-        'records': 'records'
-    }
-
-    def __init__(self, search_request=None, total_records=None, records=None, local_vars_configuration=None):  # noqa: E501
-        """RecordQueryResponse - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration()
-        self.local_vars_configuration = local_vars_configuration
-
-        self._search_request = None
-        self._total_records = None
-        self._records = None
-        self.discriminator = None
-
-        self.search_request = search_request
-        self.total_records = total_records
-        self.records = records
-
-    @property
-    def search_request(self):
-        """Gets the search_request of this RecordQueryResponse.  # noqa: E501
-
-
-        :return: The search_request of this RecordQueryResponse.  # noqa: E501
-        :rtype: SearchRequest
-        """
-        return self._search_request
-
-    @search_request.setter
-    def search_request(self, search_request):
-        """Sets the search_request of this RecordQueryResponse.
-
-
-        :param search_request: The search_request of this RecordQueryResponse.  # noqa: E501
-        :type: SearchRequest
-        """
-        if self.local_vars_configuration.client_side_validation and search_request is None:  # noqa: E501
-            raise ValueError("Invalid value for `search_request`, must not be `None`")  # noqa: E501
-
-        self._search_request = search_request
-
-    @property
-    def total_records(self):
-        """Gets the total_records of this RecordQueryResponse.  # noqa: E501
-
-        number of records in the record type  # noqa: E501
-
-        :return: The total_records of this RecordQueryResponse.  # noqa: E501
-        :rtype: int
-        """
-        return self._total_records
-
-    @total_records.setter
-    def total_records(self, total_records):
-        """Sets the total_records of this RecordQueryResponse.
-
-        number of records in the record type  # noqa: E501
-
-        :param total_records: The total_records of this RecordQueryResponse.  # noqa: E501
-        :type: int
+    RecordQueryResponse
+    """ # noqa: E501
+    search_request: SearchRequest = Field(alias="searchRequest")
+    total_records: StrictInt = Field(description="number of records in the record type", alias="totalRecords")
+    records: List[RecordResponse] = Field(description="list of records found")
+    __properties: ClassVar[List[str]] = ["searchRequest", "totalRecords", "records"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of RecordQueryResponse from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
         """
-        if self.local_vars_configuration.client_side_validation and total_records is None:  # noqa: E501
-            raise ValueError("Invalid value for `total_records`, must not be `None`")  # noqa: E501
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of search_request
+        if self.search_request:
+            _dict['searchRequest'] = self.search_request.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in records (list)
+        _items = []
+        if self.records:
+            for _item in self.records:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict['records'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of RecordQueryResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "searchRequest": SearchRequest.from_dict(obj["searchRequest"]) if obj.get("searchRequest") is not None else None,
+            "totalRecords": obj.get("totalRecords"),
+            "records": [RecordResponse.from_dict(_item) for _item in obj["records"]] if obj.get("records") is not None else None
+        })
+        return _obj
 
-        self._total_records = total_records
 
-    @property
-    def records(self):
-        """Gets the records of this RecordQueryResponse.  # noqa: E501
-
-        list of records found  # noqa: E501
-
-        :return: The records of this RecordQueryResponse.  # noqa: E501
-        :rtype: list[RecordResponse]
-        """
-        return self._records
-
-    @records.setter
-    def records(self, records):
-        """Sets the records of this RecordQueryResponse.
-
-        list of records found  # noqa: E501
-
-        :param records: The records of this RecordQueryResponse.  # noqa: E501
-        :type: list[RecordResponse]
-        """
-        if self.local_vars_configuration.client_side_validation and records is None:  # noqa: E501
-            raise ValueError("Invalid value for `records`, must not be `None`")  # noqa: E501
-
-        self._records = records
-
-    def to_dict(self):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
-                    value
-                ))
-            elif hasattr(value, "to_dict"):
-                result[attr] = value.to_dict()
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], item[1].to_dict())
-                    if hasattr(item[1], "to_dict") else item,
-                    value.items()
-                ))
-            else:
-                result[attr] = value
-
-        return result
-
-    def to_str(self):
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self):
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other):
-        """Returns true if both objects are equal"""
-        if not isinstance(other, RecordQueryResponse):
-            return False
-
-        return self.to_dict() == other.to_dict()
-
-    def __ne__(self, other):
-        """Returns true if both objects are not equal"""
-        if not isinstance(other, RecordQueryResponse):
-            return True
-
-        return self.to_dict() != other.to_dict()