Added DeleteMixin

FabioFleitas · FabioFleitas · commit 1df1294b32cf · 2017-02-09T19:37:47.000-08:00
diff --git a/quickbooks/client.py b/quickbooks/client.py
@@ -349,6 +349,12 @@ def update_object(self, qbbo, request_body, _file_path=None):
 
         return result
 
+    def delete_object(self, qbbo, request_body, _file_path=None):
+        url = self.api_url + "/company/{0}/{1}".format(self.company_id, qbbo.lower())
+        result = self.make_request("POST", url, request_body, params={'operation': 'delete'}, file_path=_file_path)
+
+        return result
+
     def batch_operation(self, request_body):
         url = self.api_url + "/company/{0}/batch".format(self.company_id)
         results = self.make_request("POST", url, request_body)
diff --git a/quickbooks/mixins.py b/quickbooks/mixins.py
@@ -80,6 +80,23 @@ def save(self, qb=None):
         return obj
 
 
+class DeleteMixin(object):
+    qbo_object_name = ""
+
+    def delete(self, qb=None):
+        if not qb:
+            qb = QuickBooks()
+
+        if not self.Id:
+            raise QuickbooksException('Cannot delete unsaved object')
+
+        data = {
+            'Id': self.Id,
+            'SyncToken': self.SyncToken,
+        }
+        return qb.delete_object(self.qbo_object_name, json.dumps(data))
+
+
 class ListMixin(object):
     qbo_object_name = ""
 
