Add an optional argument for an additional admin user

circle/vm/management/commands/mass_create_vms.py

@@ -21,14 +21,20 @@ from vm.models import Instance, InstanceTemplate
 
 
 class Command(BaseCommand):
 class Command(BaseCommand):
     def add_arguments(self, parser):
     def add_arguments(self, parser):
-        parser.add_argument('template_id', type=int)
-        parser.add_argument('users_path')
+        parser.add_argument('-t', '--template', type=int, required=True)
+        parser.add_argument('-u', '--users', required=True)
+        parser.add_argument('-a', '--admin')
 
 
     def handle(self, *args, **options):
     def handle(self, *args, **options):
-        template = InstanceTemplate.objects.get(id=options['template_id'])
-        with open(options['users_path']) as f:
+        template = InstanceTemplate.objects.get(id=options['template'])
+
+        with open(options['users']) as f:
             users = f.read().splitlines()
             users = f.read().splitlines()
-        missing_users = Instance.mass_create_for_users(template, users)
+
+        missing_users = Instance.mass_create_for_users(
+            template, users, options['admin']
+        )
+
         if len(missing_users) > 0:
         if len(missing_users) > 0:
             self.stdout.write('These users do not exist:')
             self.stdout.write('These users do not exist:')
             for user in missing_users:
             for user in missing_users:

circle/vm/models/instance.py

@@ -440,7 +440,7 @@ class Instance(AclBase, VirtualMachineDescModel, StatusModel, OperatedMixin,
                 for cps in customized_params]
                 for cps in customized_params]
 
 
     @classmethod
     @classmethod
-    def mass_create_for_users(cls, template, users, **kwargs):
+    def mass_create_for_users(cls, template, users, admin=None, **kwargs):
         """
         """
         Create and deploy an instance of a template for each user
         Create and deploy an instance of a template for each user
         in a list of users. Returns the user IDs of missing users.
         in a list of users. Returns the user IDs of missing users.
@@ -453,11 +453,11 @@ class Instance(AclBase, VirtualMachineDescModel, StatusModel, OperatedMixin,
             except User.DoesNotExist:
             except User.DoesNotExist:
                 missing_users.append(user_id)
                 missing_users.append(user_id)
 
 
-        instances = []
         for user in user_instances:
         for user in user_instances:
             instance = cls.create_from_template(template, user, **kwargs)
             instance = cls.create_from_template(template, user, **kwargs)
+            if admin:
+                instance.set_level(User.objects.get(username=admin), 'owner')
             instance.deploy(user=user)
             instance.deploy(user=user)
-            instances.append(instance)
 
 
         return missing_users
         return missing_users