aboutsummaryrefslogtreecommitdiff
path: root/app/dispatch/management/commands/insert_test_data.py
diff options
context:
space:
mode:
authorMitch Riedstra <mitch@riedstra.us>2017-10-19 20:39:25 -0400
committerMitch Riedstra <mitch@riedstra.us>2017-10-19 20:39:25 -0400
commitdd748147a7d3f8379d1ea14036b694490dd0b219 (patch)
treecf2b98ca1f813eeec6c37d191f41592703633f32 /app/dispatch/management/commands/insert_test_data.py
parenta3e719469b9efc0db361a419cd7c2fe04ccaad22 (diff)
downloaddispatch-tracker-dd748147a7d3f8379d1ea14036b694490dd0b219.tar.gz
dispatch-tracker-dd748147a7d3f8379d1ea14036b694490dd0b219.tar.xz
Remove the 'load_number' model field
Diffstat (limited to 'app/dispatch/management/commands/insert_test_data.py')
-rw-r--r--app/dispatch/management/commands/insert_test_data.py50
1 files changed, 0 insertions, 50 deletions
diff --git a/app/dispatch/management/commands/insert_test_data.py b/app/dispatch/management/commands/insert_test_data.py
deleted file mode 100644
index 8ca9a33..0000000
--- a/app/dispatch/management/commands/insert_test_data.py
+++ /dev/null
@@ -1,50 +0,0 @@
-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'])