fix models

pytdx/models/asset.py

@@ -1,12 +1,8 @@
 from typing import Optional, List, Dict
+from base_model import BaseModel
 
 
-class BaseClass:
-    def to_dict(self):
-        return {key: value for key, value in vars(self).items() if value is not None}
-
-
-class Choice(BaseClass):
+class Choice(BaseModel):
     def __init__(
         self,
         ID: Optional[int] = None,
@@ -24,7 +20,7 @@ def __init__(
         self.Order = Order
 
 
-class Attribute(BaseClass):
+class Attribute(BaseModel):
     def __init__(
         self,
         ID: Optional[int] = None,
@@ -60,7 +56,7 @@ def __init__(
         self.Choices = Choices
 
 
-class Asset(BaseClass):
+class Asset(BaseModel):
     def __init__(
         self,
         ID: Optional[int] = None,
@@ -166,7 +162,7 @@ def __init__(
         self.Uri = Uri
 
 
-class AssetModel(BaseClass):
+class AssetModel(BaseModel):
     def __init__(
         self,
         ID: Optional[int] = None,

pytdx/models/attribute.py

@@ -1,12 +1,8 @@
 from typing import Optional, List
+from base_model import BaseModel
 
 
-class BaseClass:
-    def to_dict(self):
-        return {key: value for key, value in vars(self).items() if value is not None}
-
-
-class AttributeChoice(BaseClass):
+class AttributeChoice(BaseModel):
     def __init__(
         self,
         ID: Optional[int] = None,

pytdx/models/base_model.py

@@ -0,0 +1,3 @@
+class BaseModel:
+    def to_dict(self):
+        return {key: value for key, value in vars(self).items() if value is not None}

pytdx/models/knowledge_article.py

@@ -1,7 +1,8 @@
 from typing import Optional, List
+from base_model import BaseModel
 
 
-class Attribute:
+class Attribute(BaseModel):
     def __init__(
         self,
         ID: Optional[int] = None,
@@ -37,7 +38,7 @@ def __init__(
         self.AssociatedItemIDs = AssociatedItemIDs
 
 
-class KnowledgeArticle:
+class KnowledgeArticle(BaseModel):
     def __init__(
         self,
         ID: Optional[int] = None,

pytdx/models/ticket.py

@@ -1,12 +1,8 @@
 from typing import Optional, List
+from base_model import BaseModel
 
 
-class BaseClass:
-    def to_dict(self):
-        return {key: value for key, value in vars(self).items() if value is not None}
-
-
-class NotifyItem(BaseClass):
+class NotifyItem(BaseModel):
     def __init__(
         self,
         ItemRole: Optional[str] = None,
@@ -24,7 +20,7 @@ def __init__(
         self.ProfileImageFileName = ProfileImageFileName
 
 
-class AttributeItem(BaseClass):
+class AttributeItem(BaseModel):
     def __init__(
         self,
         ID: Optional[int] = None,
@@ -60,7 +56,7 @@ def __init__(
         self.AssociatedItemIDs = AssociatedItemIDs
 
 
-class Ticket(BaseClass):
+class Ticket(BaseModel):
     def __init__(
         self,
         ID: Optional[int] = None,
@@ -300,7 +296,7 @@ def __init__(
         self.WorkflowName = WorkflowName
 
 
-class TicketFeed(BaseClass):
+class TicketFeed(BaseModel):
     def __init__(
         self,
         ID: Optional[int] = None,
@@ -370,7 +366,7 @@ def __init__(
         self.Uri = Uri
 
 
-class TicketType(BaseClass):
+class TicketType(BaseModel):
     def __init__(
         self,
         ID: Optional[int] = None,

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name="py-teamdynamix",
-    version="0.2.4",
+    version="0.2.5",
     author="Julien Rossow-Greenberg",
     author_email="[email protected]",
     description="Python client for interacting with the TeamDynamix ITSM APIs",