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.skin_path = plugin_path
 114                 self["key_red"] = Label(_("Cancel"))
 115                 self["key_green"] = Label(_("Save"))
 116                 self["key_yellow"] = Label()
 117
 118                 self.selectedFiles = config.plugins.configurationbackup.backupdirs.value
 119                 defaultDir = '/'
 120                 inhibitDirs = ["/bin", "/boot", "/dev", "/autofs", "/lib", "/proc", "/sbin", "/sys", "/hdd", "/tmp", "/mnt", "/media"]
 121                 self.filelist = MultiFileSelectList(self.selectedFiles, defaultDir, inhibitDirs = inhibitDirs )
 122                 self["checkList"] = self.filelist
 123
 124                 self["actions"] = ActionMap(["DirectionActions", "OkCancelActions", "ShortcutActions"],
 125                 {
 126                         "cancel": self.exit,
 127                         "red": self.exit,
 128                         "yellow": self.changeSelectionState,
 129                         "green": self.saveSelection,
 130                         "ok": self.okClicked,
 131                         "left": self.left,
 132                         "right": self.right,
 133                         "down": self.down,
 134                         "up": self.up
 135                 }, -1)
 136                 if not self.selectionChanged in self["checkList"].onSelectionChanged:
 137                         self["checkList"].onSelectionChanged.append(self.selectionChanged)
 138                 self.onLayoutFinish.append(self.layoutFinished)
 139
 140         def layoutFinished(self):
 141                 idx = 0
 142                 self["checkList"].moveToIndex(idx)
 143                 self.setWindowTitle()
 144                 self.selectionChanged()
 145
 146         def setWindowTitle(self):
 147                 self.setTitle(_("Select files/folders to backup..."))
 148
 149         def selectionChanged(self):
 150                 current = self["checkList"].getCurrent()[0]
 151                 if current[2] is True:
 152                         self["key_yellow"].setText(_("Deselect"))
 153                 else:
 154                         self["key_yellow"].setText(_("Select"))
 155
 156         def up(self):
 157                 self["checkList"].up()
 158
 159         def down(self):
 160                 self["checkList"].down()
 161
 162         def left(self):
 163                 self["checkList"].pageUp()
 164
 165         def right(self):
 166                 self["checkList"].pageDown()
 167
 168         def changeSelectionState(self):
 169                 self["checkList"].changeSelectionState()
 170                 self.selectedFiles = self["checkList"].getSelectedList()
 171
 172         def saveSelection(self):
 173                 self.selectedFiles = self["checkList"].getSelectedList()
 174                 config.plugins.configurationbackup.backupdirs.value = self.selectedFiles
 175                 config.plugins.configurationbackup.backupdirs.save()
 176                 config.plugins.configurationbackup.save()
 177                 config.save()
 178                 self.close(None)
 179
 180         def exit(self):
 181                 self.close(None)
 182
 183         def okClicked(self):
 184                 if self.filelist.canDescent():
 185                         self.filelist.descent()
 186
 187
 188 class RestoreMenu(Screen):
 189         skin = """
 190                 <screen position="135,144" size="450,300" title="Restore backups..." >
 191                 <widget name="filelist" position="10,10" size="430,240" scrollbarMode="showOnDemand" />
 192                 <widget name="cancel" position="120,255" size="100,40" pixmap="~/red.png" transparent="1" alphatest="on" />
 193                 <widget name="canceltext" position="0,0" size="0,0" valign="center" halign="center" zPosition="2" font="Regular;20" transparent="1" foregroundColor="black" />
 194                 <widget name="restore" position="230,255" size="100,40" pixmap="~/yellow.png" transparent="1" alphatest="on" />
 195                 <widget name="restoretext" position="0,0" size="0,0" valign="center" halign="center" zPosition="2" font="Regular;20" transparent="1"  foregroundColor="black" />
 196                 </screen>"""
 197
 198         def __init__(self, session, plugin_path):
 199                 Screen.__init__(self, session)
 200                 self.skin_path = plugin_path
 201
 202                 self["canceltext"] = Label(_("Cancel"))
 203                 self["restoretext"] = Label(_("Restore"))
 204                 self["restore"] = Pixmap()
 205                 self["cancel"] = Pixmap()
 206
 207                 self.sel = []
 208                 self.val = []
 209                 self.entry = False
 210                 self.exe = False
 211
 212                 self.path = ""
 213
 214                 self["actions"] = NumberActionMap(["SetupActions"],
 215                 {
 216                         "ok": self.KeyOk,
 217                         "cancel": self.keyCancel
 218                 }, -1)
 219
 220                 self["shortcuts"] = ActionMap(["ShortcutActions"],
 221                 {
 222                         "red": self.keyCancel,
 223                         "yellow": self.KeyOk,
 224                 })
 225                 self.flist = []
 226                 self["filelist"] = MenuList(self.flist)
 227                 self.fill_list()
 228                 self.onLayoutFinish.append(self.layoutFinished)
 229
 230         def layoutFinished(self):
 231                 self.setWindowTitle()
 232
 233         def setWindowTitle(self):
 234                 self.setTitle(_("Restore backups..."))
 235
 236
 237         def fill_list(self):
 238                 self.flist = []
 239                 self.path = getBackupPath()
 240                 if (path.exists(self.path) == False):
 241                         makedirs(self.path)
 242                 for file in listdir(self.path):
 243                         if (file.endswith(".tar.gz")):
 244                                 self.flist.append((file))
 245                                 self.entry = True
 246                                 self["filelist"].l.setList(self.flist)
 247
 248         def KeyOk(self):
 249                 if (self.exe == False) and (self.entry == True):
 250                         self.sel = self["filelist"].getCurrent()
 251                         self.val = self.path + self.sel
 252                         self.session.openWithCallback(self.startRestore, MessageBox, _("Are you sure you want to restore\nfollowing backup:\n" + self.sel + "\nSystem will restart after the restore!"))
 253
 254         def keyCancel(self):
 255                 self.close()
 256
 257         def startRestore(self, ret = False):
 258                 if (ret == True):
 259                         self.exe = True
 260                         self.session.open(Console, title = _("Restore running"), cmdlist = ["tar -xzvf " + self.path + "/" + self.sel + " -C /", "killall -9 enigma2"])
 261
 262         def Exit(self):
 263                 self.close()
 264
 265 class RestoreScreen(Screen, ConfigListScreen):
 266         skin = """
 267                 <screen position="135,144" size="350,310" title="Restore running..." >
 268                 <widget name="config" position="10,10" size="330,250" transparent="1" scrollbarMode="showOnDemand" />
 269                 </screen>"""
 270
 271         def __init__(self, session, runRestore = False):
 272                 Screen.__init__(self, session)
 273                 self.session = session
 274                 self.runRestore = runRestore
 275                 self["actions"] = ActionMap(["WizardActions", "DirectionActions"],
 276                 {
 277                         "ok": self.close,
 278                         "back": self.close,
 279                         "cancel": self.close,
 280                 }, -1)
 281                 self.finished_cb = None
 282                 self.backuppath = getBackupPath()
 283                 self.backupfile = getBackupFilename()
 284                 self.fullbackupfilename = self.backuppath + "/" + self.backupfile
 285                 self.list = []
 286                 ConfigListScreen.__init__(self, self.list)
 287                 self.onLayoutFinish.append(self.layoutFinished)
 288                 if self.runRestore:
 289                         self.onShown.append(self.doRestore)
 290
 291         def layoutFinished(self):
 292                 self.setWindowTitle()
 293
 294         def setWindowTitle(self):
 295                 self.setTitle(_("Restore running..."))
 296
 297         def doRestore(self):
 298                 if self.finished_cb:
 299                         self.session.openWithCallback(self.finished_cb, Console, title = _("Restore running"), cmdlist = ["tar -xzvf " + self.fullbackupfilename + " -C /", "killall -9 enigma2"])
 300                 else:
 301                         self.session.open(Console, title = _("Restore running"), cmdlist = ["tar -xzvf " + self.fullbackupfilename + " -C /", "killall -9 enigma2"])
 302
 303         def backupFinishedCB(self,retval = None):
 304                 self.close(True)
 305
 306         def backupErrorCB(self,retval = None):
 307                 self.close(False)
 308
 309         def runAsync(self, finished_cb):
 310                 self.finished_cb = finished_cb
 311                 self.doRestore()
 312
 313