pojo sync poc - with @DataClass

Author: orkes-harshil

src/conductor/client/http/models/action.py

@@ -1,10 +1,14 @@
 import pprint
 import re  # noqa: F401
+from dataclasses import dataclass, field, InitVar
+from typing import Optional, Dict, List, Any, Union
+from deprecated import deprecated
 
 import six
 
 
-class Action(object):
+@dataclass
+class Action:
     """NOTE: This class is auto generated by the swagger code generator program.
 
     Do not edit the class manually.
@@ -16,25 +20,46 @@ class Action(object):
       attribute_map (dict): The key is attribute name
                             and the value is json key in definition.
     """
-    swagger_types = {
+    # Initialize private fields
+    _action: Optional[str] = field(default=None, init=False)
+    _start_workflow: Any = field(default=None, init=False)
+    _complete_task: Any = field(default=None, init=False)
+    _fail_task: Any = field(default=None, init=False)
+    _expand_inline_json: Optional[bool] = field(default=None, init=False)
+    _terminate_workflow: Any = field(default=None, init=False)
+    _update_workflow_variables: Any = field(default=None, init=False)
+
+    # Init variables for backward compatibility
+    action: InitVar[Optional[str]] = None
+    start_workflow: InitVar[Any] = None
+    complete_task: InitVar[Any] = None
+    fail_task: InitVar[Any] = None
+    expand_inline_json: InitVar[Optional[bool]] = None
+    terminate_workflow: InitVar[Any] = None
+    update_workflow_variables: InitVar[Any] = None
+
+    # Class variables
+    swagger_types: Dict[str, str] = field(default_factory=lambda: {
         'action': 'str',
         'start_workflow': 'StartWorkflowRequest',
         'complete_task': 'TaskDetails',
         'fail_task': 'TaskDetails',
         'expand_inline_json': 'bool',
         'terminate_workflow': 'TerminateWorkflow',
         'update_workflow_variables': 'UpdateWorkflowVariables'
-    }
+    })
 
-    attribute_map = {
+    attribute_map: Dict[str, str] = field(default_factory=lambda: {
         'action': 'action',
         'start_workflow': 'start_workflow',
         'complete_task': 'complete_task',
         'fail_task': 'fail_task',
         'expand_inline_json': 'expandInlineJSON',
         'terminate_workflow': 'terminate_workflow',
         'update_workflow_variables': 'update_workflow_variables'
-    }
+    })
+
+    discriminator = None
 
     def __init__(self, action=None, start_workflow=None, complete_task=None, fail_task=None,
                  expand_inline_json=None, terminate_workflow=None, update_workflow_variables=None):  # noqa: E501
@@ -62,6 +87,24 @@ def __init__(self, action=None, start_workflow=None, complete_task=None, fail_ta
         if update_workflow_variables is not None:
             self.update_workflow_variables = update_workflow_variables
 
+    def __post_init__(self, action, start_workflow, complete_task, fail_task,
+                      expand_inline_json, terminate_workflow, update_workflow_variables):
+        """Handle initialization for dataclass compatibility"""
+        if action is not None:
+            self.action = action
+        if start_workflow is not None:
+            self.start_workflow = start_workflow
+        if complete_task is not None:
+            self.complete_task = complete_task
+        if fail_task is not None:
+            self.fail_task = fail_task
+        if expand_inline_json is not None:
+            self.expand_inline_json = expand_inline_json
+        if terminate_workflow is not None:
+            self.terminate_workflow = terminate_workflow
+        if update_workflow_variables is not None:
+            self.update_workflow_variables = update_workflow_variables
+
     @property
     def action(self):
         """Gets the action of this Action.  # noqa: E501
@@ -80,7 +123,8 @@ def action(self, action):
         :param action: The action of this Action.  # noqa: E501
         :type: str
         """
-        allowed_values = ["start_workflow", "complete_task", "fail_task", "terminate_workflow", "update_workflow_variables"]  # noqa: E501
+        allowed_values = ["start_workflow", "complete_task", "fail_task", "terminate_workflow",
+                          "update_workflow_variables"]  # noqa: E501
         if action not in allowed_values:
             raise ValueError(
                 "Invalid value for `action` ({0}), must be one of {1}"  # noqa: E501

src/conductor/client/http/models/authorization_request.py

@@ -1,10 +1,13 @@
 import pprint
 import re  # noqa: F401
-
+from dataclasses import dataclass, field, InitVar
+from typing import List, Optional, Dict, Any
 import six
+from deprecated import deprecated
 
 
-class AuthorizationRequest(object):
+@dataclass
+class AuthorizationRequest:
     """NOTE: This class is auto generated by the swagger code generator program.
 
     Do not edit the class manually.
@@ -16,17 +19,34 @@ class AuthorizationRequest(object):
       attribute_map (dict): The key is attribute name
                             and the value is json key in definition.
     """
-    swagger_types = {
+    subject: InitVar[Any] = None
+    target: InitVar[Any] = None
+    access: InitVar[List[str]] = None
+
+    _subject = None
+    _target = None
+    _access = None
+
+    # Class variables
+    swagger_types: Dict[str, str] = field(default_factory=lambda: {
         'subject': 'SubjectRef',
         'target': 'TargetRef',
         'access': 'list[str]'
-    }
+    })
 
-    attribute_map = {
+    attribute_map: Dict[str, str] = field(default_factory=lambda: {
         'subject': 'subject',
         'target': 'target',
         'access': 'access'
-    }
+    })
+
+    discriminator = None
+
+    def __post_init__(self, subject, target, access):
+        """Initialize fields after dataclass initialization"""
+        self.subject = subject
+        self.target = target
+        self.access = access
 
     def __init__(self, subject=None, target=None, access=None):  # noqa: E501
         """AuthorizationRequest - a model defined in Swagger"""  # noqa: E501
@@ -152,4 +172,4 @@ def __eq__(self, other):
 
     def __ne__(self, other):
         """Returns true if both objects are not equal"""
-        return not self == other
+        return not self == other

src/conductor/client/http/models/bulk_response.py

@@ -1,10 +1,16 @@
 import pprint
 import re  # noqa: F401
+from dataclasses import dataclass, field, InitVar
+from typing import Dict, List, Optional, Any, TypeVar, Generic, Union
 
 import six
+from deprecated import deprecated
 
+T = TypeVar('T')
 
-class BulkResponse(object):
+
+@dataclass
+class BulkResponse(Generic[T]):
     """NOTE: This class is auto generated by the swagger code generator program.
 
     Do not edit the class manually.
@@ -28,6 +34,13 @@ class BulkResponse(object):
         'message': 'message'
     }
 
+    bulk_error_results: InitVar[Optional[Dict[str, str]]] = None
+    bulk_successful_results: InitVar[Optional[List[T]]] = None
+
+    _bulk_error_results: Optional[Dict[str, str]] = field(default=None, init=False)
+    _bulk_successful_results: Optional[List[T]] = field(default=None, init=False)
+    _message: str = field(default="Bulk Request has been processed.", init=False)
+
     def __init__(self, bulk_error_results=None, bulk_successful_results=None):  # noqa: E501
         """BulkResponse - a model defined in Swagger"""  # noqa: E501
         self._bulk_error_results = None
@@ -39,8 +52,15 @@ def __init__(self, bulk_error_results=None, bulk_successful_results=None):  # no
         if bulk_successful_results is not None:
             self.bulk_successful_results = bulk_successful_results
 
+    def __post_init__(self, bulk_error_results, bulk_successful_results):
+        """Initialize fields after dataclass initialization"""
+        if bulk_error_results is not None:
+            self.bulk_error_results = bulk_error_results
+        if bulk_successful_results is not None:
+            self.bulk_successful_results = bulk_successful_results
+
     @property
-    def bulk_error_results(self):
+    def bulk_error_results(self) -> Optional[Dict[str, str]]:
         """Gets the bulk_error_results of this BulkResponse.  # noqa: E501
 
         Key - entityId Value - error message processing this entity
@@ -51,7 +71,7 @@ def bulk_error_results(self):
         return self._bulk_error_results
 
     @bulk_error_results.setter
-    def bulk_error_results(self, bulk_error_results):
+    def bulk_error_results(self, bulk_error_results: Optional[Dict[str, str]]):
         """Sets the bulk_error_results of this BulkResponse.
 
         Key - entityId Value - error message processing this entity
@@ -63,17 +83,17 @@ def bulk_error_results(self, bulk_error_results):
         self._bulk_error_results = bulk_error_results
 
     @property
-    def bulk_successful_results(self):
+    def bulk_successful_results(self) -> Optional[List[T]]:
         """Gets the bulk_successful_results of this BulkResponse.  # noqa: E501
 
 
         :return: The bulk_successful_results of this BulkResponse.  # noqa: E501
-        :rtype: list[str]
+        :rtype: list[T]
         """
         return self._bulk_successful_results
 
     @bulk_successful_results.setter
-    def bulk_successful_results(self, bulk_successful_results):
+    def bulk_successful_results(self, bulk_successful_results: Optional[List[T]]):
         """Sets the bulk_successful_results of this BulkResponse.
 
 
@@ -84,7 +104,7 @@ def bulk_successful_results(self, bulk_successful_results):
         self._bulk_successful_results = bulk_successful_results
 
     @property
-    def message(self):
+    def message(self) -> str:
         """Gets the message of this BulkResponse.  # noqa: E501
 
 
@@ -93,7 +113,7 @@ def message(self):
         """
         return self._message
 
-    def append_success_response(self, result):
+    def append_success_response(self, result: T):
         """Appends a successful result to bulk_successful_results.  # noqa: E501
 
         :param result: The result to append to bulk_successful_results
@@ -102,7 +122,7 @@ def append_success_response(self, result):
             self._bulk_successful_results = []
         self._bulk_successful_results.append(result)
 
-    def append_failed_response(self, id, error_message):
+    def append_failed_response(self, id: str, error_message: str):
         """Appends a failed result to bulk_error_results.  # noqa: E501
 
         :param id: The entity ID that failed
@@ -112,7 +132,7 @@ def append_failed_response(self, id, error_message):
             self._bulk_error_results = {}
         self._bulk_error_results[id] = error_message
 
-    def to_dict(self):
+    def to_dict(self) -> Dict[str, Any]:
         """Returns the model properties as a dict"""
         result = {}
 
@@ -139,21 +159,21 @@ def to_dict(self):
 
         return result
 
-    def to_str(self):
+    def to_str(self) -> str:
         """Returns the string representation of the model"""
         return pprint.pformat(self.to_dict())
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         """For `print` and `pprint`"""
         return self.to_str()
 
-    def __eq__(self, other):
+    def __eq__(self, other) -> bool:
         """Returns true if both objects are equal"""
         if not isinstance(other, BulkResponse):
             return False
 
         return self.__dict__ == other.__dict__
 
-    def __ne__(self, other):
+    def __ne__(self, other) -> bool:
         """Returns true if both objects are not equal"""
         return not self == other