openstack/os-win
Code Issues Proposed changes

Refactors wmi exceptions usage

Browse Source 
There are conditional imports (Windows specific modules), 
done in most of os-win's modules. One of the modules is wmi, 
is mostly used in try-except blocks.

Imports wmi.x_wmi and wmi.x_wmi_timed_out exceptions 
into os_win.exceptions. Refactors wmi exceptions usage.

Change-Id: I40107892c63d6186fa3d8bac36e56b28ef8c7910
This commit is contained in:
Dorin Paslaru 2016-08-17 21:58:10 +03:00 committed by Claudiu Belu
parent a5927bcbad
commit 61dcd5697e
14 changed files with 54 additions and 76 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
os_win/exceptions.py
View File

@ -17,8 +17,24 @@
Utility class for VM related operations on Hyper-V. Utility class for VM related operations on Hyper-V.
""" """
import sys
from os_win._i18n import _ from os_win._i18n import _
# Define WMI specific exceptions, so WMI won't have to be imported in any
# module that expects those exceptions.
if sys.platform == 'win32':
import wmi
x_wmi = wmi.x_wmi
x_wmi_timed_out = wmi.x_wmi_timed_out
else:
class x_wmi(Exception):
pass
class x_wmi_timed_out(x_wmi):
pass
class OSWinException(Exception): class OSWinException(Exception):
msg_fmt = 'An exception has been encountered.' msg_fmt = 'An exception has been encountered.'

5
os_win/tests/test_base.py
View File

@ -18,8 +18,10 @@ import mock
from oslotest import base from oslotest import base
from six.moves import builtins from six.moves import builtins
from os_win import exceptions
class FakeWMIExc(Exception):
class FakeWMIExc(exceptions.x_wmi):
def __init__(self, hresult=None): def __init__(self, hresult=None):
excepinfo = [None] * 5 + [hresult] excepinfo = [None] * 5 + [hresult]
self.com_error = mock.Mock(excepinfo=excepinfo) self.com_error = mock.Mock(excepinfo=excepinfo)
@ -31,7 +33,6 @@ class OsWinBaseTestCase(base.BaseTestCase):
super(OsWinBaseTestCase, self).setUp() super(OsWinBaseTestCase, self).setUp()
self._mock_wmi = mock.MagicMock() self._mock_wmi = mock.MagicMock()
self._mock_wmi.x_wmi = FakeWMIExc
mock_os = mock.MagicMock(Version='6.3.0') mock_os = mock.MagicMock(Version='6.3.0')
self._mock_wmi.WMI.return_value.Win32_OperatingSystem.return_value = ( self._mock_wmi.WMI.return_value.Win32_OperatingSystem.return_value = (

4
os_win/tests/utils/compute/test_vmutils.py
View File

@ -947,8 +947,6 @@ class VMUtilsTestCase(test_base.OsWinBaseTestCase):
@mock.patch.object(vmutils, 'patcher') @mock.patch.object(vmutils, 'patcher')
def test_vm_power_state_change_event_handler(self, mock_patcher, def test_vm_power_state_change_event_handler(self, mock_patcher,
mock_tpool, mock_sleep): mock_tpool, mock_sleep):
self._mock_wmi.x_wmi_timed_out = exceptions.HyperVException
enabled_state = constants.HYPERV_VM_STATE_ENABLED enabled_state = constants.HYPERV_VM_STATE_ENABLED
hv_enabled_state = self._vmutils._vm_power_states_map[enabled_state] hv_enabled_state = self._vmutils._vm_power_states_map[enabled_state]
fake_event = mock.Mock(ElementName=mock.sentinel.vm_name, fake_event = mock.Mock(ElementName=mock.sentinel.vm_name,
@ -957,7 +955,7 @@ class VMUtilsTestCase(test_base.OsWinBaseTestCase):
fake_listener = ( fake_listener = (
self._vmutils._conn.Msvm_ComputerSystem.watch_for.return_value) self._vmutils._conn.Msvm_ComputerSystem.watch_for.return_value)
mock_tpool.execute.side_effect = (self._mock_wmi.x_wmi_timed_out, mock_tpool.execute.side_effect = (exceptions.x_wmi_timed_out,
fake_event, Exception, fake_event, Exception,
KeyboardInterrupt) KeyboardInterrupt)

12
os_win/tests/utils/network/test_networkutils.py
View File

@ -145,15 +145,13 @@ class NetworkUtilsTestCase(test_base.OsWinBaseTestCase):
@mock.patch.object(networkutils, 'patcher') @mock.patch.object(networkutils, 'patcher')
@mock.patch.object(networkutils.tpool, 'execute') @mock.patch.object(networkutils.tpool, 'execute')
@mock.patch.object(networkutils, 'wmi', create=True)
@mock.patch.object(networkutils.NetworkUtils, '_get_event_wql_query') @mock.patch.object(networkutils.NetworkUtils, '_get_event_wql_query')
def test_get_vnic_event_listener(self, mock_get_event_query, mock_wmi, def test_get_vnic_event_listener(self, mock_get_event_query,
mock_execute, mock_patcher): mock_execute, mock_patcher):
mock_wmi.x_wmi_timed_out = ValueError
event = mock.MagicMock() event = mock.MagicMock()
port_class = self.netutils._conn.Msvm_SyntheticEthernetPortSettingData port_class = self.netutils._conn.Msvm_SyntheticEthernetPortSettingData
wmi_event_listener = port_class.watch_for.return_value wmi_event_listener = port_class.watch_for.return_value
mock_execute.side_effect = [mock_wmi.x_wmi_timed_out, event] mock_execute.side_effect = [exceptions.x_wmi_timed_out, event]
# callback will raise an exception in order to stop iteration in the # callback will raise an exception in order to stop iteration in the
# listener. # listener.
@ -254,14 +252,12 @@ class NetworkUtilsTestCase(test_base.OsWinBaseTestCase):
expected_result = conn.Msvm_ComputerSystem.return_value[0] expected_result = conn.Msvm_ComputerSystem.return_value[0]
self.assertEqual(expected_result, resulted_vm) self.assertEqual(expected_result, resulted_vm)
@mock.patch.object(networkutils, 'wmi', create=True) def test_remove_switch_port(self):
def test_remove_switch_port(self, mock_wmi):
mock_sw_port = self._mock_get_switch_port_alloc() mock_sw_port = self._mock_get_switch_port_alloc()
self.netutils._switch_ports[self._FAKE_PORT_NAME] = mock_sw_port self.netutils._switch_ports[self._FAKE_PORT_NAME] = mock_sw_port
self.netutils._vlan_sds[mock_sw_port.InstanceID] = mock.MagicMock() self.netutils._vlan_sds[mock_sw_port.InstanceID] = mock.MagicMock()
mock_wmi.x_wmi = Exception
self.netutils._jobutils.remove_virt_resource.side_effect = ( self.netutils._jobutils.remove_virt_resource.side_effect = (
mock_wmi.x_wmi) exceptions.x_wmi)
self.netutils.remove_switch_port(self._FAKE_PORT_NAME, False) self.netutils.remove_switch_port(self._FAKE_PORT_NAME, False)

3
os_win/tests/utils/storage/initiator/test_fc_utils.py
View File

@ -29,8 +29,7 @@ class FCUtilsTestCase(base.BaseTestCase):
_FAKE_ADAPTER_NAME = 'fake_adapter_name' _FAKE_ADAPTER_NAME = 'fake_adapter_name'
_FAKE_ADAPTER_WWN = list(range(8)) _FAKE_ADAPTER_WWN = list(range(8))
@mock.patch.object(fc_utils, 'wmi', create=True) def setUp(self):
def setUp(self, mock_wmi):
super(FCUtilsTestCase, self).setUp() super(FCUtilsTestCase, self).setUp()
self._setup_lib_mocks() self._setup_lib_mocks()

15
os_win/tests/utils/storage/test_smbutils.py
View File

@ -72,10 +72,8 @@ class SMBUtilsTestCase(test_base.OsWinBaseTestCase):
UserName=mock.sentinel.username, UserName=mock.sentinel.username,
Password=mock.sentinel.password) Password=mock.sentinel.password)
@mock.patch.object(smbutils, 'wmi', create=True) def test_mount_smb_share_failed(self):
def test_mount_smb_share_failed(self, mock_wmi): self._smb_conn.Msft_SmbMapping.Create.side_effect = exceptions.x_wmi
mock_wmi.x_wmi = Exception
self._smb_conn.Msft_SmbMapping.Create.side_effect = mock_wmi.x_wmi
self.assertRaises(exceptions.SMBException, self.assertRaises(exceptions.SMBException,
self._smbutils.mount_smb_share, self._smbutils.mount_smb_share,
@ -101,14 +99,13 @@ class SMBUtilsTestCase(test_base.OsWinBaseTestCase):
def test_force_unmount_smb_share(self): def test_force_unmount_smb_share(self):
self._test_unmount_smb_share(force=True) self._test_unmount_smb_share(force=True)
@mock.patch.object(smbutils, 'wmi', create=True) def test_unmount_smb_share_wmi_exception(self):
def test_unmount_smb_share_wmi_exception(self, mock_wmi):
mock_wmi.x_wmi = Exception
fake_mapping = mock.Mock() fake_mapping = mock.Mock()
fake_mapping.Remove.side_effect = mock_wmi.x_wmi fake_mapping.Remove.side_effect = exceptions.x_wmi
self._smb_conn.Msft_SmbMapping.return_value = [fake_mapping] self._smb_conn.Msft_SmbMapping.return_value = [fake_mapping]
self.assertRaises(mock_wmi.x_wmi, self._smbutils.unmount_smb_share, self.assertRaises(exceptions.SMBException,
self._smbutils.unmount_smb_share,
mock.sentinel.share_path, force=True) mock.sentinel.share_path, force=True)
@mock.patch.object(smbutils, 'ctypes') @mock.patch.object(smbutils, 'ctypes')

5
os_win/utils/compute/clusterutils.py
View File

@ -28,9 +28,6 @@ from os_win._i18n import _, _LE
from os_win import exceptions from os_win import exceptions
from os_win.utils import baseutils from os_win.utils import baseutils
if sys.platform == 'win32':
import wmi
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
@ -234,5 +231,5 @@ class ClusterUtils(baseutils.BaseUtils):
except Exception: except Exception:
LOG.exception( LOG.exception(
_LE("Exception during failover callback.")) _LE("Exception during failover callback."))
except wmi.x_wmi_timed_out: except exceptions.x_wmi_timed_out:
pass pass

6
os_win/utils/compute/livemigrationutils.py
View File

@ -14,7 +14,6 @@
# under the License. # under the License.
import platform import platform
import sys
from oslo_log import log as logging from oslo_log import log as logging
@ -26,9 +25,6 @@ from os_win.utils.compute import vmutils
from os_win.utils import jobutils from os_win.utils import jobutils
from os_win.utils.storage.initiator import iscsi_wmi_utils from os_win.utils.storage.initiator import iscsi_wmi_utils
if sys.platform == 'win32':
import wmi
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
@ -45,7 +41,7 @@ class LiveMigrationUtils(baseutils.BaseUtilsVirt):
def _get_conn_v2(self, host='localhost'): def _get_conn_v2(self, host='localhost'):
try: try:
return self._get_wmi_obj(self._wmi_namespace % host) return self._get_wmi_obj(self._wmi_namespace % host)
except wmi.x_wmi as ex: except exceptions.x_wmi as ex:
LOG.exception(_LE('Get version 2 connection error')) LOG.exception(_LE('Get version 2 connection error'))
if ex.com_error.hresult == -2147217394: if ex.com_error.hresult == -2147217394:
msg = (_('Live migration is not supported on target host "%s"') msg = (_('Live migration is not supported on target host "%s"')

6
os_win/utils/compute/vmutils.py
View File

@ -21,7 +21,6 @@ Hyper-V Server / Windows Server 2012.
""" """
import functools import functools
import sys
import time import time
import uuid import uuid
@ -42,9 +41,6 @@ from os_win.utils import baseutils
from os_win.utils import jobutils from os_win.utils import jobutils
from os_win.utils import pathutils from os_win.utils import pathutils
if sys.platform == 'win32':
import wmi
CONF = cfg.CONF CONF = cfg.CONF
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
@ -856,7 +852,7 @@ class VMUtils(baseutils.BaseUtilsVirt):
LOG.exception(err_msg, LOG.exception(err_msg,
dict(vm_name=vm_name, dict(vm_name=vm_name,
vm_power_state=vm_power_state)) vm_power_state=vm_power_state))
except wmi.x_wmi_timed_out: except exceptions.x_wmi_timed_out:
pass pass
except Exception: except Exception:
LOG.exception( LOG.exception(

6
os_win/utils/jobutils.py
View File

@ -18,7 +18,6 @@
Base Utility class for operations on Hyper-V. Base Utility class for operations on Hyper-V.
""" """
import sys
import time import time
from oslo_log import log as logging from oslo_log import log as logging
@ -32,9 +31,6 @@ from os_win.utils import win32utils
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
if sys.platform == 'win32':
import wmi
class JobUtils(baseutils.BaseUtilsVirt): class JobUtils(baseutils.BaseUtilsVirt):
@ -127,7 +123,7 @@ class JobUtils(baseutils.BaseUtilsVirt):
job.RequestStateChange( job.RequestStateChange(
self._KILL_JOB_STATE_CHANGE_REQUEST) self._KILL_JOB_STATE_CHANGE_REQUEST)
except wmi.x_wmi as ex: except exceptions.x_wmi as ex:
hresult = win32utils.Win32Utils.get_com_error_hresult( hresult = win32utils.Win32Utils.get_com_error_hresult(
ex.com_error) ex.com_error)
# The job may had been completed right before we've # The job may had been completed right before we've

8
os_win/utils/network/networkutils.py
View File

@ -23,7 +23,6 @@ import re
from eventlet import patcher from eventlet import patcher
from eventlet import tpool from eventlet import tpool
import sys
from os_win._i18n import _ from os_win._i18n import _
from os_win import exceptions from os_win import exceptions
@ -31,9 +30,6 @@ from os_win.utils import _wqlutils
from os_win.utils import baseutils from os_win.utils import baseutils
from os_win.utils import jobutils from os_win.utils import jobutils
if sys.platform == 'win32':
import wmi
class NetworkUtils(baseutils.BaseUtilsVirt): class NetworkUtils(baseutils.BaseUtilsVirt):
@ -230,7 +226,7 @@ class NetworkUtils(baseutils.BaseUtilsVirt):
try: try:
event = listen() event = listen()
callback(event.ElementName) callback(event.ElementName)
except wmi.x_wmi_timed_out: except exceptions.x_wmi_timed_out:
# no new event published. # no new event published.
pass pass
@ -295,7 +291,7 @@ class NetworkUtils(baseutils.BaseUtilsVirt):
if not vnic_deleted: if not vnic_deleted:
try: try:
self._jobutils.remove_virt_resource(sw_port) self._jobutils.remove_virt_resource(sw_port)
except wmi.x_wmi: except exceptions.x_wmi:
# port may have already been destroyed by Hyper-V # port may have already been destroyed by Hyper-V
pass pass

6
os_win/utils/storage/initiator/iscsi_wmi_utils.py
View File

@ -18,7 +18,6 @@
Helper methods for operations related to the management of volumes Helper methods for operations related to the management of volumes
and storage repositories on Windows Server 2012 and above and storage repositories on Windows Server 2012 and above
""" """
import sys
import time import time
from oslo_config import cfg from oslo_config import cfg
@ -30,9 +29,6 @@ from os_win import _utils
from os_win import exceptions from os_win import exceptions
from os_win.utils.storage.initiator import base_iscsi_utils from os_win.utils.storage.initiator import base_iscsi_utils
if sys.platform == 'win32':
import wmi
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
CONF = cfg.CONF CONF = cfg.CONF
@ -99,7 +95,7 @@ class ISCSIInitiatorWMIUtils(base_iscsi_utils.BaseISCSIInitiatorUtils):
target.Connect(NodeAddress=target_iqn, target.Connect(NodeAddress=target_iqn,
IsPersistent=True, **auth) IsPersistent=True, **auth)
time.sleep(CONF.hyperv.volume_attach_retry_interval) time.sleep(CONF.hyperv.volume_attach_retry_interval)
except wmi.x_wmi as exc: except exceptions.x_wmi as exc:
LOG.debug("Attempt %(attempt)d to connect to target " LOG.debug("Attempt %(attempt)d to connect to target "
"%(target_iqn)s failed. Retrying. " "%(target_iqn)s failed. Retrying. "
"WMI exception: %(exc)s " % "WMI exception: %(exc)s " %

5
os_win/utils/storage/smbutils.py
View File

@ -28,7 +28,6 @@ from os_win.utils import win32utils
if sys.platform == 'win32': if sys.platform == 'win32':
kernel32 = ctypes.windll.kernel32 kernel32 = ctypes.windll.kernel32
import wmi
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
@ -61,7 +60,7 @@ class SMBUtils(baseutils.BaseUtils):
self._smb_conn.Msft_SmbMapping.Create(RemotePath=share_path, self._smb_conn.Msft_SmbMapping.Create(RemotePath=share_path,
UserName=username, UserName=username,
Password=password) Password=password)
except wmi.x_wmi as exc: except exceptions.x_wmi as exc:
err_msg = (_( err_msg = (_(
'Unable to mount SMBFS share: %(share_path)s ' 'Unable to mount SMBFS share: %(share_path)s '
'WMI exception: %(wmi_exc)s') % {'share_path': share_path, 'WMI exception: %(wmi_exc)s') % {'share_path': share_path,
@ -81,7 +80,7 @@ class SMBUtils(baseutils.BaseUtils):
mapping.Remove(Force=force) mapping.Remove(Force=force)
except AttributeError: except AttributeError:
pass pass
except wmi.x_wmi: except exceptions.x_wmi:
# If this fails, a 'Generic Failure' exception is raised. # If this fails, a 'Generic Failure' exception is raised.
# This happens even if we unforcefully unmount an in-use # This happens even if we unforcefully unmount an in-use
# share, for which reason we'll simply ignore it in this # share, for which reason we'll simply ignore it in this

33
os_win/utils/storage/target/iscsi_target_utils.py
View File

@ -13,8 +13,6 @@
# License for the specific language governing permissions and limitations # License for the specific language governing permissions and limitations
# under the License. # under the License.
import sys
from oslo_log import log as logging from oslo_log import log as logging
from os_win._i18n import _, _LI from os_win._i18n import _, _LI
@ -25,9 +23,6 @@ from os_win.utils import hostutils
from os_win.utils import pathutils from os_win.utils import pathutils
from os_win.utils import win32utils from os_win.utils import win32utils
if sys.platform == 'win32':
import wmi
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
@ -119,7 +114,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
"""Creates ISCSI target.""" """Creates ISCSI target."""
try: try:
self._conn_wmi.WT_Host.NewHost(HostName=target_name) self._conn_wmi.WT_Host.NewHost(HostName=target_name)
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_code = self._win32utils.get_com_err_code(wmi_exc.com_error) err_code = self._win32utils.get_com_err_code(wmi_exc.com_error)
target_exists = err_code == self._ERR_FILE_EXISTS target_exists = err_code == self._ERR_FILE_EXISTS
@ -141,7 +136,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
return return
wt_host.RemoveAllWTDisks() wt_host.RemoveAllWTDisks()
wt_host.Delete_() wt_host.Delete_()
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_msg = _("Failed to delete ISCSI target %s") err_msg = _("Failed to delete ISCSI target %s")
raise exceptions.ISCSITargetWMIException(err_msg % target_name, raise exceptions.ISCSITargetWMIException(err_msg % target_name,
wmi_exc=wmi_exc) wmi_exc=wmi_exc)
@ -173,7 +168,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
wt_host.CHAPUserName = chap_username wt_host.CHAPUserName = chap_username
wt_host.CHAPSecret = chap_password wt_host.CHAPSecret = chap_password
wt_host.put() wt_host.put()
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_msg = _('Failed to set CHAP credentials on target %s.') err_msg = _('Failed to set CHAP credentials on target %s.')
raise exceptions.ISCSITargetWMIException(err_msg % target_name, raise exceptions.ISCSITargetWMIException(err_msg % target_name,
wmi_exc=wmi_exc) wmi_exc=wmi_exc)
@ -191,7 +186,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
wt_idmethod.Method = id_method wt_idmethod.Method = id_method
wt_idmethod.Value = initiator wt_idmethod.Value = initiator
wt_idmethod.put() wt_idmethod.put()
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_msg = _('Could not associate initiator %(initiator)s to ' err_msg = _('Could not associate initiator %(initiator)s to '
'iSCSI target: %(target_name)s.') 'iSCSI target: %(target_name)s.')
raise exceptions.ISCSITargetWMIException( raise exceptions.ISCSITargetWMIException(
@ -204,7 +199,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
wt_idmethod = self._get_wt_idmethod(initiator, target_name) wt_idmethod = self._get_wt_idmethod(initiator, target_name)
if wt_idmethod: if wt_idmethod:
wt_idmethod.Delete_() wt_idmethod.Delete_()
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_msg = _('Could not deassociate initiator %(initiator)s from ' err_msg = _('Could not deassociate initiator %(initiator)s from '
'iSCSI target: %(target_name)s.') 'iSCSI target: %(target_name)s.')
raise exceptions.ISCSITargetWMIException( raise exceptions.ISCSITargetWMIException(
@ -217,7 +212,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
self._conn_wmi.WT_Disk.NewWTDisk(DevicePath=vhd_path, self._conn_wmi.WT_Disk.NewWTDisk(DevicePath=vhd_path,
Description=wtd_name, Description=wtd_name,
SizeInMB=size_mb) SizeInMB=size_mb)
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_msg = _('Failed to create WT Disk. ' err_msg = _('Failed to create WT Disk. '
'VHD path: %(vhd_path)s ' 'VHD path: %(vhd_path)s '
'WT disk name: %(wtd_name)s') 'WT disk name: %(wtd_name)s')
@ -231,7 +226,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
try: try:
self._conn_wmi.WT_Disk.ImportWTDisk(DevicePath=vhd_path, self._conn_wmi.WT_Disk.ImportWTDisk(DevicePath=vhd_path,
Description=wtd_name) Description=wtd_name)
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_msg = _("Failed to import WT disk: %s.") err_msg = _("Failed to import WT disk: %s.")
raise exceptions.ISCSITargetWMIException(err_msg % vhd_path, raise exceptions.ISCSITargetWMIException(err_msg % vhd_path,
wmi_exc=wmi_exc) wmi_exc=wmi_exc)
@ -241,7 +236,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
wt_disk = self._get_wt_disk(wtd_name) wt_disk = self._get_wt_disk(wtd_name)
wt_disk.Enabled = enabled wt_disk.Enabled = enabled
wt_disk.put() wt_disk.put()
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_msg = _('Could not change disk status. WT Disk name: %s') err_msg = _('Could not change disk status. WT Disk name: %s')
raise exceptions.ISCSITargetWMIException(err_msg % wtd_name, raise exceptions.ISCSITargetWMIException(err_msg % wtd_name,
wmi_exc=wmi_exc) wmi_exc=wmi_exc)
@ -251,7 +246,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
wt_disk = self._get_wt_disk(wtd_name, fail_if_not_found=False) wt_disk = self._get_wt_disk(wtd_name, fail_if_not_found=False)
if wt_disk: if wt_disk:
wt_disk.Delete_() wt_disk.Delete_()
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_msg = _("Failed to remove WT disk: %s.") err_msg = _("Failed to remove WT disk: %s.")
raise exceptions.ISCSITargetWMIException(err_msg % wtd_name, raise exceptions.ISCSITargetWMIException(err_msg % wtd_name,
wmi_exc=wmi_exc) wmi_exc=wmi_exc)
@ -260,7 +255,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
try: try:
wt_disk = self._get_wt_disk(wtd_name) wt_disk = self._get_wt_disk(wtd_name)
wt_disk.Extend(additional_mb) wt_disk.Extend(additional_mb)
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_msg = _('Could not extend WT Disk %(wtd_name)s ' err_msg = _('Could not extend WT Disk %(wtd_name)s '
'with additional %(additional_mb)s MB.') 'with additional %(additional_mb)s MB.')
raise exceptions.ISCSITargetWMIException( raise exceptions.ISCSITargetWMIException(
@ -274,7 +269,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
wt_disk = self._get_wt_disk(wtd_name) wt_disk = self._get_wt_disk(wtd_name)
wt_host = self._get_wt_host(target_name) wt_host = self._get_wt_host(target_name)
wt_host.AddWTDisk(wt_disk.WTD) wt_host.AddWTDisk(wt_disk.WTD)
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_msg = _('Could not add WTD Disk %(wtd_name)s to ' err_msg = _('Could not add WTD Disk %(wtd_name)s to '
'iSCSI target %(target_name)s.') 'iSCSI target %(target_name)s.')
raise exceptions.ISCSITargetWMIException( raise exceptions.ISCSITargetWMIException(
@ -291,7 +286,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
wt_snap = self._conn_wmi.WT_Snapshot(Id=snap_id)[0] wt_snap = self._conn_wmi.WT_Snapshot(Id=snap_id)[0]
wt_snap.Description = snapshot_name wt_snap.Description = snapshot_name
wt_snap.put() wt_snap.put()
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_msg = _('Failed to create snapshot. ' err_msg = _('Failed to create snapshot. '
'WT Disk name: %(wtd_name)s ' 'WT Disk name: %(wtd_name)s '
'Snapshot name: %(snapshot_name)s') 'Snapshot name: %(snapshot_name)s')
@ -315,7 +310,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
self._pathutils.copy(src_path, dest_path) self._pathutils.copy(src_path, dest_path)
wt_disk.Delete_() wt_disk.Delete_()
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_msg = _('Failed to export snapshot %(snapshot_name)s ' err_msg = _('Failed to export snapshot %(snapshot_name)s '
'to %(dest_path)s.') 'to %(dest_path)s.')
raise exceptions.ISCSITargetWMIException( raise exceptions.ISCSITargetWMIException(
@ -330,7 +325,7 @@ class ISCSITargetUtils(baseutils.BaseUtils):
fail_if_not_found=False) fail_if_not_found=False)
if wt_snapshot: if wt_snapshot:
wt_snapshot.Delete_() wt_snapshot.Delete_()
except wmi.x_wmi as wmi_exc: except exceptions.x_wmi as wmi_exc:
err_msg = _('Failed delete snapshot %s.') err_msg = _('Failed delete snapshot %s.')
raise exceptions.ISCSITargetWMIException(err_msg % snapshot_name, raise exceptions.ISCSITargetWMIException(err_msg % snapshot_name,
wmi_exc=wmi_exc) wmi_exc=wmi_exc)