code.vuplus.com Git - vuplus_dvbapp-plugin/blob - epgrefresh/src/EPGRefreshResource.py

   1 from twisted.web import http, resource
   2 from EPGRefresh import epgrefresh
   3 from EPGRefreshService import EPGRefreshService
   4 from enigma import eServiceReference
   5 from Components.config import config
   6
   7 class EPGRefreshStartRefreshResource(resource.Resource):
   8         def render(self, req):
   9                 state = False
  10
  11                 if epgrefresh.forceRefresh():
  12                         output = "initiated refresh"
  13                         state = True
  14                 else:
  15                         output = "could not initiate refresh"
  16
  17                 req.setResponseCode(http.OK)
  18                 req.setHeader('Content-type', 'application; xhtml+xml')
  19                 req.setHeader('charset', 'UTF-8')
  20
  21                 return """<?xml version=\"1.0\" encoding=\"UTF-8\" ?>
  22                         <e2simplexmlresult>
  23                                 <e2state>%s</e2state>
  24                                 <e2statetext>%s</e2statetext>
  25                         </e2simplexmlresult>
  26                         """ % ('true' if state else 'false', output)
  27
  28 class EPGRefreshAddRemoveServiceResource(resource.Resource):
  29         TYPE_ADD = 0
  30         TYPE_DEL = 1
  31
  32         def __init__(self, type):
  33                 assert(type in (self.TYPE_ADD, self.TYPE_DEL))
  34                 self.type = type
  35
  36         def render(self, req):
  37                 do_add = self.type == self.TYPE_ADD
  38                 state = False
  39
  40                 if 'sref' in req.args:
  41                         sref = req.args["sref"][0]
  42                         if do_add:
  43                                 # strip all after last : (custom name)
  44                                 pos = sref.rfind(':')
  45                                 if pos != -1:
  46                                         sref = sref[:pos+1]
  47
  48                         duration = req.args.get("duration", None)
  49                         try:
  50                                 duration = duration and int(duration)
  51                         except ValueError, ve:
  52                                 output = 'invalid value for "duration": ' + str(duration)
  53                         else:
  54                                 epgservice = EPGRefreshService(sref, duration)
  55
  56                                 if sref:
  57                                         ref = eServiceReference(str(sref))
  58                                         if not ref.valid():
  59                                                 output = 'invalid value for "sref": ' + str(sref)
  60                                         elif (ref.flags & 7) == 7:
  61                                                 # bouquet
  62                                                 if epgservice in epgrefresh.services[1]:
  63                                                         if do_add:
  64                                                                 output = "bouquet already in list"
  65                                                         else:
  66                                                                 epgrefresh.services[1].remove(epgservice)
  67                                                                 output = "bouquet removed from list"
  68                                                                 state = True
  69                                                 else:
  70                                                         if do_add:
  71                                                                 epgrefresh.services[1].add(epgservice)
  72                                                                 output = "bouquet added to list"
  73                                                                 state = True
  74                                                         else:
  75                                                                 output = "bouquet not in list"
  76                                         else:
  77                                                 # assume service
  78                                                 if epgservice in epgrefresh.services[0]:
  79                                                         if do_add:
  80                                                                 output = "service already in list"
  81                                                         else:
  82                                                                 epgrefresh.services[0].remove(epgservice)
  83                                                                 output = "service removed from list"
  84                                                                 state = True
  85                                                 else:
  86                                                         if do_add:
  87                                                                 epgrefresh.services[0].add(epgservice)
  88                                                                 output = "service added to list"
  89                                                                 state = True
  90                                                         else:
  91                                                                 output = "service not in list"
  92
  93                                         # save if list changed
  94                                         if state:
  95                                                 epgrefresh.saveConfiguration()
  96                                 else:
  97                                         output = 'invalid value for "sref": ' + str(sref)
  98                 else:
  99                         output = 'missing argument "sref"'
 100
 101                 req.setResponseCode(http.OK)
 102                 req.setHeader('Content-type', 'application; xhtml+xml')
 103                 req.setHeader('charset', 'UTF-8')
 104
 105                 return """<?xml version=\"1.0\" encoding=\"UTF-8\" ?>
 106                         <e2simplexmlresult>
 107                                 <e2state>%s</e2state>
 108                                 <e2statetext>%s</e2statetext>
 109                         </e2simplexmlresult>
 110                         """ % ('true' if state else 'false', output)
 111
 112 class EPGRefreshListServicesResource(resource.Resource):
 113         def render(self, req):
 114                 # show xml
 115                 req.setResponseCode(http.OK)
 116                 req.setHeader('Content-type', 'application; xhtml+xml')
 117                 req.setHeader('charset', 'UTF-8')
 118                 return ''.join(epgrefresh.buildConfiguration(webif = True))
 119
 120 class EPGRefreshChangeSettingsResource(resource.Resource):
 121         def render(self, req):
 122                 for key, value in req.args.iteritems():
 123                         value = value[0]
 124                         if key == "enabled":
 125                                 config.plugins.epgrefresh.enabled.value = True if value == "true" else False
 126                         elif key == "begin":
 127                                 config.plugins.epgrefresh.begin.value = int(value)
 128                         elif key == "end":
 129                                 config.plugins.epgrefresh.end.value = int(value)
 130                         elif key == "interval":
 131                                 config.plugins.epgrefresh.interval.value = int(value)
 132                         elif key == "delay_standby":
 133                                 config.plugins.epgrefresh.delay_standby.value = int(value)
 134                         elif key == "inherit_autotimer":
 135                                 config.plugins.epgrefresh.inherit_autotimer.value = True if value == "true" else False
 136                         elif key == "afterevent":
 137                                 config.plugins.epgrefresh.afterevent.value = True if value == "true" else False
 138                         elif key == "force":
 139                                 config.plugins.epgrefresh.force.value = True if value == "true" else False
 140                         elif key == "wakeup":
 141                                 config.plugins.epgrefresh.wakeup.value = True if value == "true" else False
 142                         elif key == "pase_autotimer":
 143                                 config.plugins.epgrefresh.parse_autotimer.value = True if value == "true" else False
 144
 145                 config.plugins.epgrefresh.save()
 146
 147                 if config.plugins.epgrefresh.enabled.value:
 148                         epgrefresh.start()
 149                 else:
 150                         epgrefresh.stop()
 151
 152                 req.setResponseCode(http.OK)
 153                 req.setHeader('Content-type', 'application; xhtml+xml')
 154                 req.setHeader('charset', 'UTF-8')
 155
 156                 return """<?xml version=\"1.0\" encoding=\"UTF-8\" ?>
 157                         <e2simplexmlresult>
 158                                 <e2state>true</e2state>
 159                                 <e2statetext>config changed.</e2statetext>
 160                         </e2simplexmlresult>
 161                         """
 162
 163 class EPGRefreshSettingsResource(resource.Resource):
 164         def render(self, req):
 165                 req.setResponseCode(http.OK)
 166                 req.setHeader('Content-type', 'application; xhtml+xml')
 167                 req.setHeader('charset', 'UTF-8')
 168
 169                 from time import time, localtime, mktime
 170                 now = localtime()
 171                 begin_h = config.plugins.epgrefresh.begin.value
 172                 begin = mktime((
 173                         now.tm_year, now.tm_mon, now.tm_mday, begin_h[0], begin_h[1],
 174                         0, now.tm_wday, now.tm_yday, now.tm_isdst)
 175                 )
 176                 end_h = config.plugins.epgrefresh.end.value
 177                 end = mktime((
 178                         now.tm_year, now.tm_mon, now.tm_mday, end_h[0], end_h[1],
 179                         0, now.tm_wday, now.tm_yday, now.tm_isdst)
 180                 )
 181
 182                 return """<?xml version=\"1.0\" encoding=\"UTF-8\" ?>
 183                         <e2settings>
 184                                 <e2setting>
 185                                         <e2settingname>config.plugins.epgrefresh.enabled</e2settingname>
 186                                         <e2settingvalue>%s</e2settingvalue>
 187                                 </e2setting>
 188                                 <e2setting>
 189                                         <e2settingname>config.plugins.epgrefresh.begin</e2settingname>
 190                                         <e2settingvalue>%d</e2settingvalue>
 191                                 </e2setting>
 192                                 <e2setting>
 193                                         <e2settingname>config.plugins.epgrefresh.end</e2settingname>
 194                                         <e2settingvalue>%d</e2settingvalue>
 195                                 </e2setting>
 196                                 <e2setting>
 197                                         <e2settingname>config.plugins.epgrefresh.interval</e2settingname>
 198                                         <e2settingvalue>%d</e2settingvalue>
 199                                 </e2setting>
 200                                 <e2setting>
 201                                         <e2settingname>config.plugins.epgrefresh.delay_standby</e2settingname>
 202                                         <e2settingvalue>%d</e2settingvalue>
 203                                 </e2setting>
 204                                 <e2setting>
 205                                         <e2settingname>config.plugins.epgrefresh.inherit_autotimer</e2settingname>
 206                                         <e2settingvalue>%s</e2settingvalue>
 207                                 </e2setting>
 208                                 <e2setting>
 209                                         <e2settingname>config.plugins.epgrefresh.afterevent</e2settingname>
 210                                         <e2settingvalue>%s</e2settingvalue>
 211                                 </e2setting>
 212                                 <e2setting>
 213                                         <e2settingname>config.plugins.epgrefresh.force</e2settingname>
 214                                         <e2settingvalue>%s</e2settingvalue>
 215                                 </e2setting>
 216                                 <e2setting>
 217                                         <e2settingname>config.plugins.epgrefresh.wakeup</e2settingname>
 218                                         <e2settingvalue>%s</e2settingvalue>
 219                                 </e2setting>
 220                                 <e2setting>
 221                                         <e2settingname>config.plugins.epgrefresh.parse_autotimer</e2settingname>
 222                                         <e2settingvalue>%s</e2settingvalue>
 223                                 </e2setting>
 224                         </e2settings>
 225                         """ % (
 226                                 config.plugins.epgrefresh.enabled.value,
 227                                 begin,
 228                                 end,
 229                                 config.plugins.epgrefresh.interval.value,
 230                                 config.plugins.epgrefresh.delay_standby.value,
 231                                 config.plugins.epgrefresh.inherit_autotimer.value,
 232                                 config.plugins.epgrefresh.afterevent.value,
 233                                 config.plugins.epgrefresh.force.value,
 234                                 config.plugins.epgrefresh.wakeup.value,
 235                                 config.plugins.epgrefresh.parse_autotimer.value,
 236                         )
 237