forked from p34709852/monkey
Fix missing WindowsError on linux
This commit is contained in:
parent
6233fec0f7
commit
2d83657bd9
|
@ -19,6 +19,12 @@ if "win32" == sys.platform:
|
||||||
else:
|
else:
|
||||||
DETACHED_PROCESS = 0
|
DETACHED_PROCESS = 0
|
||||||
|
|
||||||
|
# Linux doesn't have WindowsError
|
||||||
|
try:
|
||||||
|
WindowsError
|
||||||
|
except NameError:
|
||||||
|
WindowsError = None
|
||||||
|
|
||||||
__author__ = 'itamar'
|
__author__ = 'itamar'
|
||||||
|
|
||||||
LOG = logging.getLogger(__name__)
|
LOG = logging.getLogger(__name__)
|
||||||
|
@ -62,7 +68,7 @@ class MonkeyDrops(object):
|
||||||
self._config['source_path'], self._config['destination_path'])
|
self._config['source_path'], self._config['destination_path'])
|
||||||
|
|
||||||
file_moved = True
|
file_moved = True
|
||||||
except (WindowsError, IOError, OSError), exc:
|
except (WindowsError, IOError, OSError) as exc:
|
||||||
LOG.debug("Error moving source file '%s' into '%s': %s",
|
LOG.debug("Error moving source file '%s' into '%s': %s",
|
||||||
self._config['source_path'], self._config['destination_path'],
|
self._config['source_path'], self._config['destination_path'],
|
||||||
exc)
|
exc)
|
||||||
|
@ -75,7 +81,7 @@ class MonkeyDrops(object):
|
||||||
|
|
||||||
LOG.info("Copied source file '%s' into '%s'",
|
LOG.info("Copied source file '%s' into '%s'",
|
||||||
self._config['source_path'], self._config['destination_path'])
|
self._config['source_path'], self._config['destination_path'])
|
||||||
except (WindowsError, IOError, OSError), exc:
|
except (WindowsError, IOError, OSError) as exc:
|
||||||
LOG.error("Error copying source file '%s' into '%s': %s",
|
LOG.error("Error copying source file '%s' into '%s': %s",
|
||||||
self._config['source_path'], self._config['destination_path'],
|
self._config['source_path'], self._config['destination_path'],
|
||||||
exc)
|
exc)
|
||||||
|
@ -131,7 +137,7 @@ class MonkeyDrops(object):
|
||||||
# try removing the file first
|
# try removing the file first
|
||||||
try:
|
try:
|
||||||
os.remove(self._config['source_path'])
|
os.remove(self._config['source_path'])
|
||||||
except Exception, exc:
|
except Exception as exc:
|
||||||
LOG.debug("Error removing source file '%s': %s", self._config['source_path'], exc)
|
LOG.debug("Error removing source file '%s': %s", self._config['source_path'], exc)
|
||||||
|
|
||||||
# mark the file for removal on next boot
|
# mark the file for removal on next boot
|
||||||
|
|
|
@ -6,6 +6,12 @@ from enum import IntEnum
|
||||||
|
|
||||||
from network.info import get_host_subnets
|
from network.info import get_host_subnets
|
||||||
|
|
||||||
|
# Linux doesn't have WindowsError
|
||||||
|
try:
|
||||||
|
WindowsError
|
||||||
|
except NameError:
|
||||||
|
WindowsError = None
|
||||||
|
|
||||||
__author__ = 'uri'
|
__author__ = 'uri'
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue