Merge branch 'mass_create_for_users' into 'master'

Mass create for users

See merge request !415

circle/vm/management/commands/mass_create_vms.py

+# Copyright 2014 Budapest University of Technology and Economics (BME IK)
+#
+# This file is part of CIRCLE Cloud.
+#
+# CIRCLE is free software: you can redistribute it and/or modify it under
+# the terms of the GNU General Public License as published by the Free
+# Software Foundation, either version 3 of the License, or (at your option)
+# any later version.
+#
+# CIRCLE is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+# FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
+# details.
+#
+# You should have received a copy of the GNU General Public License along
+# with CIRCLE.  If not, see <http://www.gnu.org/licenses/>.
+
+from django.core.management.base import BaseCommand
+from vm.models import Instance, InstanceTemplate
+
+
+class Command(BaseCommand):
+    def add_arguments(self, parser):
+        parser.add_argument('template_id', type=int)
+        parser.add_argument('users_path')
+
+    def handle(self, *args, **options):
+        template = InstanceTemplate.objects.get(id=options['template_id'])
+        with open(options['users_path']) as f:
+            users = f.read().splitlines()
+        missing_users = Instance.mass_create_for_users(template, users)
+        if len(missing_users) > 0:
+            self.stdout.write('These users do not exist:')
+            for user in missing_users:
+                self.stdout.write(user)

circle/vm/models/instance.py

@@ -439,6 +439,28 @@ class Instance(AclBase, VirtualMachineDescModel, StatusModel, OperatedMixin,
         return [cls.create(cps, disks, networks, req_traits, tags)
         return [cls.create(cps, disks, networks, req_traits, tags)
                 for cps in customized_params]
                 for cps in customized_params]
 
 
+    @classmethod
+    def mass_create_for_users(cls, template, users, **kwargs):
+        """
+        Create and deploy an instance of a template for each user
+        in a list of users. Returns the user IDs of missing users.
+        """
+        user_instances = []
+        missing_users = []
+        for user_id in users:
+            try:
+                user_instances.append(User.objects.get(profile__org_id=user_id))
+            except User.DoesNotExist:
+                missing_users.append(user_id)
+
+        instances = []
+        for user in user_instances:
+            instance = cls.create_from_template(template, user, **kwargs)
+            instance.deploy(user=user)
+            instances.append(instance)
+
+        return missing_users
+
     def clean(self, *args, **kwargs):
     def clean(self, *args, **kwargs):
         self.time_of_suspend, self.time_of_delete = self.get_renew_times()
         self.time_of_suspend, self.time_of_delete = self.get_renew_times()
         super(Instance, self).clean(*args, **kwargs)
         super(Instance, self).clean(*args, **kwargs)