rec10/trunk/src/epgdb.py

   1 #!/usr/bin/python
   2 # coding: UTF-8
   3 # Rec10 TS Recording Tools
   4 # Copyright (C) 2009 Yukikaze
   5 import datetime
   6 import os
   7 import shutil
   8
   9 import chdb
  10 import n_gram
  11 import rec10d
  12 import ts2epg
  13 import xml2db
  14 path = str(os.path.dirname(os.path.abspath(__file__))) + "/"
  15 tmppath = path
  16 def update(chtxt):
  17     print chtxt
  18     tnow = datetime.datetime.now()
  19     try:
  20         lastup = datetime.datetime.strptime(chdb.chtxtsearch(chtxt)[0]['update'], "%Y-%m-%d %H:%M:%S")
  21     except:
  22         lastup = datetime.datetime.strptime("2009-04-01 00:00:00", "%Y-%m-%d %H:%M:%S")
  23     dt = tnow-lastup
  24     dt = dt.days * 24 * 60 + dt.seconds
  25     if dt > 2 * 60 * 60:
  26         ts2epg.write(tmppath + chtxt + "epgdate.xml", chdb.chtxtsearch(chtxt)['ch'])
  27         xml2db.xml2db(tmppath + chtxt + "epgdate.xml", chdb.chtxtsearch(chtxt)['bctype'])
  28 def updatebc(bctype):
  29     bctypel = chdb.bctypesearch(bctype)
  30     print bctypel
  31     try:
  32         ts2epg.write(tmppath + bctype + "epgdate.xml", chdb.bctypesearch(bctype)['ch'])
  33         xml2db.xml2db(tmppath + bctype + "epgdate.xml", bctype)
  34         shutil.copyfile(tmppath + bctype + "epgdate.xml", tmppath + bctype + "epgdate.bak.xml")
  35         rec10d.rec10db.update_by_bctype_chdata(bctype)
  36     except Exception, inst:
  37         print type(inst)
  38         print inst
  39         xml2db.xml2db(tmppath + bctype + "epgdate.bak.xml", bctype)
  40         if rec10d.rec10db.select_by_bctype_chdata(bctype)[0][6] != "0":
  41             rec10d.rec10db.update_status_by_bctype_chdata(bctype, "2")
  42 def updatebc_bak(bctype):
  43     """
  44     update epg data by .bak file
  45     """
  46     xml2db.xml2db(tmppath + bctype + "epgdate.bak.xml", bctype)
  47 def updateall_bak():
  48     updatebc_bak("te13")
  49     updatebc_bak("te20")
  50     updatebc_bak("te22")
  51     updatebc_bak("te18")
  52     updatebc_bak("te19")
  53     updatebc_bak("te21")
  54     updatebc_bak("te23")
  55     updatebc_bak("cs1")
  56     updatebc_bak("cs2")
  57     updatebc_bak("bs")
  58 def updateall():
  59     updatebc("te13")
  60     updatebc("te20")
  61     updatebc("te22")
  62     updatebc("te18")
  63     updatebc("te19")
  64     updatebc("te21")
  65     updatebc("te23")
  66     updatebc("cs1")
  67     updatebc("cs2")
  68     updatebc("bs")
  69 #def search(titletxt):
  70 ##   return rec10d.rec10db.select_by_title_tv(titletxt)
  71 def searchtime2(titletxt, timet, deltatime, chtxt):
  72     time = datetime.datetime.strptime(timet, "%Y-%m-%d %H:%M:%S")
  73     ret = []
  74     deltatime = int(deltatime)
  75     beforetime = time.strftime("%Y%m%d%H%M%S")
  76     aftertime = time + datetime.timedelta(hours=deltatime)
  77     aftertime = aftertime.strftime("%Y%m%d%H%M%S")
  78     one = ["", "", "", ""]
  79     tempt = 0
  80     d = rec10d.rec10db.select_by_time_ngram_tv(beforetime, aftertime, chtxt)
  81     for chtxtt, title, start, stop in d:
  82         #print chtxtt+":"+title+":"+start+":"+stop
  83         p = n_gram.bigram(titletxt, title)
  84         btime = start
  85         btime = btime[0:4] + "-" + btime[4:6] + "-" + btime[6:8] + " " + btime[8:10] + ":" + btime[10:12] + ":00"
  86         etime = stop
  87         etime = etime[0:4] + "-" + etime[4:6] + "-" + etime[6:8] + " " + etime[8:10] + ":" + etime[10:12] + ":00"
  88         bt = datetime.datetime.strptime(btime, "%Y-%m-%d %H:%M:%S")
  89         dt = bt-time
  90         dt = 24 * 60 * 60 * dt.days + dt.seconds + 1
  91         p = p * 90
  92         dt = 1000-1000 * dt / (7 * 24 * 60 * 60)
  93         if p > 0:
  94             p = p + dt
  95             print dt
  96         if p > tempt:
  97             one = [chtxtt, title.encode('utf-8'), btime, etime]
  98             tempt = p
  99         if p > 0:
 100             print title + ":" + str(p) + "点"
 101     ret = one
 102     print ret
 103     return ret