diff options
| author | Mitch Riedstra <mitch@riedstra.us> | 2017-10-19 20:39:25 -0400 |
|---|---|---|
| committer | Mitch Riedstra <mitch@riedstra.us> | 2017-10-19 20:39:25 -0400 |
| commit | dd748147a7d3f8379d1ea14036b694490dd0b219 (patch) | |
| tree | cf2b98ca1f813eeec6c37d191f41592703633f32 /app/dispatch/management/commands/insert_test_data.py | |
| parent | a3e719469b9efc0db361a419cd7c2fe04ccaad22 (diff) | |
| download | dispatch-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.py | 50 |
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']) |
