Update hypervisors working dir when the project is moved.

This commit is contained in:
Jeremy committed 2015-02-27 16:51:17 -07:00
1 parent b55719186e
commit 0e8b8fa66f
4 files changed
+40 -10

No files matched your search

+21 -5
View File
@@ -133,17 +133,18 @@ class Dynamips(BaseManager):
continue
@asyncio.coroutine
def project_closed(self, project_dir):
def project_closed(self, project):
"""
Called when a project is closed.
:param project_dir: project directory
:param project: Project instance
"""
# delete the Dynamips devices
# delete the Dynamips devices corresponding to the project
tasks = []
for device in self._devices.values():
tasks.append(asyncio.async(device.delete()))
if device.project.id == project.id:
tasks.append(asyncio.async(device.delete()))
if tasks:
done, _ = yield from asyncio.wait(tasks)
@@ -154,7 +155,7 @@ class Dynamips(BaseManager):
log.error("Could not delete device {}".format(e), exc_info=1)
# delete useless files
project_dir = os.path.join(project_dir, 'project-files', self.module_name.lower())
project_dir = project.module_working_directory(self.module_name.lower())
files = glob.glob(os.path.join(project_dir, "*.ghost"))
files += glob.glob(os.path.join(project_dir, "*_lock"))
files += glob.glob(os.path.join(project_dir, "ilt_*"))
@@ -170,6 +171,21 @@ class Dynamips(BaseManager):
log.warn("Could not delete file {}: {}".format(file, e))
continue
@asyncio.coroutine
def project_moved(self, project):
"""
Called when a project is moved.
:param project: Project instance
"""
for vm in project.vms:
yield from vm.hypervisor.set_working_dir(project.module_working_directory(self.module_name.lower()))
for device in self._devices.values():
if device.project.id == project.id:
yield from device.hypervisor.set_working_dir(project.module_working_directory(self.module_name.lower()))
@property
def dynamips_path(self):
"""
@@ -146,7 +146,7 @@ class DynamipsHypervisor:
"""
# encase working_dir in quotes to protect spaces in the path
yield from self.send("hypervisor working_dir {}".format('"' + working_dir + '"'))
yield from self.send('hypervisor working_dir "{}"'.format(working_dir))
self._working_dir = working_dir
log.debug("Working directory set to {}".format(self._working_dir))