2016-07-26 08:32:43 +00:00
#!/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 ( )
2016-08-15 11:30:02 +00:00
# We don't send close notif to clients because the close / open dance is purely internal
yield from self . _project . close ( ignore_notification = True )
2016-08-19 15:05:28 +00:00
self . _project . controller . notification . emit ( " snapshot.restored " , self . __json__ ( ) )
2016-07-26 08:32:43 +00:00
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
}