From 0957977a439d0ca8f33cf6b3c7f2457b0bbf7cd6 Mon Sep 17 00:00:00 2001
From: Mitch Riedstra <Mitch@riedstra.us>
Date: Mon, 9 Oct 2017 15:04:39 -0400
Subject: Delete all of the old migrations, remove redundant contact model and
 add a script for adding fake data

---
 .../management/commands/insert_test_data.py        | 50 ++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 app/dispatch/management/commands/insert_test_data.py

(limited to 'app/dispatch/management/commands/insert_test_data.py')

diff --git a/app/dispatch/management/commands/insert_test_data.py b/app/dispatch/management/commands/insert_test_data.py
new file mode 100644
index 0000000..8ca9a33
--- /dev/null
+++ b/app/dispatch/management/commands/insert_test_data.py
@@ -0,0 +1,50 @@
+from django.core.management.base import BaseCommand, CommandError
+from django.contrib.auth import get_user_model
+from dispatch.models import Company, Contact, Load
+
+import yaml
+
+User = get_user_model()
+
+class Command(BaseCommand):
+    help = "Import Objects from a YML file"
+
+    def add_arguments(self, parser):
+        parser.add_argument('--file', type=str, dest='filename')
+
+
+    def handle(self, *args, **options):
+
+        yml_data = open(options['filename']).read()
+        parsed = yaml.load(yml_data)
+
+        self.parse_users(parsed['users'])
+        self.parse_loads(parsed['loads'])
+
+    def parse_loads(self, loads):
+        for l in loads:
+            try:
+                new_user = Load(
+                    load_number=u['load_number'],
+                    date=u['date'],
+                    user=u['user_email'],
+                    user=u['user'],
+                )
+                new_user.save()
+            except:
+                print("Error on user %s" % u['email'])
+
+
+    def parse_users(self, users):
+        for u in users:
+            try:
+                new_user = User(
+                    first_name=u['firstName'],
+                    last_name=u['lastName'],
+                    email=u['email'],
+                    username=u['firstName']
+                    is_active=u['active']
+                )
+                new_user.save()
+            except:
+                print("Error on user %s" % u['email'])
-- 
cgit v1.2.3