rec10/trunk/src/epgdb.py

   1 #!/usr/bin/python\r
   2 # coding: UTF-8\r
   3 # Rec10 TS Recording Tools\r
   4 # Copyright (C) 2009-2011 Yukikaze\r
   5 import datetime\r
   6 import os\r
   7 import shutil\r
   8 import time\r
   9 import os.path\r
  10 import traceback\r
  11 \r
  12 import chdb\r
  13 import n_gram\r
  14 import rec10d\r
  15 import ts2epg\r
  16 import xml2db_dom\r
  17 import configreader\r
  18 import recdblist\r
  19 path = str(os.path.dirname(os.path.abspath(__file__))) + "/"\r
  20 tmppath = configreader.getConfPath("tmp")+"/"\r
  21 if tmppath=="/":\r
  22     tmppath=path\r
  23 if not os.path.exists(tmppath):\r
  24     os.mkdir(tmppath)\r
  25 def update(chtxt):\r
  26     recdblist.printutf8(chtxt)\r
  27     tnow = datetime.datetime.now()\r
  28     try:\r
  29         lastup = datetime.datetime.strptime(chdb.searchCHFromChtxt(chtxt)[0]['update'], "%Y-%m-%d %H:%M:%S")\r
  30     except:\r
  31         lastup = datetime.datetime.strptime("2009-04-01 00:00:00", "%Y-%m-%d %H:%M:%S")\r
  32     dt = tnow-lastup\r
  33     dt = dt.days * 24 * 60 + dt.seconds\r
  34     if dt > 2 * 60 * 60:\r
  35         ts2epg.writeEpgXML(os.path.join(tmppath, chtxt + "epgdata.xml"), chdb.searchCHFromChtxt(chtxt)['ch'])\r
  36         xml2db_dom.xml2db_dom(os.path.join(tmppath, chtxt + "epgdata.xml"), chdb.searchCHFromChtxt(chtxt)['bctype'])\r
  37 def updatebc(bctype):\r
  38     bctypel = chdb.searchCHFromBctype(bctype)\r
  39     #print bctype\r
  40     try:\r
  41         ts2epg.writeEpgXML(tmppath + bctype + "epgdata.xml", chdb.searchCHFromBctype(bctype)['ch'])\r
  42         xml2db_dom.xml2db_dom(tmppath + bctype + "epgdata.xml", bctype)\r
  43         shutil.copyfile(tmppath + bctype + "epgdata.xml", tmppath + bctype + "epgdata.bak.xml")\r
  44         time.sleep(5)\r
  45         rec10d.rec10db.update_by_bctype_epg_ch(bctype)\r
  46     except Exception, inst:\r
  47         recdblist.addCommonlogEX("Error","updatebc(epgdb.py)", str(type(inst)), str(inst)+"\n"+traceback.format_exc(),log_level=200)\r
  48         xml2db_dom.xml2db_dom(tmppath + bctype + "epgdata.bak.xml", bctype)\r
  49         rec10d.rec10db.update_status_by_bctype_epg_ch(bctype, "2")\r
  50 def updateLogo_bc(bctype):\r
  51     bctypel = chdb.searchCHFromBctype(bctype)\r
  52     #print bctype\r
  53     try:\r
  54         ts2epg.writeLogoEpgXML(tmppath + bctype + "logo.xml", chdb.searchCHFromBctype(bctype)['ch'])\r
  55         xml2db_dom.xml2db_dom_logo(tmppath + bctype + "logo.xml", bctype)\r
  56         shutil.copyfile(tmppath + bctype + "logo.xml", tmppath + bctype + "logo.bak.xml")\r
  57         time.sleep(5)\r
  58         rec10d.rec10db.update_logostatus_by_bctype_epg_ch(bctype,"1")\r
  59     except Exception, inst:\r
  60         recdblist.addCommonlogEX("Error","updatebc(epgdb.py)", str(type(inst)), str(inst)+"\n"+traceback.format_exc(),log_level=200)\r
  61         xml2db_dom.xml2db_dom(tmppath + bctype + "logo.bak.xml", bctype)\r
  62         rec10d.rec10db.update_logostatus_by_bctype_epg_ch(bctype,"2")\r
  63 def searchTime(titletxt, timet, deltatime, chtxt):\r
  64     time = datetime.datetime.strptime(timet, "%Y-%m-%d %H:%M:%S")\r
  65     ret = []\r
  66     deltatime = int(deltatime)\r
  67     beforetime = time.strftime("%Y%m%d%H%M%S")\r
  68     beforetime = time - datetime.timedelta(hours=deltatime)\r
  69     aftertime = time + datetime.timedelta(hours=deltatime)\r
  70     beforetime = beforetime.strftime("%Y%m%d%H%M%S")\r
  71     aftertime = aftertime.strftime("%Y%m%d%H%M%S")\r
  72     one = ["", "", "", "","","",""]\r
  73     tempt = 0\r
  74     chtxtt=chtxt.replace("_0","_%")\r
  75     d = rec10d.rec10db.select_by_time_ngram_epg_timeline(beforetime, aftertime, chtxtt)\r
  76     for chtxtt, title, start, stop ,exp,longexp,category in d:\r
  77         #recdblist.printutf8(chtxtt+":"+title+":"+start+":"+stop)\r
  78         p = n_gram.bigram(titletxt, title)\r
  79         btime = start\r
  80         btime = btime[0:4] + "-" + btime[4:6] + "-" + btime[6:8] + " " + btime[8:10] + ":" + btime[10:12] + ":00"\r
  81         etime = stop\r
  82         etime = etime[0:4] + "-" + etime[4:6] + "-" + etime[6:8] + " " + etime[8:10] + ":" + etime[10:12] + ":00"\r
  83         bt = datetime.datetime.strptime(btime, "%Y-%m-%d %H:%M:%S")\r
  84         dt = bt-time\r
  85         dt = abs(dt)\r
  86         dt = 24 * 60 * 60 * dt.days + dt.seconds + 1\r
  87         p = p * 90\r
  88         dt = 1000-1000 * dt / (7 * 24 * 60 * 60)\r
  89         if p > 0:\r
  90             p = p + dt\r
  91         if p > tempt:\r
  92             one = [chtxtt, title, btime, etime,exp,longexp,category]\r
  93             tempt = p\r
  94         if p > 0:\r
  95             #recdblist.printutf8(title + ":" + str(p) + u"点")\r
  96             ""\r
  97     ret = one\r
  98     #print ret\r
  99     return ret\r
 100 def searchTimeAuto(keytxt, timet, deltatime):\r
 101     time = datetime.datetime.strptime(timet, "%Y-%m-%d %H:%M:%S")\r
 102     ret = []\r
 103     deltatime = int(deltatime)\r
 104     beforetime = time.strftime("%Y%m%d%H%M%S")\r
 105     aftertime = time + datetime.timedelta(hours=deltatime)\r
 106     aftertime = aftertime.strftime("%Y%m%d%H%M%S")\r
 107     d = rec10d.rec10db.select_by_time_keyword_auto_suggest_epg_timeline(keytxt,beforetime, aftertime)\r
 108     for chtxtt, title, start, stop ,exp,longexp,category in d:\r
 109         btime = start\r
 110         btime = btime[0:4] + "-" + btime[4:6] + "-" + btime[6:8] + " " + btime[8:10] + ":" + btime[10:12] + ":00"\r
 111         etime = stop\r
 112         etime = etime[0:4] + "-" + etime[4:6] + "-" + etime[6:8] + " " + etime[8:10] + ":" + etime[10:12] + ":00"\r
 113         ret.append([chtxtt, title, btime, etime,exp,longexp,category])\r
 114     return ret\r
 115 def countSchedule(btime,etime):\r
 116     """\r
 117     return rec num (int return[0]:TE /int return[1]:BS/CS)\r
 118     """\r
 119     def cmpare(x, y):\r
 120         xt = x\r
 121         yt = y\r
 122         if xt > yt:\r
 123             return -1\r
 124         elif xt == yt:\r
 125             return 0\r
 126         else:\r
 127             return 1\r
 128     dls = rec10d.rec10db.select_bytime_all_timeline(btime, etime)\r
 129     times = []\r
 130     btd = datetime.datetime.strptime(btime, "%Y-%m-%d %H:%M:%S")\r
 131     etd = datetime.datetime.strptime(etime, "%Y-%m-%d %H:%M:%S")\r
 132     for dl in dls:\r
 133         dd1 = datetime.datetime.strptime(dl['btime'], "%Y-%m-%d %H:%M:%S")\r
 134         dd2 = datetime.datetime.strptime(dl['etime'], "%Y-%m-%d %H:%M:%S")\r
 135         if dd1 < btd:\r
 136             dd1 = btd\r
 137         if dd2 > etd:\r
 138             dd2 = etd\r
 139         times.append(dd1)\r
 140         times.append(dd2)\r
 141     times = list(set(times))\r
 142     times.sort(cmpare)\r
 143     times.reverse()\r
 144     retcount = [0, 0]\r
 145     for i in xrange(0, len(times)-2, 1):\r
 146         bt1 = times[i]\r
 147         et1 = times[i + 1]\r
 148         btime1 = bt1.strftime("%Y-%m-%d %H:%M:%S")\r
 149         etime1 = et1.strftime("%Y-%m-%d %H:%M:%S")\r
 150         ret1 = rec10d.rec10db.count_schedule_timeline(btime1, etime1)\r
 151         if retcount[0] < ret1[0]:\r
 152             retcount[0] = ret1[0]\r
 153         if retcount[1] < ret1[1]:\r
 154             retcount[1] = ret1[1]\r
 155     return retcount\r
 156 \r
 157 def searchSchedule(btime,etime):\r
 158     """\r
 159     return rec num (int return[0]:TE /int return[1]:BS/CS)\r
 160     """\r
 161     def cmpare(x, y):\r
 162         xt = x\r
 163         yt = y\r
 164         if xt > yt:\r
 165             return -1\r
 166         elif xt == yt:\r
 167             return 0\r
 168         else:\r
 169             return 1\r
 170     dls = rec10d.rec10db.select_bytime_all_timeline(btime, etime)\r
 171     times = []\r
 172     btd = datetime.datetime.strptime(btime, "%Y-%m-%d %H:%M:%S")\r
 173     etd = datetime.datetime.strptime(etime, "%Y-%m-%d %H:%M:%S")\r
 174     for dl in dls:\r
 175         dd1 = datetime.datetime.strptime(dl['btime'], "%Y-%m-%d %H:%M:%S")\r
 176         dd2 = datetime.datetime.strptime(dl['etime'], "%Y-%m-%d %H:%M:%S")\r
 177         if dd1 < btd:\r
 178             dd1 = btd\r
 179         if dd2 > etd:\r
 180             dd2 = etd\r
 181         times.append(dd1)\r
 182         times.append(dd2)\r
 183     times = list(set(times))\r
 184     times.sort(cmpare)\r
 185     times.reverse()\r
 186     ret=[]\r
 187     for i in xrange(0, len(times)-2, 1):\r
 188         bt1 = times[i]\r
 189         et1 = times[i + 1]\r
 190         btime1 = bt1.strftime("%Y-%m-%d %H:%M:%S")\r
 191         etime1 = et1.strftime("%Y-%m-%d %H:%M:%S")\r
 192         ret1 = rec10d.rec10db.search_schedule_timeline(btime1, etime1)\r
 193         ret.append(ret1)\r
 194     ret = list(set(ret))\r
 195     return ret\r
 196 \r
 197 def countEpgSchedule(epgbtime,epgetime):\r
 198     """\r
 199     return rec num (int return[0]:TE /int return[1]:BS/CS)\r
 200     """\r
 201     def cmpare(x, y):\r
 202         xt = x\r
 203         yt = y\r
 204         if xt > yt:\r
 205             return -1\r
 206         elif xt == yt:\r
 207             return 0\r
 208         else:\r
 209             return 1\r
 210     dls = rec10d.rec10db.select_byepgtime_over_timeline(epgbtime, epgetime)\r
 211     #print dls\r
 212     times = []\r
 213     btd = datetime.datetime.strptime(epgbtime, "%Y-%m-%d %H:%M:%S")\r
 214     etd = datetime.datetime.strptime(epgetime, "%Y-%m-%d %H:%M:%S")\r
 215     for dl in dls:\r
 216         dd1 = datetime.datetime.strptime(dl['epgbtime'], "%Y-%m-%d %H:%M:%S")\r
 217         dd2 = datetime.datetime.strptime(dl['epgetime'], "%Y-%m-%d %H:%M:%S")\r
 218         if dd1 < btd:\r
 219             dd1 = btd\r
 220         if dd2 > etd:\r
 221             dd2 = etd\r
 222         times.append(dd1)\r
 223         times.append(dd2)\r
 224     times = list(set(times))\r
 225     times.sort(cmpare)\r
 226     times.reverse()\r
 227     retcount = [0, 0]\r
 228     for i in xrange(0, len(times)-1, 1):\r
 229         bt1 = times[i]\r
 230         et1 = times[i + 1]\r
 231         epgbtime1 = bt1.strftime("%Y-%m-%d %H:%M:%S")\r
 232         epgetime1 = et1.strftime("%Y-%m-%d %H:%M:%S")\r
 233         ret1 = rec10d.rec10db.count_epgschedule_timeline(epgbtime1, epgetime1)\r
 234         if retcount[0] < ret1[0]:\r
 235             retcount[0] = ret1[0]\r
 236         if retcount[1] < ret1[1]:\r
 237             retcount[1] = ret1[1]\r
 238     return retcount\r
 239 def searchEpgSchedule(epgbtime,epgetime):\r
 240     """\r
 241     return rec num (int return[0]:TE /int return[1]:BS/CS)\r
 242     """\r
 243     def cmpare(x, y):\r
 244         xt = x\r
 245         yt = y\r
 246         if xt > yt:\r
 247             return -1\r
 248         elif xt == yt:\r
 249             return 0\r
 250         else:\r
 251             return 1\r
 252     dls = rec10d.rec10db.select_byepgtime_all_timeline(epgbtime, epgetime)\r
 253     times = []\r
 254     btd = datetime.datetime.strptime(epgbtime, "%Y-%m-%d %H:%M:%S")\r
 255     etd = datetime.datetime.strptime(epgetime, "%Y-%m-%d %H:%M:%S")\r
 256     for dl in dls:\r
 257         dd1 = datetime.datetime.strptime(dl['epgbtime'], "%Y-%m-%d %H:%M:%S")\r
 258         dd2 = datetime.datetime.strptime(dl['epgetime'], "%Y-%m-%d %H:%M:%S")\r
 259         if dd1 < btd:\r
 260             dd1 = btd\r
 261         if dd2 > etd:\r
 262             dd2 = etd\r
 263         times.append(dd1)\r
 264         times.append(dd2)\r
 265     times = list(set(times))\r
 266     times.sort(cmpare)\r
 267     times.reverse()\r
 268     ret=[]\r
 269     for i in xrange(0, len(times)-2, 1):\r
 270         bt1 = times[i]\r
 271         et1 = times[i + 1]\r
 272         epgbtime1 = bt1.strftime("%Y-%m-%d %H:%M:%S")\r
 273         epgetime1 = et1.strftime("%Y-%m-%d %H:%M:%S")\r
 274         ret1 = rec10d.rec10db.search_epgschedule_timeline(epgbtime1, epgetime1)\r
 275         ret.append(ret1)\r
 276     ret = list(set(ret))\r
 277     return ret\r