1 files changed, 11 insertions, 12 deletions
diff --git a/app/dispatch/management/commands/insert_fake_data.py b/app/dispatch/management/commands/insert_fake_data.py
index 0919181..5d4203d 100644
--- a/app/dispatch/management/commands/insert_fake_data.py
+++ b/app/dispatch/management/commands/insert_fake_data.py
@@ -1,6 +1,6 @@
 from django.core.management.base import BaseCommand, CommandError
 from django.contrib.auth import get_user_model
-from dispatch.models import Company, Load
+from dispatch.models import Customer, Load
 from faker import Faker
 import random
 
@@ -35,15 +35,15 @@ class Command(BaseCommand):
         self.end_date   = options['enddate']
         self.start_date = options['startdate']
 
-        company_ids = []
+        customer_ids = []
         user_ids = []
 
         for i in range(0, options['companies']):
-            co = self.fake_company()
+            co = self.fake_customer()
 
             # We're going to hold the model IDs in memory to access randomly
             # later on
-            company_ids.append(co.pk)
+            customer_ids.append(co.pk)
 
         for i in range(0, options['users']):
             u = self.fake_user()
@@ -51,37 +51,36 @@ class Command(BaseCommand):
             user_ids.append(u.pk)
 
         for i in range(0, options['loads']):
-            co_id = company_ids[random.randint(0, len(company_ids)-1)]
-            co = Company.objects.get(pk=co_id)
+            co_id = customer_ids[random.randint(0, len(customer_ids)-1)]
+            co = Customer.objects.get(pk=co_id)
 
             u_id = user_ids[random.randint(0, len(user_ids)-1)]
             u = User.objects.get(pk=u_id)
 
             l = self.fake_load(u, co)
 
-    def fake_company(self):
-        new_company = Company(
+    def fake_customer(self):
+        new_customer = Customer(
                 name=self.fake.company(),
                 address=self.fake.address(),
                 phone_number=self.fake.msisdn(),
                 email_address=self.fake.company_email(),
                 contact_name=self.fake.name(),
             )
-        new_company.save()
-        return new_company
+        new_customer.save()
+        return new_customer
 
     def fake_load(self, usr, co):
         fake_description = "{}-{} {} {}K".format(
             self.fake.month(), self.fake.day_of_month(),
             self.fake.license_plate(), random.randint(5, 50))
-        print(fake_description)
         new_load = Load(
             user=usr,
             # Because that's going to be random enough
             # load_number=self.fake.license_plate(),
             # description=self.fake.sentence(nb_words=6),
             description=fake_description,
-            company=co,
+            customer=co,
             amount=random.randint(150,2500),
             delivered_to=self.fake.city(),
         )