1
0
mirror of https://github.com/GNS3/gns3-server synced 2024-11-28 11:18:11 +00:00
gns3-server/gns3server/controller/project.py
2016-03-10 21:51:29 +01:00

107 lines
3.1 KiB
Python

#!/usr/bin/env python
#
# Copyright (C) 2016 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
from uuid import UUID, uuid4
from .vm import VM
class Project:
"""
A project inside controller
:param project_id: force project identifier (None by default auto generate an UUID)
:param path: path of the project. (None use the standard directory)
:param temporary: boolean to tell if the project is a temporary project (destroy when closed)
"""
def __init__(self, name=None, project_id=None, path=None, temporary=False):
self._name = name
if project_id is None:
self._id = str(uuid4())
else:
try:
UUID(project_id, version=4)
except ValueError:
raise aiohttp.web.HTTPBadRequest(text="{} is not a valid UUID".format(project_id))
self._id = project_id
self._path = path
self._temporary = temporary
self._hypervisors = set()
self._vms = {}
@property
def name(self):
return self._name
@property
def id(self):
return self._id
@property
def temporary(self):
return self._temporary
@property
def path(self):
return self._path
@asyncio.coroutine
def addHypervisor(self, hypervisor):
self._hypervisors.add(hypervisor)
yield from hypervisor.post("/projects", self)
@asyncio.coroutine
def addVM(self, hypervisor, vm_id, **kwargs):
"""
Create a vm or return an existing vm
:param kwargs: See the documentation of VM
"""
if vm_id not in self._vms:
vm = VM(self, hypervisor, vm_id=vm_id, **kwargs)
yield from vm.create()
self._vms[vm.id] = vm
return vm
return self._vms[vm_id]
@asyncio.coroutine
def close(self):
for hypervisor in self._hypervisors:
yield from hypervisor.post("/projects/{}/close".format(self._id))
@asyncio.coroutine
def commit(self):
for hypervisor in self._hypervisors:
yield from hypervisor.post("/projects/{}/commit".format(self._id))
@asyncio.coroutine
def delete(self):
for hypervisor in self._hypervisors:
yield from hypervisor.delete("/projects/{}".format(self._id))
def __json__(self):
return {
"name": self._name,
"project_id": self._id,
"temporary": self._temporary,
"path": self._path
}