Replace snapshots by import / export

Fix #1042

gns3server/controller/snapshot.py

@@ -0,0 +1,93 @@
+#!/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 uuid
+import shutil
+import asyncio
+from datetime import datetime, timezone
+
+
+from .import_project import import_project
+
+
+# The string use to extract the date from the filename
+FILENAME_TIME_FORMAT = "%d%m%y_%H%M%S"
+
+
+class Snapshot:
+    """
+    A snapshot object
+    """
+
+    def __init__(self, project, name=None, filename=None):
+
+        assert filename or name, "You need to pass a name or a filename"
+
+        self._id = str(uuid.uuid4())  # We don't need to keep id between project loading because they are use only as key for operation like delete, update.. but have no impact on disk
+        self._project = project
+        if name:
+            self._name = name
+            self._created_at = datetime.now().timestamp()
+            filename = self._name + "_" + datetime.utcfromtimestamp(self._created_at).replace(tzinfo=None).strftime(FILENAME_TIME_FORMAT) + ".gns3project"
+        else:
+            self._name = filename.split("_")[0]
+            datestring = filename.replace(self._name + "_", "").split(".")[0]
+            try:
+                self._created_at = datetime.strptime(datestring, FILENAME_TIME_FORMAT).replace(tzinfo=timezone.utc).timestamp()
+            except ValueError:
+                self._created_at = datetime.utcnow().timestamp()
+        self._path = os.path.join(project.path, "snapshots", filename)
+
+    @property
+    def id(self):
+        return self._id
+
+    @property
+    def name(self):
+        return self._name
+
+    @property
+    def path(self):
+        return self._path
+
+    @property
+    def created_at(self):
+        return int(self._created_at)
+
+    @asyncio.coroutine
+    def restore(self):
+        """
+        Restore the snapshot
+        """
+
+        yield from self._project.delete_on_computes()
+        yield from self._project.close()
+        shutil.rmtree(os.path.join(self._project.path, "project-files"))
+        with open(self._path, "rb") as f:
+            project = yield from import_project(self._project.controller, self._project.id, f, location=self._project.path)
+        yield from project.open()
+        return project
+
+    def __json__(self):
+        return {
+            "snapshot_id": self._id,
+            "name": self._name,
+            "created_at": int(self._created_at),
+            "project_id": self._project.id
+        }