rsdownloader/src/RSMain.py

   1 ##
   2 ## RS Downloader
   3 ## by AliAbdul
   4 ##
   5 from Components.ActionMap import ActionMap
   6 from Components.MenuList import MenuList
   7 from os import path as os_path
   8 from RS import rapidshare
   9 from RSConfig import RSConfig, config
  10 from RSDownloadBrowser import RSDownloadBrowser
  11 from RSListBrowser import RSListBrowser
  12 from RSProgress import RSProgress
  13 from RSSearch import RSSearch
  14 from RSTranslation import _, TitleScreen
  15 from Screens.Console import Console
  16 from Screens.MessageBox import MessageBox
  17 from Tools.Directories import fileExists
  18
  19 if fileExists("/usr/lib/enigma2/python/Screens/LTKeyBoard.pyc"):
  20         from Screens.LTKeyBoard import LTKeyBoard
  21         LT = True
  22 else:
  23         from Components.Input import Input
  24         from Screens.InputBox import InputBox
  25         LT = False
  26
  27 ##############################################################################
  28
  29 class RSMain(TitleScreen):
  30         skin = """
  31                 <screen position="200,165" size="320,270" title="RS Downloader">
  32                         <widget name="list" position="10,10" size="300,250" />
  33                 </screen>"""
  34
  35         def __init__(self, session, args = None):
  36                 TitleScreen.__init__(self, session)
  37                 self.session = session
  38
  39                 self["list"] = MenuList([
  40                         _("Start downloading..."),
  41                         _("Show lists..."),
  42                         _("Show downloads..."),
  43                         _("Show progress..."),
  44                         _("Edit configs..."),
  45                         _("Show status..."),
  46                         _("Show log..."),
  47                         _("Delete log..."),
  48                         _("Use search-engine..."),
  49                         _("Show info...")])
  50
  51                 self["actions"] = ActionMap(["OkCancelActions"], {"ok": self.okClicked, "cancel": self.close}, -1)
  52
  53         def okClicked(self):
  54                 selected = self["list"].getCurrent()
  55                 if selected is not None:
  56                         if selected == (_("Start downloading...")):
  57                                 if not rapidshare.downloading:
  58                                         ret = rapidshare.startDownloading()
  59                                         if ret:
  60                                                 self.session.open(MessageBox, _("Started downloading..."), MessageBox.TYPE_INFO)
  61                                         else:
  62                                                 self.session.open(MessageBox, _("Could not start downloading!"), MessageBox.TYPE_ERROR)
  63                                 else:
  64                                         self.session.open(MessageBox, _("Already downloading!"), MessageBox.TYPE_ERROR)
  65
  66                         elif selected == (_("Show lists...")):
  67                                 path = config.plugins.RSDownloader.lists_directory.value
  68                                 if os_path.exists(path):
  69                                         self.session.open(RSListBrowser, path)
  70                                 else:
  71                                         self.session.open(MessageBox, (_("Couldn't find the directory %s!") % path), MessageBox.TYPE_ERROR)
  72
  73                         elif selected == (_("Show downloads...")):
  74                                 path = config.plugins.RSDownloader.downloads_directory.value
  75                                 if os_path.exists(path):
  76                                         self.session.open(RSDownloadBrowser, path)
  77                                 else:
  78                                         self.session.open(MessageBox, (_("Couldn't find the directory %s!") % path), MessageBox.TYPE_ERROR)
  79
  80                         elif selected == (_("Show progress...")):
  81                                 self.session.open(RSProgress)
  82
  83                         elif selected == (_("Edit configs...")):
  84                                 self.session.open(RSConfig)
  85
  86                         elif selected == (_("Show status...")):
  87                                 if rapidshare.downloading:
  88                                         self.session.open(MessageBox, (_("Downloading %s.") % rapidshare.downloading_file), MessageBox.TYPE_INFO)
  89                                 else:
  90                                         self.session.open(MessageBox, _("Stopped!"), MessageBox.TYPE_INFO)
  91
  92                         elif selected == (_("Show log...")):
  93                                 self.session.open(Console, "RS Downloader", ["cat /tmp/rapidshare.log"])
  94
  95                         elif selected == (_("Delete log...")):
  96                                 self.session.open(Console, "RS Downloader", ["rm -f /tmp/rapidshare.log"])
  97
  98                         elif selected == (_("Use search-engine...")):
  99                                 title = _("Search http://rapidshare-search-engine.com for:")
 100                                 if LT:
 101                                         self.session.openWithCallback(self.searchCallback, LTKeyBoard, title=title)
 102                                 else:
 103                                         self.session.openWithCallback(self.searchCallback, InputBox, title=title, text="", maxSize=False, type=Input.TEXT)
 104
 105                         elif selected == (_("Show info...")):
 106                                 self.session.open(MessageBox, (_("RS Downloader\nby AliAbdul\n\nThis plugin allows you to download files from rapidshare in the background.")), MessageBox.TYPE_INFO)
 107
 108         def searchCallback(self, callback):
 109                 if callback is not None and callback != "":
 110                         self.session.open(RSSearch, callback)
 111