polyaxon-sdk 1.18.0rc7__py2.py3-none-any.whl → 2.1.6__py2.py3-none-any.whl

polyaxon_sdk/models/v1_agent_state_response.py

@@ -1,226 +1,82 @@
-#!/usr/bin/python
-#
-# Copyright 2018-2022 Polyaxon, Inc.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
 # coding: utf-8
 
 """
     Polyaxon SDKs and REST API specification.
 
-    Polyaxon SDKs and REST API specification.  # noqa: E501
+         # noqa: E501
 
-    The version of the OpenAPI document: 1.18.0
+    The version of the OpenAPI document: 2.1.6
     Contact: contact@polyaxon.com
-    Generated by: https://openapi-generator.tech
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
 """
 
 
-try:
-    from inspect import getfullargspec
-except ImportError:
-    from inspect import getargspec as getfullargspec
+from __future__ import annotations
+from inspect import getfullargspec
 import pprint
 import re  # noqa: F401
-import six
-
-from polyaxon_sdk.configuration import Configuration
+import json
 
 
-class V1AgentStateResponse(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
-    """
+from typing import Any, Dict, Optional
+from pydantic import BaseModel, StrictInt
+from polyaxon_sdk.models.agent_state_response_agent_state import AgentStateResponseAgentState
+from polyaxon_sdk.models.v1_statuses import V1Statuses
 
+class V1AgentStateResponse(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.
+    V1AgentStateResponse
     """
-    openapi_types = {
-        'status': 'V1Statuses',
-        'state': 'AgentStateResponseAgentState',
-        'live_state': 'int',
-        'compatible_updates': 'object'
-    }
-
-    attribute_map = {
-        'status': 'status',
-        'state': 'state',
-        'live_state': 'live_state',
-        'compatible_updates': 'compatible_updates'
-    }
-
-    def __init__(self, status=None, state=None, live_state=None, compatible_updates=None, local_vars_configuration=None):  # noqa: E501
-        """V1AgentStateResponse - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration.get_default_copy()
-        self.local_vars_configuration = local_vars_configuration
-
-        self._status = None
-        self._state = None
-        self._live_state = None
-        self._compatible_updates = None
-        self.discriminator = None
-
-        if status is not None:
-            self.status = status
-        if state is not None:
-            self.state = state
-        if live_state is not None:
-            self.live_state = live_state
-        if compatible_updates is not None:
-            self.compatible_updates = compatible_updates
-
-    @property
-    def status(self):
-        """Gets the status of this V1AgentStateResponse.  # noqa: E501
-
-
-        :return: The status of this V1AgentStateResponse.  # noqa: E501
-        :rtype: V1Statuses
-        """
-        return self._status
-
-    @status.setter
-    def status(self, status):
-        """Sets the status of this V1AgentStateResponse.
-
-
-        :param status: The status of this V1AgentStateResponse.  # noqa: E501
-        :type status: V1Statuses
-        """
-
-        self._status = status
-
-    @property
-    def state(self):
-        """Gets the state of this V1AgentStateResponse.  # noqa: E501
-
-
-        :return: The state of this V1AgentStateResponse.  # noqa: E501
-        :rtype: AgentStateResponseAgentState
-        """
-        return self._state
-
-    @state.setter
-    def state(self, state):
-        """Sets the state of this V1AgentStateResponse.
-
-
-        :param state: The state of this V1AgentStateResponse.  # noqa: E501
-        :type state: AgentStateResponseAgentState
-        """
-
-        self._state = state
-
-    @property
-    def live_state(self):
-        """Gets the live_state of this V1AgentStateResponse.  # noqa: E501
-
-
-        :return: The live_state of this V1AgentStateResponse.  # noqa: E501
-        :rtype: int
-        """
-        return self._live_state
-
-    @live_state.setter
-    def live_state(self, live_state):
-        """Sets the live_state of this V1AgentStateResponse.
-
-
-        :param live_state: The live_state of this V1AgentStateResponse.  # noqa: E501
-        :type live_state: int
-        """
-
-        self._live_state = live_state
-
-    @property
-    def compatible_updates(self):
-        """Gets the compatible_updates of this V1AgentStateResponse.  # noqa: E501
-
-
-        :return: The compatible_updates of this V1AgentStateResponse.  # noqa: E501
-        :rtype: object
-        """
-        return self._compatible_updates
-
-    @compatible_updates.setter
-    def compatible_updates(self, compatible_updates):
-        """Sets the compatible_updates of this V1AgentStateResponse.
-
-
-        :param compatible_updates: The compatible_updates of this V1AgentStateResponse.  # noqa: E501
-        :type compatible_updates: object
-        """
-
-        self._compatible_updates = compatible_updates
-
-    def to_dict(self, serialize=False):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        def convert(x):
-            if hasattr(x, "to_dict"):
-                args = getfullargspec(x.to_dict).args
-                if len(args) == 1:
-                    return x.to_dict()
-                else:
-                    return x.to_dict(serialize)
-            else:
-                return x
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            attr = self.attribute_map.get(attr, attr) if serialize else attr
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: convert(x),
-                    value
-                ))
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], convert(item[1])),
-                    value.items()
-                ))
-            else:
-                result[attr] = convert(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, V1AgentStateResponse):
-            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, V1AgentStateResponse):
-            return True
+    status: Optional[V1Statuses] = None
+    state: Optional[AgentStateResponseAgentState] = None
+    live_state: Optional[StrictInt] = None
+    compatible_updates: Optional[Dict[str, Any]] = None
+    __properties = ["status", "state", "live_state", "compatible_updates"]
+
+    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) -> V1AgentStateResponse:
+        """Create an instance of V1AgentStateResponse 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) -> V1AgentStateResponse:
+        """Create an instance of V1AgentStateResponse from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1AgentStateResponse.parse_obj(obj)
+
+        _obj = V1AgentStateResponse.parse_obj({
+            "status": obj.get("status"),
+            "state": AgentStateResponseAgentState.from_dict(obj.get("state")) if obj.get("state") is not None else None,
+            "live_state": obj.get("live_state"),
+            "compatible_updates": obj.get("compatible_updates")
+        })
+        return _obj
 
-        return self.to_dict() != other.to_dict()

polyaxon_sdk/models/v1_agent_status_body_request.py

@@ -1,200 +1,79 @@
-#!/usr/bin/python
-#
-# Copyright 2018-2022 Polyaxon, Inc.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
 # coding: utf-8
 
 """
     Polyaxon SDKs and REST API specification.
 
-    Polyaxon SDKs and REST API specification.  # noqa: E501
+         # noqa: E501
 
-    The version of the OpenAPI document: 1.18.0
+    The version of the OpenAPI document: 2.1.6
     Contact: contact@polyaxon.com
-    Generated by: https://openapi-generator.tech
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
 """
 
 
-try:
-    from inspect import getfullargspec
-except ImportError:
-    from inspect import getargspec as getfullargspec
+from __future__ import annotations
+from inspect import getfullargspec
 import pprint
 import re  # noqa: F401
-import six
-
-from polyaxon_sdk.configuration import Configuration
+import json
 
 
-class V1AgentStatusBodyRequest(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
-    """
+from typing import Optional
+from pydantic import BaseModel, StrictStr
+from polyaxon_sdk.models.v1_status_condition import V1StatusCondition
 
+class V1AgentStatusBodyRequest(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.
+    V1AgentStatusBodyRequest
     """
-    openapi_types = {
-        'owner': 'str',
-        'uuid': 'str',
-        'condition': 'V1StatusCondition'
-    }
-
-    attribute_map = {
-        'owner': 'owner',
-        'uuid': 'uuid',
-        'condition': 'condition'
-    }
-
-    def __init__(self, owner=None, uuid=None, condition=None, local_vars_configuration=None):  # noqa: E501
-        """V1AgentStatusBodyRequest - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration.get_default_copy()
-        self.local_vars_configuration = local_vars_configuration
-
-        self._owner = None
-        self._uuid = None
-        self._condition = None
-        self.discriminator = None
-
-        if owner is not None:
-            self.owner = owner
-        if uuid is not None:
-            self.uuid = uuid
-        if condition is not None:
-            self.condition = condition
-
-    @property
-    def owner(self):
-        """Gets the owner of this V1AgentStatusBodyRequest.  # noqa: E501
-
-
-        :return: The owner of this V1AgentStatusBodyRequest.  # noqa: E501
-        :rtype: str
-        """
-        return self._owner
-
-    @owner.setter
-    def owner(self, owner):
-        """Sets the owner of this V1AgentStatusBodyRequest.
-
-
-        :param owner: The owner of this V1AgentStatusBodyRequest.  # noqa: E501
-        :type owner: str
-        """
-
-        self._owner = owner
-
-    @property
-    def uuid(self):
-        """Gets the uuid of this V1AgentStatusBodyRequest.  # noqa: E501
-
-
-        :return: The uuid of this V1AgentStatusBodyRequest.  # noqa: E501
-        :rtype: str
-        """
-        return self._uuid
-
-    @uuid.setter
-    def uuid(self, uuid):
-        """Sets the uuid of this V1AgentStatusBodyRequest.
-
-
-        :param uuid: The uuid of this V1AgentStatusBodyRequest.  # noqa: E501
-        :type uuid: str
-        """
-
-        self._uuid = uuid
-
-    @property
-    def condition(self):
-        """Gets the condition of this V1AgentStatusBodyRequest.  # noqa: E501
-
-
-        :return: The condition of this V1AgentStatusBodyRequest.  # noqa: E501
-        :rtype: V1StatusCondition
-        """
-        return self._condition
-
-    @condition.setter
-    def condition(self, condition):
-        """Sets the condition of this V1AgentStatusBodyRequest.
-
-
-        :param condition: The condition of this V1AgentStatusBodyRequest.  # noqa: E501
-        :type condition: V1StatusCondition
-        """
-
-        self._condition = condition
-
-    def to_dict(self, serialize=False):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        def convert(x):
-            if hasattr(x, "to_dict"):
-                args = getfullargspec(x.to_dict).args
-                if len(args) == 1:
-                    return x.to_dict()
-                else:
-                    return x.to_dict(serialize)
-            else:
-                return x
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            attr = self.attribute_map.get(attr, attr) if serialize else attr
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: convert(x),
-                    value
-                ))
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], convert(item[1])),
-                    value.items()
-                ))
-            else:
-                result[attr] = convert(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, V1AgentStatusBodyRequest):
-            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, V1AgentStatusBodyRequest):
-            return True
+    owner: Optional[StrictStr] = None
+    uuid: Optional[StrictStr] = None
+    condition: Optional[V1StatusCondition] = None
+    __properties = ["owner", "uuid", "condition"]
+
+    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) -> V1AgentStatusBodyRequest:
+        """Create an instance of V1AgentStatusBodyRequest 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 condition
+        if self.condition:
+            _dict['condition'] = self.condition.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> V1AgentStatusBodyRequest:
+        """Create an instance of V1AgentStatusBodyRequest from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1AgentStatusBodyRequest.parse_obj(obj)
+
+        _obj = V1AgentStatusBodyRequest.parse_obj({
+            "owner": obj.get("owner"),
+            "uuid": obj.get("uuid"),
+            "condition": V1StatusCondition.from_dict(obj.get("condition")) if obj.get("condition") is not None else None
+        })
+        return _obj
 
-        return self.to_dict() != other.to_dict()