code.vuplus.com Git - vuplus_dvbapp/blob - lib/python/Plugins/SystemPlugins/ConfigurationBackup/plugin.py

   1 from enigma import *
   2 from Screens.Screen import Screen
   3 from Screens.MessageBox import MessageBox
   4 from Screens.Console import Console
   5 from Components.ActionMap import ActionMap, NumberActionMap
   6 from Components.Pixmap import *
   7 from Components.Pixmap import Pixmap
   8 from Components.Label import Label
   9 from Components.MenuList import MenuList
  10 from Components.config import ConfigSelection, ConfigSubsection, KEY_LEFT, KEY_RIGHT, KEY_0, getConfigListEntry
  11 from Components.ConfigList import ConfigList
  12 from Plugins.Plugin import PluginDescriptor
  13
  14 from Tools.NumericalTextInput import *
  15 from Tools.Directories import *
  16 import os
  17 import string
  18 import time
  19 import datetime
  20
  21 plugin_path = ""
  22
  23 # FIXME: harddiskmanager has a better overview about available mointpoints!
  24 BackupPath = {
  25                 "hdd" : "/media/hdd/backup",
  26                 "usb" : "/media/usb/backup",
  27                 "cf" : "/media/cf/backup"
  28         }
  29
  30 MountPoints = {
  31                 "hdd" : "/media/hdd",
  32                 "usb" : "/media/usb",
  33                 "cf" : "/media/cf"
  34         }
  35
  36 class BackupSetup(Screen):
  37         skin = """
  38                 <screen position="135,144" size="450,300" title="Backup and Restore" >
  39                         <widget name="config" position="10,10" size="430,240" />
  40                         <widget name="ok" position="10,255" size="100,40" pixmap="~/green.png" transparent="1" alphatest="on" />
  41                         <widget name="oktext" position="0,0" size="0,0" valign="center" halign="center" zPosition="2" font="Regular;20" transparent="1"  foregroundColor="black" />
  42                         <widget name="cancel" position="120,255" size="100,40" pixmap="~/red.png" transparent="1" alphatest="on" />
  43                         <widget name="canceltext" position="0,0" size="0,0" valign="center" halign="center" zPosition="2" font="Regular;20" transparent="1" foregroundColor="black" />
  44                         <widget name="restore" position="230,255" size="100,40" pixmap="~/yellow.png" transparent="1" alphatest="on" />
  45                         <widget name="restoretext" position="0,0" size="0,0" valign="center" halign="center" zPosition="2" font="Regular;20" transparent="1"  foregroundColor="black" />
  46                         <widget name="backup" position="340,255" size="100,40" pixmap="~/blue.png" transparent="1" alphatest="on" />
  47                         <widget name="backuptext" position="0,0" size="0,0" valign="center" halign="center" zPosition="2" font="Regular;20" transparent="1"  foregroundColor="black" />
  48                 </screen>"""
  49
  50         def keyLeft(self):
  51                 self["config"].handleKey(KEY_LEFT)
  52
  53         def keyRight(self):
  54                 self["config"].handleKey(KEY_RIGHT)
  55
  56         def keyNumberGlobal(self, number):
  57                 print "You pressed number", number
  58                 if (self["config"].getCurrent()[1].parent.enabled == True):
  59                         self["config"].handleKey(KEY_0+number)
  60
  61         def keyCancel(self):
  62                 for x in self["config"].list:
  63                         x[1].cancel()
  64                 self.close()
  65
  66         def keySave(self):
  67                 for x in self["config"].list:
  68                         x[1].save()
  69                 self.close()
  70
  71         def __init__(self, session, args = None):
  72                 Screen.__init__(self, session)
  73                 self.skin_path = plugin_path
  74
  75                 self["oktext"] = Label(_("OK"))
  76                 self["canceltext"] = Label(_("Cancel"))
  77                 self["backuptext"] = Label(_("Backup"))
  78                 self["restoretext"] = Label(_("Restore"))
  79                 self["restore"] = Pixmap()
  80                 self["backup"] = Pixmap()
  81                 self["ok"] = Pixmap()
  82                 self["cancel"] = Pixmap()
  83
  84                 self.path = ""
  85                 self.list = []
  86                 self["config"] = ConfigList(self.list)
  87                 self.createSetup()
  88
  89                 self["actions"] = NumberActionMap(["SetupActions"],
  90                 {
  91                         "ok": self.keySave,
  92                         "cancel": self.keyCancel,
  93                         "left": self.keyLeft,
  94                         "right": self.keyRight
  95                 }, -1)
  96
  97                 self["shortcuts"] = ActionMap(["ShortcutActions"],
  98                 {
  99                         "red": self.keyCancel,
 100                         "green": self.keySave,
 101                         "blue": self.Backup,
 102                         "yellow": self.Restore,
 103                 })
 104
 105
 106         def createSetup(self):
 107                 print "Creating BackupSetup"
 108                 self.list = [ ]
 109                 self["config"] = ConfigList(self.list)
 110                 self.backup = ConfigSubsection()
 111                 self.backup.type = ConfigSelection(choices = [("full", _("full /etc directory")), ("settings", _("only /etc/enigma2 directory")), ("var", _("/var directory")), ("skin", _("/usr/share/enigma2 directory"))])
 112                 self.backup.location = ConfigSelection(choices = [("usb", _("USB Stick")), ("cf", _("CF Drive")), ("hdd", _("Harddisk"))])
 113                 self.list.append(getConfigListEntry(_("Backup Mode"), self.backup.type))
 114                 self.list.append(getConfigListEntry(_("Backup Location"), self.backup.location))
 115
 116         def createBackupfolders(self):
 117                 self.path = BackupPath[self.backup.location.value]
 118                 print "Creating Backup Folder if not already there..."
 119                 if (os.path.exists(self.path) == False):
 120                         os.makedirs(self.path)
 121
 122         def Backup(self):
 123                 print "this will start the backup now!"
 124                 self.session.openWithCallback(self.runBackup, MessageBox, _("Do you want to backup now?\nAfter pressing OK, please wait!"))
 125
 126         def Restore(self):
 127                 print "this will start the restore now!"
 128                 self.session.open(RestoreMenu, self.backup)
 129
 130         def runBackup(self, result):
 131                 if result:
 132                         if os.path.ismount(MountPoints[self.backup.location.value]):
 133                                 self.createBackupfolders()
 134                                 d = time.localtime()
 135                                 dt = datetime.date(d.tm_year, d.tm_mon, d.tm_mday)
 136                                 self.path = BackupPath[self.backup.location.value]
 137                                 if self.backup.type.value == "full":
 138                                         print "Backup Mode: Full"
 139                                         self.session.open(Console, title = "Backup running", cmdlist = ["tar -czvf " + self.path + "/" + str(dt) + "_full_backup.tar.gz /etc/"])
 140                                 elif self.backup.type.value == "settings":
 141                                         print "Backup Mode: Settings"
 142                                         self.session.open(Console, title = "Backup running", cmdlist = ["tar -czvf " + self.path + "/" + str(dt) + "_settings_backup.tar.gz /etc/enigma2/"])
 143                                 elif self.backup.type.value == "var":
 144                                         print "Backup Mode: var"
 145                                         self.session.open(Console, title = "Backup running", cmdlist = [ "tar -czvf " + self.path + "/" + str(dt) + "_var_backup.tar.gz /var/"])
 146                                 elif self.backup.type.value == "skin":
 147                                         print "Backup Mode: skin"
 148                                         self.session.open(Console, title ="Backup running", cmdlist = [ "tar -czvf " + self.path + "/" + str(dt) + "_skin_backup.tar.gz /usr/share/enigma2/"])
 149                         else:
 150                                 self.session.open(MessageBox, _("Sorry your Backup destination does not exist\n\nPlease choose an other one."), MessageBox.TYPE_INFO)
 151
 152 class RestoreMenu(Screen):
 153         skin = """
 154                 <screen position="135,144" size="450,300" title="Restore Backups" >
 155                 <widget name="filelist" position="10,10" size="430,240" scrollbarMode="showOnDemand" />
 156                 <widget name="cancel" position="120,255" size="100,40" pixmap="~/red.png" transparent="1" alphatest="on" />
 157                 <widget name="canceltext" position="0,0" size="0,0" valign="center" halign="center" zPosition="2" font="Regular;20" transparent="1" foregroundColor="black" />
 158                 <widget name="restore" position="230,255" size="100,40" pixmap="~/yellow.png" transparent="1" alphatest="on" />
 159                 <widget name="restoretext" position="0,0" size="0,0" valign="center" halign="center" zPosition="2" font="Regular;20" transparent="1"  foregroundColor="black" />
 160                 </screen>"""
 161
 162         def __init__(self, session, backup):
 163                 Screen.__init__(self, session)
 164                 self.skin_path = plugin_path
 165                 self.backup = backup
 166
 167                 self["canceltext"] = Label(_("Cancel"))
 168                 self["restoretext"] = Label(_("Restore"))
 169                 self["restore"] = Pixmap()
 170                 self["cancel"] = Pixmap()
 171
 172                 self.sel = []
 173                 self.val = []
 174                 self.entry = False
 175                 self.exe = False
 176
 177                 self.path = ""
 178
 179                 self["actions"] = NumberActionMap(["SetupActions"],
 180                 {
 181                         "ok": self.KeyOk,
 182                         "cancel": self.keyCancel
 183                 }, -1)
 184
 185                 self["shortcuts"] = ActionMap(["ShortcutActions"],
 186                 {
 187                         "red": self.keyCancel,
 188                         "yellow": self.KeyOk,
 189                 })
 190                 self.flist = []
 191                 self["filelist"] = MenuList(self.flist)
 192                 self.fill_list()
 193
 194         def fill_list(self):
 195                 self.flist = []
 196                 self.path = BackupPath[self.backup.location.value]
 197                 if (os.path.exists(self.path) == False):
 198                         os.makedirs(self.path)
 199                 for file in os.listdir(self.path):
 200                         if (file.endswith(".tar.gz")):
 201                                 self.flist.append((file))
 202                                 self.entry = True
 203                                 self["filelist"].l.setList(self.flist)
 204
 205         def KeyOk(self):
 206                 if (self.exe == False) and (self.entry == True):
 207                         self.sel = self["filelist"].getCurrent()
 208                         self.val = self.path + self.sel
 209                         self.session.openWithCallback(self.startRestore, MessageBox, _("are you sure you want to restore\nfollowing backup:\n" + self.sel + "\nEnigma2 will restart after the restore"))
 210
 211         def keyCancel(self):
 212                 self.close()
 213
 214         def startRestore(self, ret = False):
 215                 if (ret == True):
 216                         self.exe = True
 217                         self.session.open(Console, title = "Restore running", cmdlist = ["tar -xzvf " + self.path + "/" + self.sel + " -C /", "killall -9 enigma2"])
 218
 219         def Exit(self):
 220                 self.close()
 221
 222 def BackupMain(session, **kwargs):
 223         session.open(BackupSetup)
 224
 225 def Plugins(path, **kwargs):
 226         global plugin_path
 227         plugin_path = path
 228         return PluginDescriptor(name="Backup/Restore", description="Backup and Restore your Settings", icon="backup.png", where = PluginDescriptor.WHERE_PLUGINMENU, fnc=BackupMain)