ExternalVendorCode/gns3-server
1
0
Fork 0
mirror of https://github.com/GNS3/gns3-server.git synced 2025-01-19 03:06:32 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Attempt to fix error when loading wmi module. Fixes #1712

Browse Source
This commit is contained in:
grossmj 2020-01-23 19:04:17 +08:00
parent ab4b391a02
commit fb4ece5c67
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
gns3server/controller/gns3vm/hyperv_gns3_vm.py
View File

@ -21,9 +21,6 @@ import asyncio
import psutil
import ipaddress
if sys.platform.startswith("win"):
import wmi
from .base_gns3_vm import BaseGNS3VM
from .gns3_vm_error import GNS3VMError
log = logging.getLogger(__name__)
@ -47,6 +44,7 @@ class HyperVGNS3VM(BaseGNS3VM):
self._conn = None
self._vm = None
self._management = None
self._wmi = None
def _check_requirements(self):
"""
@ -64,8 +62,12 @@ class HyperVGNS3VM(BaseGNS3VM):
raise GNS3VMError("Hyper-V with nested virtualization is only supported on Windows 10 Anniversary Update (build 10.0.14393) or later")
try:
conn = wmi.WMI()
except wmi.x_wmi as e:
import pythoncom
pythoncom.CoInitialize()
import wmi
self._wmi = wmi
conn = self._wmi.WMI()
except self._wmi.x_wmi as e:
raise GNS3VMError("Could not connect to WMI: {}".format(e))
if not conn.Win32_ComputerSystem()[0].HypervisorPresent:
@ -86,8 +88,8 @@ class HyperVGNS3VM(BaseGNS3VM):
self._check_requirements()
try:
self._conn = wmi.WMI(namespace=r"root\virtualization\v2")
except wmi.x_wmi as e:
self._conn = self._wmi.WMI(namespace=r"root\virtualization\v2")
except self._wmi.x_wmi as e:
raise GNS3VMError("Could not connect to WMI: {}".format(e))
if not self._conn.Msvm_VirtualSystemManagementService():
@ -185,7 +187,7 @@ class HyperVGNS3VM(BaseGNS3VM):
for vm in self._conn.Msvm_ComputerSystem():
if vm.ElementName != self._management.SystemName:
vms.append({"vmname": vm.ElementName})
except wmi.x_wmi as e:
except self._wmi.x_wmi as e:
raise GNS3VMError("Could not list Hyper-V VMs: {}".format(e))
return vms
@ -194,7 +196,7 @@ class HyperVGNS3VM(BaseGNS3VM):
Gets the WMI object.
"""
return wmi.WMI(moniker=path.replace('\\', '/'))
return self._wmi.WMI(moniker=path.replace('\\', '/'))
async def _set_state(self, state):
"""