Dirty stop start for VPCS

gns3server/modules/base_manager.py

@@ -0,0 +1,89 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2015 GNS3 Technologies Inc.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+
+import asyncio
+import aiohttp
+
+from .device_error import DeviceError
+
+
+class BaseManager:
+    """
+    Base class for all Manager.
+    Responsible of management of a VM pool
+    """
+
+    def __init__(self):
+        self._vms = {}
+
+    @classmethod
+    def instance(cls):
+        """
+        Singleton to return only one instance of Manager.
+
+        :returns: instance of Manager
+        """
+
+        if not hasattr(cls, "_instance"):
+            cls._instance = cls()
+        return cls._instance
+
+    @classmethod
+    @asyncio.coroutine
+    def destroy(cls):
+        cls._instance = None
+
+    def _get_vm_instance(self, vm_id):
+        """
+        Returns a VM instance.
+
+        :param vm_id: VM identifier
+
+        :returns: VM instance
+        """
+
+        if vm_id not in self._vms:
+            raise aiohttp.web.HTTPNotFound(text="ID {} doesn't exist".format(vm_id))
+        return self._vms[vm_id]
+
+    @asyncio.coroutine
+    def create_vm(self, vmname, identifier=None):
+        if not identifier:
+            for i in range(1, 1024):
+                if i not in self._vms:
+                    identifier = i
+                    break
+            if identifier == 0:
+                raise DeviceError("Maximum number of VM instances reached")
+        else:
+            if identifier in self._vms:
+                raise DeviceError("VM identifier {} is already used by another VM instance".format(identifier))
+        vm = self._VM_CLASS(vmname, identifier)
+        yield from vm.wait_for_creation()
+        self._vms[vm.id] = vm
+        return vm
+
+    @asyncio.coroutine
+    def start_vm(self, vm_id):
+        vm = self._get_vm_instance(vm_id)
+        yield from vm.start()
+
+    @asyncio.coroutine
+    def stop_vm(self, vm_id):
+        vm = self._get_vm_instance(vm_id)
+        yield from vm.stop()