menu.py

from __future__ import absolute_import, print_function

import os
import sys
import tempfile

from Components.ActionMap import ActionMap
from Components.Button import Button
from Components.config import config
from Components.Label import Label
from Components.ScrollLabel import ScrollLabel
from Components.Sources.List import List
from enigma import eEPGCache, eTimer
from Screens.MessageBox import MessageBox
from Screens.Screen import Screen
from Screens.Setup import Setup
from Tools.LoadPixmap import LoadPixmap

from . import _, e2m3u2bouquet, log
from . import plugin as E2m3u2b_Plugin
from .about import E2m3u2b_About
from .providers import E2m3u2b_Providers

try:
    import Plugins.Extensions.EPGImport.EPGImport as EPGImport
except ImportError:
    EPGImport = None


class E2m3u2b_Menu(Screen):
    skin = """
    <screen position="center,center" size="600,500">
        <widget name="key_red" position="0,0" size="140,40" valign="center" halign="center" zPosition="5"  foregroundColor="white" font="Regular;20" transparent="1" shadowColor="background" shadowOffset="-2,-2" />
        <widget name="key_green" position="140,0" size="140,40" valign="center" halign="center" zPosition="5"  foregroundColor="white" font="Regular;20" transparent="1" shadowColor="background" shadowOffset="-2,-2" />
        <ePixmap pixmap="skin_default/buttons/red.png" position="0,0" size="140,40" transparent="1" alphatest="on" />
        <ePixmap pixmap="skin_default/buttons/green.png" position="140,0" size="140,40" transparent="1" alphatest="on" />

        <widget source="list" render="Listbox" position="0,50" size="600,420" scrollbarMode="showOnDemand">
            <convert type="TemplatedMultiContent">
                {"template": [
                    MultiContentEntryPixmapAlphaTest(pos = (12, 4), size = (32, 32), png=0),
                    MultiContentEntryText(pos = (58, 5), size = (440, 38), font=0, flags = RT_HALIGN_LEFT|RT_VALIGN_TOP, text = 1),
                    ],
                    "fonts": [gFont("Regular",22)],
                    "itemHeight": 40
                }
            </convert>
        </widget>
    </screen>
    """

    def __init__(self, session):
        Screen.__init__(self, session)
        self.setTitle(_("IPTV Bouquet Maker"))
        self.skinName = ['E2m3u2b_Menu', 'AutoBouquetsMaker_Menu']

        self.onChangedEntry = []
        l = []
        self['list'] = List(l)

        self["actions"] = ActionMap(["ColorActions", "SetupActions", "MenuActions"],
                                {
                                    'red': self.keyCancel,
                                    'green': self.manual_update,
                                    'cancel': self.keyCancel,
                                    'ok': self.openSelected,
                                    'menu': self.keyCancel
                                }, -2)
        self['key_red'] = Button(_('Exit'))
        self['key_green'] = Button(_('Run'))
        self.epgimport = None

        if EPGImport and config.plugins.e2m3u2b.do_epgimport.value is True:
            # skip channelfilter for IPTV
            self.epgimport = EPGImport.EPGImport(eEPGCache.getInstance(), lambda x: True)
        self.createSetup()

    def createSetup(self):
        l = [self.build_list_entry(_('Configure')),
             self.build_list_entry(_('Providers')),
             self.build_list_entry(_('Create Bouquets')),
             self.build_list_entry(_('Status')),
             self.build_list_entry(_('Reset Bouquets')),
             self.build_list_entry(_('Show Log')),
             self.build_list_entry(_('About'))]
        self['list'].list = l

    def build_list_entry(self, description):
        pixmap = LoadPixmap(cached=True, path='{}/images/{}'.format(os.path.dirname(sys.modules[__name__].__file__), 'blank.png'))
        return((pixmap, description))

    def openSelected(self):
        index = self['list'].getIndex()

        if index == 0:
            self.session.openWithCallback(E2m3u2b_Plugin.done_configuring, E2m3u2b_Config)
            return
        if index == 1:
            self.session.open(E2m3u2b_Providers)
            return
        if index == 2:
            self.manual_update()
            return
        if index == 3:
            self.session.open(E2m3u2b_Status)
            return
        if index == 4:
            self.reset_bouquets()
            return
        if index == 5:
            self.session.open(E2m3u2b_Log)
            return
        if index == 6:
            self.session.open(E2m3u2b_About)
            return

    def manual_update(self):
        self.session.open(E2m3u2b_Update, self.epgimport)

    def reset_bouquets(self):
        """Remove any generated bouquets
        and epg importer config
        """
        self.session.openWithCallback(self.reset_bouquets_callback, MessageBox, _("This will remove the IPTV Bouquets and Epg Importer configs that have been created.\nProceed?"), MessageBox.TYPE_YESNO,
                                      default=False)

    def reset_bouquets_callback(self, confirmed):
        if not confirmed:
            return
        try:
            E2m3u2b_Plugin.do_reset()
        except Exception as e:
            print("[e2m3u2b] reset_bouquets_callback Error:", e, file=log)
            if config.plugins.e2m3u2b.debug.value:
                raise

    def keyCancel(self):
        self.close()


class E2m3u2b_Config(Setup):
    def __init__(self, session):
        Setup.__init__(self, session, "E2m3u2b_Config", plugin="Extensions/E2m3u2bouquet", PluginLanguageDomain="E2m3u2bouquet")


class E2m3u2b_Status(Screen):
    skin = """
        <screen position="center,center" size="600,500">
            <widget name="key_red" position="0,0" size="140,40" valign="center" halign="center" zPosition="5"  foregroundColor="white" font="Regular;20" transparent="1" shadowColor="background" shadowOffset="-2,-2" />
            <ePixmap pixmap="skin_default/buttons/red.png" position="0,0" size="140,40" transparent="1" alphatest="on" />
            <widget name="about" position="10,50" size="580,430" font="Regular;18"/>
        </screen>
        """

    def __init__(self, session):
        self.session = session
        Screen.__init__(self, session)
        Screen.setTitle(self, "IPTV Bouquet Maker - Status")
        self.skinName = ['E2m3u2b_Status', 'AutoBouquetsMaker_About']

        self["about"] = Label("")
        self["actions"] = ActionMap(["SetupActions", "ColorActions", "MenuActions"],
                                    {
                                        "red": self.keyCancel,
                                        "cancel": self.keyCancel,
                                        "menu": self.keyCancel
                                    }, -2)
        self["key_red"] = Button(_("Close"))

        if config.plugins.e2m3u2b.last_update:
            self["about"].setText(_('Last channel update: {}').format(config.plugins.e2m3u2b.last_update.value))

    def keyCancel(self):
        self.close()


class E2m3u2b_Log(Screen):
    skin = """
    <screen position="center,center" size="600,500">
    <ePixmap name="red" position="0,0" zPosition="2" size="140,40" pixmap="skin_default/buttons/red.png" transparent="1" alphatest="on" />
    <ePixmap name="green" position="140,0" zPosition="2" size="140,40" pixmap="skin_default/buttons/green.png" transparent="1" alphatest="on" />
    <ePixmap name="blue" position="280,0" zPosition="2" size="140,40" pixmap="skin_default/buttons/blue.png" transparent="1" alphatest="on" />
    <widget name="key_red" position="0,0" size="140,40" valign="center" halign="center" zPosition="4" foregroundColor="white" font="Regular;20" transparent="1" shadowColor="background" shadowOffset="-2,-2" />
    <widget name="key_green" position="140,0" size="140,40" valign="center" halign="center" zPosition="4" foregroundColor="white" font="Regular;20" transparent="1" shadowColor="background" shadowOffset="-2,-2" />
    <widget name="key_blue" position="280,0" size="140,40" valign="center" halign="center" zPosition="4" foregroundColor="white" font="Regular;20" transparent="1" shadowColor="background" shadowOffset="-2,-2" />
    <widget name="list" position="10,40" size="540,340" />
    </screen>"""

    def __init__(self, session):
        self.session = session
        Screen.__init__(self, session)
        Screen.setTitle(self, "IPTV Bouquet Maker - Log")
        self.skinName = ["E2m3u2b_Log", "AutoBouquetsMaker_Log"]

        self["key_red"] = Button(_("Close"))
        self["key_green"] = Button(_("Save"))
#        self["key_blue"] = Button("Clear")
        self["list"] = ScrollLabel(log.getvalue())
        self["actions"] = ActionMap(["DirectionActions", "OkCancelActions", "ColorActions", "MenuActions"],
                                    {
                                        "red": self.keyCancel,
                                        "green": self.keySave,
#                                        "blue": self.keyClear,
                                        "cancel": self.keyCancel,
                                        "ok": self.keyCancel,
                                        "left": self["list"].pageUp,
                                        "right": self["list"].pageDown,
                                        "up": self["list"].pageUp,
                                        "down": self["list"].pageDown,
                                        "pageUp": self["list"].pageUp,
                                        "pageDown": self["list"].pageDown,
                                        "menu": self.keyCancel,
                                    }, -2)

    def keyCancel(self):
        self.close(False)

    def keyClear(self):
        log.logfile.reset()  # this crashes Py3
        log.logfile.truncate()
        self.close(False)

    def keySave(self):
        path = tempfile.gettempdir()
        filename = os.path.join(path, 'e2m3u2bouquet.log')
        with open(filename, 'w') as f:
            f.write(log.getvalue())
        self.session.open(MessageBox, _('Log file has been saved to the tmp directory'), MessageBox.TYPE_INFO, timeout=30)


class E2m3u2b_Update(Screen):
    skin = """
            <screen position="center,center" size="600,500">
                <widget name="key_red" position="0,0" size="140,40" valign="center" halign="center" zPosition="5"  foregroundColor="white" font="Regular;20" transparent="1" shadowColor="background" shadowOffset="-2,-2" />
                <ePixmap pixmap="skin_default/buttons/red.png" position="0,0" size="140,40" transparent="1" alphatest="on" />
                <widget name="about" position="10,50" size="580,430" font="Regular;16"/>
            </screen>
            """

    def __init__(self, session, epgimport):
        self.session = session
        Screen.__init__(self, session)
        Screen.setTitle(self, "IPTV Bouquet Maker - Create Bouquets")
        self.skinName = ['E2m3u2b_Update', 'AutoBouquetsMaker_About']

        self["actions"] = ActionMap(["SetupActions", "ColorActions", "MenuActions"],
                                    {
                                        "red": self.keyCancel,
                                        "cancel": self.keyCancel,
                                        "menu": self.keyCancel
                                    }, -2)
        self["key_red"] = Button(_("Close"))

        self['about'] = Label()
        self['about'].setText(_('Starting...'))

        self.activityTimer = eTimer()
        try:
            self.activityTimer_conn = self.activityTimer.timeout.connect(self.prepare)
        except:
            self.activityTimer.timeout.get().append(self.prepare)

        self.update_status_timer = eTimer()
        try:
            self.update_status_timer_conn = self.update_status_timer.timeout.connect(self.update_status)
        except:
            self.update_status_timer.callback.append(self.update_status)

        self.epgimport = epgimport
        self.onLayoutFinish.append(self.populate)

    def populate(self):
        self.activityTimer.start(1)

    def prepare(self):
        self.activityTimer.stop()
        self.manual_update()

    def keyCancel(self):
        self.update_status_timer.stop()
        self.close()

    def manual_update(self):
        """Manual update
        """
        is_epgimport_running = False
        if self.epgimport:
            is_epgimport_running = self.epgimport.isImportRunning()

        if is_epgimport_running or e2m3u2bouquet.Status.is_running:
            self.session.open(MessageBox, _("Update still in progress. Please wait."), MessageBox.TYPE_ERROR, timeout=10, close_on_any_key=True)
            self.close()
            return
        else:
            self.session.openWithCallback(self.manual_update_callback, MessageBox, _("Update of channels will start.\n"
                                          "This may take a few minutes.\n"
                                          "Proceed?"), MessageBox.TYPE_YESNO,
                                          timeout=15, default=True)

    def manual_update_callback(self, confirmed):
        if not confirmed:
            self.close()
            return
        try:
            self.start_update()
        except Exception as e:
            print("[e2m3u2b] manual_update_callback Error:", e, file=log)
            if config.plugins.e2m3u2b.debug.value:
                raise

    def start_update(self):
        self.update_status_timer.start(2000)
        E2m3u2b_Plugin.start_update(self.epgimport)

    def update_status(self):
        self['about'].setText(e2m3u2bouquet.Status.message)

        if self.epgimport and self.epgimport.isImportRunning():
            self['about'].setText(_('EPG Import: Importing {} {} events').format(self.epgimport.source.description,
                                                                              self.epgimport.eventCount))


class E2m3u2b_Check(Screen):
    def __init__(self, session):
        Screen.__init__(self, session)
        self.session = session
        self.onShown.append(self.epimport_check)

    def epimport_check(self):
        if EPGImport is None:
            self.session.open(MessageBox, _('EPG Import not found\nPlease install the EPG Import plugin'),
                              MessageBox.TYPE_WARNING, timeout=10)
            self.close()


#class SetupSummary(Screen):
#    def __init__(self, session, parent):
#        Screen.__init__(self, session, parent=parent)
#        self['SetupTitle'] = StaticText(parent.setup_title)
#        self['SetupEntry'] = StaticText('')
#        self['SetupValue'] = StaticText('')
#        self.onShow.append(self.addWatcher)
#        self.onHide.append(self.removeWatcher)#
#
#    def addWatcher(self):
#        self.parent.onChangedEntry.append(self.selectionChanged)
#        self.parent['config'].onSelectionChanged.append(self.selectionChanged)
#        self.selectionChanged()#
#
#    def removeWatcher(self):
#        self.parent.onChangedEntry.remove(self.selectionChanged)
#        self.parent['config'].onSelectionChanged.remove(self.selectionChanged)#
#
#    def selectionChanged(self):
#        print('SetupSummary -> selectionChanged')
#        self['SetupEntry'].text = self.parent.getCurrentEntry()
#        self['SetupValue'].text = self.parent.getCurrentValue()
#        if hasattr(self.parent, 'getCurrentDescription'):
#            self.parent['description'].text = self.parent.getCurrentDescription()