lib/python/Plugins/SystemPlugins/SoftwareManager/BackupRestore.py

   1 from Screens.Screen import Screen
   2 from Screens.MessageBox import MessageBox
   3 from Screens.Console import Console
   4 from Components.ActionMap import ActionMap, NumberActionMap
   5 from Components.Pixmap import Pixmap
   6 from Components.Label import Label
   7 from Components.MenuList import MenuList
   8 from Components.config import getConfigListEntry, configfile, ConfigSelection, ConfigSubsection, ConfigText, ConfigLocations
   9 from Components.config import config
  10 from Components.ConfigList import ConfigList,ConfigListScreen
  11 from Components.FileList import MultiFileSelectList
  12 from Plugins.Plugin import PluginDescriptor
  13 from enigma import eTimer
  14 from Tools.Directories import *
  15 from os import popen, path, makedirs, listdir, access, stat, rename, remove, W_OK, R_OK
  16 from time import gmtime, strftime, localtime
  17 from datetime import date
  18
  19
  20 config.plugins.configurationbackup = ConfigSubsection()
  21 config.plugins.configurationbackup.backuplocation = ConfigText(default = '/media/hdd/', visible_width = 50, fixed_size = False)
  22 config.plugins.configurationbackup.backupdirs = ConfigLocations(default=['/etc/enigma2/', '/etc/network/interfaces', '/etc/wpa_supplicant.conf'])
  23
  24 def getBackupPath():
  25         backuppath = config.plugins.configurationbackup.backuplocation.value
  26         if backuppath.endswith('/'):
  27                 return backuppath + 'backup'
  28         else:
  29                 return backuppath + '/backup'
  30
  31 def getBackupFilename():
  32         return "enigma2settingsbackup.tar.gz"
  33
  34
  35 class BackupScreen(Screen, ConfigListScreen):
  36         skin = """
  37                 <screen position="135,144" size="350,310" title="Backup running..." >
  38                 <widget name="config" position="10,10" size="330,250" transparent="1" scrollbarMode="showOnDemand" />
  39                 </screen>"""
  40
  41         def __init__(self, session, runBackup = False):
  42                 Screen.__init__(self, session)
  43                 self.session = session
  44                 self.runBackup = runBackup
  45                 self["actions"] = ActionMap(["WizardActions", "DirectionActions"],
  46                 {
  47                         "ok": self.close,
  48                         "back": self.close,
  49                         "cancel": self.close,
  50                 }, -1)
  51                 self.finished_cb = None
  52                 self.backuppath = getBackupPath()
  53                 self.backupfile = getBackupFilename()
  54                 self.fullbackupfilename = self.backuppath + "/" + self.backupfile
  55                 self.list = []
  56                 ConfigListScreen.__init__(self, self.list)
  57                 self.onLayoutFinish.append(self.layoutFinished)
  58                 if self.runBackup:
  59                         self.onShown.append(self.doBackup)
  60
  61         def layoutFinished(self):
  62                 self.setWindowTitle()
  63
  64         def setWindowTitle(self):
  65                 self.setTitle(_("Backup running..."))
  66
  67         def doBackup(self):
  68                 try:
  69                         if (path.exists(self.backuppath) == False):
  70                                 makedirs(self.backuppath)
  71                         self.backupdirs = ' '.join( config.plugins.configurationbackup.backupdirs.value )
  72                         if path.exists(self.fullbackupfilename):
  73                                 dt = str(date.fromtimestamp(stat(self.fullbackupfilename).st_ctime))
  74                                 self.newfilename = self.backuppath + "/" + dt + '-' + self.backupfile
  75                                 if path.exists(self.newfilename):
  76                                         remove(self.newfilename)
  77                                 rename(self.fullbackupfilename,self.newfilename)
  78                         if self.finished_cb:
  79                                 self.session.openWithCallback(self.finished_cb, Console, title = _("Backup running"), cmdlist = ["tar -czvf " + self.fullbackupfilename + " " + self.backupdirs],finishedCallback = self.backupFinishedCB,closeOnSuccess = True)
  80                         else:
  81                                 self.session.open(Console, title = _("Backup running"), cmdlist = ["tar -czvf " + self.fullbackupfilename + " " + self.backupdirs],finishedCallback = self.backupFinishedCB, closeOnSuccess = True)
  82                 except OSError:
  83                         if self.finished_cb:
  84                                 self.session.openWithCallback(self.finished_cb, MessageBox, _("Sorry your backup destination is not writeable.\nPlease choose an other one."), MessageBox.TYPE_INFO)
  85                         else:
  86                                 self.session.openWithCallback(self.backupErrorCB,MessageBox, _("Sorry your backup destination is not writeable.\nPlease choose an other one."), MessageBox.TYPE_INFO)
  87
  88         def backupFinishedCB(self,retval = None):
  89                 self.close(True)
  90
  91         def backupErrorCB(self,retval = None):
  92                 self.close(False)
  93
  94         def runAsync(self, finished_cb):
  95                 self.finished_cb = finished_cb
  96                 self.doBackup()
  97
  98
  99 class BackupSelection(Screen):
 100         skin = """
 101         <screen position="135,125" size="450,310" title="Select files/folders to backup...">
 102                 <widget name="checkList" position="10,10" size="430,250" transparent="1" scrollbarMode="showOnDemand" />
 103                 <ePixmap position="0,265" zPosition="1" size="135,40" pixmap="skin_default/buttons/red.png" transparent="1" alphatest="on" />
 104                 <widget name="key_red" position="0,265" zPosition="2" size="135,40" halign="center" valign="center" font="Regular;22" transparent="1" shadowColor="black" shadowOffset="-1,-1" />
 105                 <ePixmap position="135,265" zPosition="1" size="135,40" pixmap="skin_default/buttons/green.png" transparent="1" alphatest="on" />
 106                 <widget name="key_green" position="135,265" zPosition="2" size="135,40" halign="center" valign="center" font="Regular;22" transparent="1" shadowColor="black" shadowOffset="-1,-1" />
 107                 <ePixmap position="270,265" zPosition="1" size="135,40" pixmap="skin_default/buttons/yellow.png" transparent="1" alphatest="on" />
 108                 <widget name="key_yellow" position="270,265" zPosition="2" size="135,40" halign="center" valign="center" font="Regular;22" transparent="1" shadowColor="black" shadowOffset="-1,-1" />
 109         </screen>"""
 110
 111         def __init__(self, session):
 112                 Screen.__init__(self, session)
 113                 self["key_red"] = Label(_("Cancel"))
 114                 self["key_green"] = Label(_("Save"))
 115                 self["key_yellow"] = Label()
 116
 117                 self.selectedFiles = config.plugins.configurationbackup.backupdirs.value
 118                 defaultDir = '/'
 119                 inhibitDirs = ["/bin", "/boot", "/dev", "/autofs", "/lib", "/proc", "/sbin", "/sys", "/hdd", "/tmp", "/mnt", "/media"]
 120                 self.filelist = MultiFileSelectList(self.selectedFiles, defaultDir, inhibitDirs = inhibitDirs )
 121                 self["checkList"] = self.filelist
 122
 123                 self["actions"] = ActionMap(["DirectionActions", "OkCancelActions", "ShortcutActions"],
 124                 {
 125                         "cancel": self.exit,
 126                         "red": self.exit,
 127                         "yellow": self.changeSelectionState,
 128                         "green": self.saveSelection,
 129                         "ok": self.okClicked,
 130                         "left": self.left,
 131                         "right": self.right,
 132                         "down": self.down,
 133                         "up": self.up
 134                 }, -1)
 135                 if not self.selectionChanged in self["checkList"].onSelectionChanged:
 136                         self["checkList"].onSelectionChanged.append(self.selectionChanged)
 137                 self.onLayoutFinish.append(self.layoutFinished)
 138
 139         def layoutFinished(self):
 140                 idx = 0
 141                 self["checkList"].moveToIndex(idx)
 142                 self.setWindowTitle()
 143                 self.selectionChanged()
 144
 145         def setWindowTitle(self):
 146                 self.setTitle(_("Select files/folders to backup..."))
 147
 148         def selectionChanged(self):
 149                 current = self["checkList"].getCurrent()[0]
 150                 if current[2] is True:
 151                         self["key_yellow"].setText(_("Deselect"))
 152                 else:
 153                         self["key_yellow"].setText(_("Select"))
 154
 155         def up(self):
 156                 self["checkList"].up()
 157
 158         def down(self):
 159                 self["checkList"].down()
 160
 161         def left(self):
 162                 self["checkList"].pageUp()
 163
 164         def right(self):
 165                 self["checkList"].pageDown()
 166
 167         def changeSelectionState(self):
 168                 self["checkList"].changeSelectionState()
 169                 self.selectedFiles = self["checkList"].getSelectedList()
 170
 171         def saveSelection(self):
 172                 self.selectedFiles = self["checkList"].getSelectedList()
 173                 config.plugins.configurationbackup.backupdirs.value = self.selectedFiles
 174                 config.plugins.configurationbackup.backupdirs.save()
 175                 config.plugins.configurationbackup.save()
 176                 config.save()
 177                 self.close(None)
 178
 179         def exit(self):
 180                 self.close(None)
 181
 182         def okClicked(self):
 183                 if self.filelist.canDescent():
 184                         self.filelist.descent()
 185
 186
 187 class RestoreMenu(Screen):
 188         skin = """
 189                 <screen position="135,144" size="450,300" title="Restore backups..." >
 190                 <widget name="filelist" position="10,10" size="430,240" scrollbarMode="showOnDemand" />
 191                 <widget name="cancel" position="120,255" size="100,40" pixmap="~/red.png" transparent="1" alphatest="on" />
 192                 <widget name="canceltext" position="0,0" size="0,0" valign="center" halign="center" zPosition="2" font="Regular;20" transparent="1" foregroundColor="black" />
 193                 <widget name="restore" position="230,255" size="100,40" pixmap="~/yellow.png" transparent="1" alphatest="on" />
 194                 <widget name="restoretext" position="0,0" size="0,0" valign="center" halign="center" zPosition="2" font="Regular;20" transparent="1"  foregroundColor="black" />
 195                 </screen>"""
 196
 197         def __init__(self, session, plugin_path):
 198                 Screen.__init__(self, session)
 199                 self.skin_path = plugin_path
 200
 201                 self["canceltext"] = Label(_("Cancel"))
 202                 self["restoretext"] = Label(_("Restore"))
 203                 self["restore"] = Pixmap()
 204                 self["cancel"] = Pixmap()
 205
 206                 self.sel = []
 207                 self.val = []
 208                 self.entry = False
 209                 self.exe = False
 210
 211                 self.path = ""
 212
 213                 self["actions"] = NumberActionMap(["SetupActions"],
 214                 {
 215                         "ok": self.KeyOk,
 216                         "cancel": self.keyCancel
 217                 }, -1)
 218
 219                 self["shortcuts"] = ActionMap(["ShortcutActions"],
 220                 {
 221                         "red": self.keyCancel,
 222                         "yellow": self.KeyOk,
 223                 })
 224                 self.flist = []
 225                 self["filelist"] = MenuList(self.flist)
 226                 self.fill_list()
 227                 self.onLayoutFinish.append(self.layoutFinished)
 228
 229         def layoutFinished(self):
 230                 self.setWindowTitle()
 231
 232         def setWindowTitle(self):
 233                 self.setTitle(_("Restore backups..."))
 234
 235
 236         def fill_list(self):
 237                 self.flist = []
 238                 self.path = getBackupPath()
 239                 if (path.exists(self.path) == False):
 240                         makedirs(self.path)
 241                 for file in listdir(self.path):
 242                         if (file.endswith(".tar.gz")):
 243                                 self.flist.append((file))
 244                                 self.entry = True
 245                                 self["filelist"].l.setList(self.flist)
 246
 247         def KeyOk(self):
 248                 if (self.exe == False) and (self.entry == True):
 249                         self.sel = self["filelist"].getCurrent()
 250                         self.val = self.path + self.sel
 251                         self.session.openWithCallback(self.startRestore, MessageBox, _("Are you sure you want to restore\nfollowing backup:\n" + self.sel + "\nSystem will restart after the restore!"))
 252
 253         def keyCancel(self):
 254                 self.close()
 255
 256         def startRestore(self, ret = False):
 257                 if (ret == True):
 258                         self.exe = True
 259                         self.session.open(Console, title = _("Restore running"), cmdlist = ["tar -xzvf " + self.path + "/" + self.sel + " -C /", "killall -9 enigma2"])
 260
 261         def Exit(self):
 262                 self.close()
 263
 264 class RestoreScreen(Screen, ConfigListScreen):
 265         skin = """
 266                 <screen position="135,144" size="350,310" title="Restore running..." >
 267                 <widget name="config" position="10,10" size="330,250" transparent="1" scrollbarMode="showOnDemand" />
 268                 </screen>"""
 269
 270         def __init__(self, session, runRestore = False):
 271                 Screen.__init__(self, session)
 272                 self.session = session
 273                 self.runRestore = runRestore
 274                 self["actions"] = ActionMap(["WizardActions", "DirectionActions"],
 275                 {
 276                         "ok": self.close,
 277                         "back": self.close,
 278                         "cancel": self.close,
 279                 }, -1)
 280                 self.finished_cb = None
 281                 self.backuppath = getBackupPath()
 282                 self.backupfile = getBackupFilename()
 283                 self.fullbackupfilename = self.backuppath + "/" + self.backupfile
 284                 self.list = []
 285                 ConfigListScreen.__init__(self, self.list)
 286                 self.onLayoutFinish.append(self.layoutFinished)
 287                 if self.runRestore:
 288                         self.onShown.append(self.doRestore)
 289
 290         def layoutFinished(self):
 291                 self.setWindowTitle()
 292
 293         def setWindowTitle(self):
 294                 self.setTitle(_("Restore running..."))
 295
 296         def doRestore(self):
 297                 if self.finished_cb:
 298                         self.session.openWithCallback(self.finished_cb, Console, title = _("Restore running"), cmdlist = ["tar -xzvf " + self.fullbackupfilename + " -C /", "killall -9 enigma2"])
 299                 else:
 300                         self.session.open(Console, title = _("Restore running"), cmdlist = ["tar -xzvf " + self.fullbackupfilename + " -C /", "killall -9 enigma2"])
 301
 302         def backupFinishedCB(self,retval = None):
 303                 self.close(True)
 304
 305         def backupErrorCB(self,retval = None):
 306                 self.close(False)
 307
 308         def runAsync(self, finished_cb):
 309                 self.finished_cb = finished_cb
 310                 self.doRestore()
 311
 312