1 files changed, 37 insertions, 0 deletions
diff --git a/app/dispatch/management/commands/setup_identity.py b/app/dispatch/management/commands/setup_identity.py
new file mode 100644
index 0000000..3251d4e
--- /dev/null
+++ b/app/dispatch/management/commands/setup_identity.py
@@ -0,0 +1,37 @@
+from django.core.management.base import BaseCommand, CommandError
+from django.contrib.auth import get_user_model
+from dispatch.models import Identity
+from faker import Faker
+import random
+
+from django.db.utils import IntegrityError
+
+# import yaml
+
+User = get_user_model()
+
+
+class Command(BaseCommand):
+    help = "with "
+
+    def add_arguments(self, parser):
+        # parser.add_argument('--file', type=str, dest='filename')
+        parser.add_argument('--name', type=str, dest='name', default='My Company')
+        parser.add_argument('--address', type=str, dest='address', default='123 Main St')
+        parser.add_argument('--city', type=str, dest='city', default='Demoville')
+        parser.add_argument('--state', type=str, dest='state', default='Michigan')
+        parser.add_argument('--zip', type=str, dest='zip', default='49503')
+
+
+    def handle(self, *args, **options):
+        u = User.objects.first()
+
+        ident = Identity(
+                user=u,
+                name=options['name'],
+                address=options['address'],
+                city=options['city'],
+                state=options['state'],
+                zip_code=options['zip'])
+        ident.save()
+        ident.set_default()