rec10/trunk/src/auto_process.py

   1 #!/usr/bin/python
   2 # coding: UTF-8
   3 # Rec10 TS Recording Tools
   4 # Copyright (C) 2009-2010 Yukikaze
   5 import os
   6 import glob
   7 import time
   8 import datetime
   9
  10 import recdblist
  11 import rec10d
  12 import os.path
  13 import tv2mp4
  14 def Update_to_MP4(path):
  15     """
  16     録画一時フォルダ内mp4ファイルを検索
  17     """
  18     avilist = glob.glob(path + "/*.avi")
  19     mkvlist = glob.glob(path+"/*.mkv")
  20     avilist=avilist+mkvlist
  21     #recdblist.printutf8(avilist)
  22     for avif in avilist:
  23         if avif.rfind("sa.avi")==-1:
  24             ##b25f is title.ts.b25  avi is title.avi
  25             dir = os.path.split(avif)[0]
  26             title = os.path.split(avif)[1]
  27             title = title.replace(".avi", "")
  28             title = title.replace(".mkv", "")
  29             avipath = os.path.join(dir, title + ".avi")
  30             mkvpath = os.path.join(dir, title + ".mkv")
  31             mp4path = os.path.join(dir, title + ".mp4")
  32             if not os.path.exists(mp4path):
  33                 if os.path.exists(avipath):
  34                     avidtime = int(time.time()-os.path.getmtime(avipath))
  35                     if avidtime>300:
  36                         if os.path.exists(mkvpath):
  37                             mkvdtime = int(time.time()-os.path.getmtime(mkvpath))
  38                             if mkvdtime>300:
  39                                 recdblist.printutf8(mkvpath+":"+mp4path)
  40                                 tv2mp4.mkv2mp4(mkvpath,mp4path)
  41                             else:
  42                                 recdblist.printutf8(avipath+":"+mp4path)
  43                                 tv2mp4.avi2mp4(avipath,mp4path)
  44                         else:
  45                             recdblist.printutf8(avipath+":"+mp4path)
  46                             tv2mp4.avi2mp4(avipath,mp4path)
  47                 else:
  48                     if not os.path.exists(avipath):
  49                         if os.path.exists(mkvpath):
  50                             mkvdtime = int(time.time()-os.path.getmtime(mkvpath))
  51                             if mkvdtime>300:
  52                                 recdblist.printutf8(mkvpath+":"+mp4path)
  53                                 tv2mp4.mkv2mp4(mkvpath,mp4path)
  54 def auto_check(path):
  55     avilist = glob.glob(path + "/*.avi")
  56     mkvlist = glob.glob(path+"/*.mkv")
  57     mp4list = glob.glob(path+"/*.mp4")
  58     tslist = glob.glob(path+"/*.ts")
  59     b25list = glob.glob(path+"/*.ts.b25")
  60     filelist=avilist+mkvlist+tslist+b25list+mp4list
  61     add=[]
  62     tbtime=datetime.datetime.now()+datetime.timedelta(seconds=60)
  63     tbtime2=datetime.datetime.now()+datetime.timedelta(seconds=1200)
  64     for fn in filelist:
  65         if fn.rfind("sa.avi")==-1 and fn.rfind(".120.avi")==-1 and os.path.getsize(fn)>1*1000*1000:
  66             dir = os.path.split(fn)[0]
  67             file = os.path.split(fn)[1]
  68             title = os.path.splitext(file)[0]
  69             title = unicode(title,'utf-8')
  70
  71             #print dbs
  72             ext = os.path.splitext(file)[1]
  73             if ext == ".b25":
  74                 title=title.replace(".ts","")
  75             dbs=rec10d.rec10db.select_by_name_time_timeline(title,tbtime,tbtime2)
  76             if len(dbs)==0:
  77                 proc=check_process(dir, title)
  78                 chtxtt=rec10d.rec10db.select_chtxt_by_title_timeline_log(title)
  79                 nchtxt=""
  80                 if chtxtt!=None:
  81                     nchtxt=chtxtt
  82                 if proc=="b25":
  83                     """chtxtt=rec10d.rec10db.select_chtxt_by_title_timeline_log(title)
  84                     nchtxt=""
  85                     if chtxtt!=None:
  86                         nchtxt=chtxtt"""
  87                     add.append([recdblist.REC_AUTO_SUGGEST_DECODE,title,nchtxt])
  88                 elif proc=="ts":
  89                     add.append([recdblist.REC_AUTO_SUGGEST_ENCODE,title,nchtxt])
  90                 elif proc =="avi":
  91                     add.append([recdblist.REC_AUTO_SUGGEST_AVI2FP,title,nchtxt])
  92                 #elif proc =="mp4":
  93                 #    add.append([recdblist.REC_AUTO_SUGGEST_AP2FP,title,nchtxt])
  94     #print add
  95     if len(add)>0:
  96         rec10d.rec10db.new_auto_proc()
  97         for a in add:
  98             rec10d.rec10db.add_auto_proc(a[0],a[1],a[2])
  99         time.sleep(1)
 100 """
 101     処理がどの段階まで言ったのかを調査し返す。
 102     return
 103     recording
 104     b25
 105     b25decoding
 106     tssplitting
 107     ts
 108     encoding
 109     avi
 110     mp4making
 111     mp4
 112 """
 113 def check_process(path,title):
 114     path1 = os.path.join(path,title+".mkv")
 115     if os.path.exists(path1):
 116         if int(time.time()-os.path.getmtime(path1))>300:
 117             return "mkv"
 118         else:
 119             return "mkvmaking"
 120     elif os.path.exists(os.path.join(path,title+".mp4")):
 121         if int(time.time()-os.path.getmtime(os.path.join(path,title+".mp4")))>300:
 122             return "mp4"
 123         else:
 124             return "mp4making"
 125     elif os.path.exists(os.path.join(path,title+".avi")):
 126         if int(time.time()-os.path.getmtime(os.path.join(path,title+".avi")))>300:
 127             return "avi"
 128         else:
 129             return "encoding"
 130     elif os.path.exists(os.path.join(path,title+".ts.log")):
 131         if int(time.time()-os.path.getmtime(os.path.join(path,title+".ts")))<300:
 132             return "encoding"
 133         else:
 134             return "ts"
 135     elif os.path.exists(os.path.join(path,title+".ts")):
 136         if int(time.time()-os.path.getmtime(os.path.join(path,title+".ts")))>300:
 137             return "ts"
 138         else:
 139             return "tssplitting"
 140     elif os.path.exists(os.path.join(path,title+".sa.avi")):
 141         if int(time.time()-os.path.getmtime(os.path.join(path,title+".sa.avi")))>300:
 142             return "ts"
 143         else:
 144             return "tssplitting"
 145     elif os.path.exists(os.path.join(path,title+".ts.b25")):
 146         if int(time.time()-os.path.getmtime(os.path.join(path,title+".ts.b25")))>300:
 147             return "b25"
 148         else:
 149             return "recording"
 150
 151