Merge pull request #2 from InfraSIM/mark-dev-branch

Support to modify EXSi vm memory size

Author: sharkconi

AutoDeployUI/urls.py

@@ -42,6 +42,7 @@
     url(r'^api/v1/esxi/(?P<id>[0-9]+)/datastores$', 'app.view.views.esxi_datastore', name='esxi_get_datastore'),
     url(r'^api/v1/esxi/(?P<id>[0-9]+)/networks$', 'app.view.views.esxi_network', name='esxi_get_network'),
     url(r'^api/v1/esxi/(?P<id>[0-9]+)/deploy$', 'app.view.views.esxi_deploy', name='esxi_deploy'),
+    url(r'^api/v1/esxi/(?P<id>[0-9]+)/changemem$', 'app.view.views.esxi_change_memory', name='esxi_change_memory'),
     url(r'^api/v1/esxi/(?P<id>[0-9]+)/adddrive$', 'app.view.views.esxi_add_drive', name='esxi_add_drive'),
     url(r'^api/v1/esxi/(?P<id>[0-9]+)/addnic$', 'app.view.views.esxi_add_nic', name='esxi_add_nic'),
     url(r'^api/v1/esxi/(?P<id>[0-9]+)/vpduhostlist$', 'app.view.views.esxi_vpdu_host_config_list', name='esxi_vpdu_host_config_list'),

app/view/views.py

@@ -76,6 +76,39 @@ def esxi_network(request, id, format=None):
     else:
         return Response(network, status=status.HTTP_200_OK)
 
+
+@api_view(('POST'),)
+def esxi_change_memory(request, id, format=None):
+    """
+    Change VM Memory
+    ---
+    parameters:
+       - name: name
+         description: Please input VM name
+         required: true
+         type: string
+         paramType: form
+       - name: size
+         description: Please input memory size(MB)
+         required: true
+         type: string
+         paramType: form
+    """
+    try:
+        esxi = ESXi.objects.get(id = id)
+    except ESXi.DoesNotExist:
+        content = "Cannot find the ESXi."
+        return HttpResponse(content, status = status.HTTP_404_NOT_FOUND)
+
+    usr = esxi.username
+    pwd = esxi.password
+    host = esxi.esxiIP
+
+    content = vRackBuilder.esxi_change_memory(host, usr, pwd, request.data["name"], request.data["size"])
+    if "fail" in content.lower() or "can't" in content.lower():
+        return Response("Fail to Change VM memory size, please check your VM or ESXi", status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+
 @api_view(('POST',))
 def esxi_add_drive(request, id, format=None):
     """

lib/vRackBuilder/api.py

@@ -222,6 +222,30 @@ def esxi_destory_vm(host, usr, pwd, vm_name):
 
     return "Destory VM(%s) success!" % vm_name
 
+def esxi_change_memory(host, usr, pwd, vm_name, mem_size):
+    try:
+        si = connect.Connect(host=host, user=usr, pwd=pwd)
+        vm_view = si.content.viewManager.CreateContainerView(si.content.rootFolder, [vim.VirtualMachine], True)
+        uuid = ""
+        for vm in vm_view.view:
+            if vm.summary.config.name == vm_name:
+                uuid = vm.summary.config.instanceUuid
+                break
+        if uuid == "":
+            connect.Disconnect(si)
+            return "Can't find VM(%s)!" % vm_name
+
+        vm = si.content.searchIndex.FindByUuid(None, uuid, True, True)
+        spec = vim.vm.ConfigSpec()
+        spec.memoryMB = int(mem_size) + 512
+        task = vm.ReconfigVM_Task(spec)
+        tasks.wait_for_tasks(si, [task])
+
+        connect.Disconnect(si)
+    except Exception, e:
+        return "Change VM Memory on VM(%s) failed!" % vm_name
+    return "Change VM(%s) Memory successfully!" % vm_name
+
 def esxi_add_drive(host, usr, pwd, vm_name, disk_size):
     try:
         si = connect.Connect(host=host, user=usr, pwd=pwd)