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