Early import project api

gns3server/controller/__init__.py

@@ -286,6 +286,20 @@ class Controller:
             yield from project.open()
         return project
 
+    def get_free_project_name(self, base_name):
+        """
+        Generate a free project name base on the base name
+        """
+        names = [ p.name for p in self._projects.values() ]
+        if base_name not in names:
+            return base_name
+        i = 1
+        while "{}-{}".format(base_name, i) in names:
+            i += 1
+            if i > 1000000:
+                raise aiohttp.web.HTTPConflict(text="A project name could not be allocated (node limit reached?)")
+        return "{}-{}".format(base_name, i)
+
     @property
     def projects(self):
         """

gns3server/controller/import_project.py

@@ -0,0 +1,69 @@
+#!/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 os
+import json
+import uuid
+import asyncio
+import zipfile
+import aiohttp
+
+from ..config import Config
+
+
+"""
+Handle the import of project from a .gns3project
+"""
+
+@asyncio.coroutine
+def import_project(controller, project_id, stream, gns3vm=True):
+    """
+    Import a project contain in a zip file
+
+    You need to handle OSerror exceptions
+
+    :param stream: A io.BytesIO of the zipfile
+    :param gns3vm: True move Docker, IOU and Qemu to the GNS3 VM
+    :returns: Project
+    """
+    server_config = Config.instance().get_section_config("Server")
+    projects_path = os.path.expanduser(server_config.get("projects_path", "~/GNS3/projects"))
+    os.makedirs(projects_path, exist_ok=True)
+
+    with zipfile.ZipFile(stream) as myzip:
+
+        try:
+            topology = json.loads(myzip.read("project.gns3").decode())
+            # If the project name is already used we generate a new one
+            topology["name"] = controller.get_free_project_name(topology["name"])
+        except KeyError:
+            raise aiohttp.web.HTTPConflict(text="Can't import topology the .gns3 is corrupted or missing")
+
+        path = os.path.join(projects_path, topology["name"])
+        os.makedirs(path)
+        myzip.extractall(path)
+
+        dot_gns3_path = os.path.join(path, topology["name"] + ".gns3")
+        # We change the project_id to avoid erasing the project
+        topology["project_id"] = project_id
+        with open(dot_gns3_path, "w+") as f:
+            json.dump(topology, f, indent=4)
+        os.remove(os.path.join(path, "project.gns3"))
+
+    project = yield from controller.load_project(dot_gns3_path)
+    return project
+