1
0
mirror of https://github.com/GNS3/gns3-server synced 2024-11-13 20:08:55 +00:00

Cleanup old temporary project at startup

This commit is contained in:
Julien Duponchelle 2015-02-04 17:18:53 +01:00
parent ca354ae7f2
commit 2ace014a3c
3 changed files with 70 additions and 3 deletions

View File

@ -26,6 +26,8 @@ from gns3server.server import Server
from gns3server.web.logger import init_logger
from gns3server.version import __version__
from gns3server.config import Config
from gns3server.modules.project import Project
import logging
log = logging.getLogger(__name__)
@ -140,6 +142,8 @@ def main():
log.critical("The current working directory doesn't exist")
return
Project.clean_project_directory()
host = server_config["host"]
port = int(server_config["port"])
server = Server(host, port)

View File

@ -58,7 +58,6 @@ class Project:
if config.get("local", False) is False:
raise aiohttp.web.HTTPForbidden(text="You are not allowed to modifiy the project directory location")
self._temporary = temporary
self._vms = set()
self._vms_to_destroy = set()
self._path = os.path.join(self._location, self._uuid)
@ -66,9 +65,11 @@ class Project:
os.makedirs(os.path.join(self._path, "vms"), exist_ok=True)
except OSError as e:
raise aiohttp.web.HTTPInternalServerError(text="Could not create project directory: {}".format(e))
self.temporary = temporary
log.debug("Create project {uuid} in directory {path}".format(path=self._path, uuid=self._uuid))
def _get_default_project_directory(self):
@classmethod
def _get_default_project_directory(cls):
"""
Return the default location for the project directory
depending of the operating system
@ -109,8 +110,21 @@ class Project:
@temporary.setter
def temporary(self, temporary):
if hasattr(self, 'temporary') and temporary == self._temporary:
return
self._temporary = temporary
if self._temporary:
try:
with open(os.path.join(self._path, ".gns3_temporary"), 'w+') as f:
f.write("1")
except OSError as e:
raise aiohttp.web.HTTPInternalServerError(text="Could not create temporary project: {}".format(e))
else:
if os.path.exists(os.path.join(self._path, ".gns3_temporary")):
os.remove(os.path.join(self._path, ".gns3_temporary"))
def vm_working_directory(self, vm):
"""
Return a working directory for a specific VM.
@ -222,3 +236,16 @@ class Project:
"""Remove project from disk"""
yield from self._close_and_clean(True)
@classmethod
def clean_project_directory(cls):
"""At startup drop old temporary project. After a crash for example"""
config = Config.instance().get_section_config("Server")
directory = config.get("project_directory", cls._get_default_project_directory())
if os.path.exists(directory):
for project in os.listdir(directory):
path = os.path.join(directory, project)
if os.path.exists(os.path.join(path, ".gns3_temporary")):
log.warning("Purge old temporary project {}".format(project))
shutil.rmtree(path)

View File

@ -21,6 +21,7 @@ import asyncio
import pytest
import aiohttp
import shutil
from uuid import uuid4
from unittest.mock import patch
from gns3server.modules.project import Project
@ -53,11 +54,21 @@ def test_path(tmpdir):
assert p.path == os.path.join(str(tmpdir), p.uuid)
assert os.path.exists(os.path.join(str(tmpdir), p.uuid))
assert os.path.exists(os.path.join(str(tmpdir), p.uuid, 'vms'))
assert not os.path.exists(os.path.join(p.path, '.gns3_temporary'))
def test_temporary_path():
p = Project()
p = Project(temporary=True)
assert os.path.exists(p.path)
assert os.path.exists(os.path.join(p.path, '.gns3_temporary'))
def test_remove_temporary_flag():
p = Project(temporary=True)
assert os.path.exists(p.path)
assert os.path.exists(os.path.join(p.path, '.gns3_temporary'))
p.temporary = False
assert not os.path.exists(os.path.join(p.path, '.gns3_temporary'))
def test_changing_location_not_allowed(tmpdir):
@ -164,3 +175,28 @@ def test_get_default_project_directory():
path = os.path.normpath(os.path.expanduser("~/GNS3/projects"))
assert project._get_default_project_directory() == path
assert os.path.exists(path)
def test_clean_project_directory(tmpdir):
# A non anonymous project with uuid.
project1 = tmpdir / uuid4()
project1.mkdir()
# A non anonymous project.
oldproject = tmpdir / uuid4()
oldproject.mkdir()
# an anonymous project
project2 = tmpdir / uuid4()
project2.mkdir()
tmp = (project2 / ".gns3_temporary")
with open(str(tmp), 'w+') as f:
f.write("1")
with patch("gns3server.config.Config.get_section_config", return_value={"project_directory": str(tmpdir)}):
Project.clean_project_directory()
assert os.path.exists(str(project1))
assert os.path.exists(str(oldproject))
assert not os.path.exists(str(project2))