From 3fca9dbb0df0d564f73a111d13c9c8eae825836f Mon Sep 17 00:00:00 2001
From: PCoder <purple.coder@yahoo.co.uk>
Date: Mon, 16 Apr 2018 05:03:48 +0200
Subject: [PATCH] Fix a bug creating hostingorder

---
 datacenterlight/tasks.py | 34 +++++++++++++---------------------
 1 file changed, 13 insertions(+), 21 deletions(-)

diff --git a/datacenterlight/tasks.py b/datacenterlight/tasks.py
index c97c6c54..db479b43 100644
--- a/datacenterlight/tasks.py
+++ b/datacenterlight/tasks.py
@@ -97,30 +97,22 @@ def create_vm_task(self, vm_template_id, user, specs, template,
         if vm_id is None:
             raise Exception("Could not create VM")
 
-        if 'pricing_name' in specs:
-            vm_pricing = VMPricing.get_vm_pricing_by_name(
-                name=specs['pricing_name']
-            )
-            # Create a Hosting Order
-            order = HostingOrder.create(
-                price=final_price,
-                vm_id=vm_id,
-                customer=customer,
-                billing_address=billing_address,
-                vm_pricing=vm_pricing
-            )
-        else:
-            # Create a Hosting Order
-            order = HostingOrder.create(
-                price=final_price,
-                vm_id=vm_id,
-                customer=customer,
-                billing_address=billing_address
-            )
+        vm_pricing = VMPricing.get_vm_pricing_by_name(
+            name=specs['pricing_name']
+        ) if 'pricing_name' in specs else VMPricing.get_default_pricing()
+        # Create a Hosting Order
+        order = HostingOrder.create(
+            price=final_price,
+            vm_id=vm_id,
+            customer=customer,
+            billing_address=billing_address,
+            vm_pricing=vm_pricing
+        )
 
         # Create a Hosting Bill
         HostingBill.create(
-            customer=customer, billing_address=billing_address)
+            customer=customer, billing_address=billing_address
+        )
 
         # Create Billing Address for User if he does not have one
         if not customer.user.billing_addresses.count():