Updated budget.

ej2 · ej2 · commit 2d8199f78aaf · 2015-09-10T10:17:29.000-05:00
diff --git a/quickbooks/objects/budget.py b/quickbooks/objects/budget.py
@@ -1,10 +1,12 @@
-from base import QuickbooksBaseObject, Ref, QuickbooksManagedObject
+from base import QuickbooksBaseObject, Ref, QuickbooksManagedObject, QuickbooksTransactionEntity
 
 
 class BudgetDetail(QuickbooksBaseObject):
     class_dict = {
         "AccountRef": Ref,
-        "CustomerRef": Ref
+        "CustomerRef": Ref,
+        "ClassRef": Ref,
+        "DepartmentRef": Ref,
     }
 
     def __init__(self):
@@ -14,20 +16,22 @@ def __init__(self):
 
         self.AccountRef = None
         self.CustomerRef = None
+        self.ClassRef = None
+        self.DepartmentRef = None
 
     def __unicode__(self):
         return str(self.Amount)
 
 
-class Budget(QuickbooksManagedObject):
+class Budget(QuickbooksManagedObject, QuickbooksTransactionEntity):
     """
     QBO definition: The Budget endpoint allows you to retrieve the current state of budgets already set up in the user's
     company file. A budget allows for an amount to be assigned on a monthly, quarterly, or annual basis for a specific
     account or customer and are created to give a business measurable expense goals. This amount represents how much
     should be spent against that account or customer in the give time period.
     """
 
-    class_dict = {
+    list_dict = {
         "BudgetDetail": BudgetDetail,
     }
 
@@ -42,7 +46,7 @@ def __init__(self):
         self.BudgetEntryType = ""
         self.Active = True
 
-        self.BudgetDetail = None
+        self.BudgetDetail = []
 
     def __unicode__(self):
         return self.Name
