diff options
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']) |
