OSDN Git Service

fix some miss.
[rec10/rec10-git.git] / rec10 / trunk / src / auto_process.py
index 45de50b..edf1083 100644 (file)
@@ -1,12 +1,13 @@
 #!/usr/bin/python
 # coding: UTF-8
 # Rec10 TS Recording Tools
-# Copyright (C) 2009 Yukikaze
+# Copyright (C) 2009-2010 Yukikaze
 import os
 import glob
 import time
+import datetime
 
-import recdb
+import recdblist
 import rec10d
 import os.path
 import tv2mp4
@@ -17,7 +18,7 @@ def Update_to_MP4(path):
     avilist = glob.glob(path + "/*.avi")
     mkvlist = glob.glob(path+"/*.mkv")
     avilist=avilist+mkvlist
-    #print avilist
+    #recdblist.printutf8(avilist)
     for avif in avilist:
         if avif.rfind("sa.avi")==-1:
             ##b25f is title.ts.b25  avi is title.avi
@@ -35,45 +36,67 @@ def Update_to_MP4(path):
                         if os.path.exists(mkvpath):
                             mkvdtime = int(time.time()-os.path.getmtime(mkvpath))
                             if mkvdtime>300:
-                                print mkvpath+":"+mp4path
+                                recdblist.printutf8(mkvpath+":"+mp4path)
                                 tv2mp4.mkv2mp4(mkvpath,mp4path)
                             else:
-                                print avipath+":"+mp4path
+                                recdblist.printutf8(avipath+":"+mp4path)
                                 tv2mp4.avi2mp4(avipath,mp4path)
                         else:
-                            print avipath+":"+mp4path
+                            recdblist.printutf8(avipath+":"+mp4path)
                             tv2mp4.avi2mp4(avipath,mp4path)
                 else:
                     if not os.path.exists(avipath):
                         if os.path.exists(mkvpath):
                             mkvdtime = int(time.time()-os.path.getmtime(mkvpath))
                             if mkvdtime>300:
-                                print mkvpath+":"+mp4path
+                                recdblist.printutf8(mkvpath+":"+mp4path)
                                 tv2mp4.mkv2mp4(mkvpath,mp4path)
 def auto_check(path):
     avilist = glob.glob(path + "/*.avi")
     mkvlist = glob.glob(path+"/*.mkv")
+    mp4list = glob.glob(path+"/*.mp4")
     tslist = glob.glob(path+"/*.ts")
     b25list = glob.glob(path+"/*.ts.b25")
-    filelist=avilist+mkvlist+tslist+b25list
-    rec10d.rec10db.new_auto_proc()
+    filelist=avilist+mkvlist+tslist+b25list+mp4list
+    add=[]
+    tbtime=datetime.datetime.now()+datetime.timedelta(seconds=60)
+    tbtime2=datetime.datetime.now()+datetime.timedelta(seconds=1200)
     for fn in filelist:
-        if fn.rfind("sa.avi")==-1:
+        if fn.rfind("sa.avi")==-1 and fn.rfind(".120.avi")==-1 and os.path.getsize(fn)>1*1000*1000:
             dir = os.path.split(fn)[0]
             file = os.path.split(fn)[1]
             title = os.path.splitext(file)[0]
+            title = unicode(title,'utf-8')
+            
+            #print dbs
             ext = os.path.splitext(file)[1]
-            if ext == "b25":
+            if ext == ".b25":
                 title=title.replace(".ts","")
-            proc=check_process(dir, title)
-            if proc=="b25":
-                rec10d.rec10db.add_auto_proc(recdb.REC_AUTO_SUGGEST_DECODE,title)
-            elif proc=="ts":
-                rec10d.rec10db.add_auto_proc(recdb.REC_AUTO_SUGGEST_ENCODE,title)
-            elif proc =="avi":
-                rec10d.rec10db.add_auto_proc(recdb.REC_AUTO_SUGGEST_AVI2MP4,title)
-            elif proc =="mkv":
-                rec10d.rec10db.add_auto_proc(recdb.REC_AUTO_SUGGEST_MKV2MP4,title)
+            dbs=rec10d.rec10db.select_by_name_time_timeline(title,tbtime,tbtime2)
+            if len(dbs)==0:
+                proc=check_process(dir, title)
+                chtxtt=rec10d.rec10db.select_chtxt_by_title_timeline_log(title)
+                nchtxt=""
+                if chtxtt!=None:
+                    nchtxt=chtxtt
+                if proc=="b25":
+                    """chtxtt=rec10d.rec10db.select_chtxt_by_title_timeline_log(title)
+                    nchtxt=""
+                    if chtxtt!=None:
+                        nchtxt=chtxtt"""
+                    add.append([recdblist.REC_AUTO_SUGGEST_DECODE,title,nchtxt])
+                elif proc=="ts":
+                    add.append([recdblist.REC_AUTO_SUGGEST_ENCODE,title,nchtxt])
+                elif proc =="264":
+                    add.append([recdblist.REC_AUTO_SUGGEST_AVI2FP,title,nchtxt])
+                #elif proc =="mp4":
+                #    add.append([recdblist.REC_AUTO_SUGGEST_AP2FP,title,nchtxt])
+    #print add
+    if len(add)>0:
+        rec10d.rec10db.new_auto_proc()
+        for a in add:
+            rec10d.rec10db.add_auto_proc(a[0],a[1],a[2])
+        time.sleep(1)
 """
     処理がどの段階まで言ったのかを調査し返す。
     return
@@ -88,17 +111,27 @@ def auto_check(path):
     mp4
 """
 def check_process(path,title):
-    path1 = os.path.join(path,title+".mp4")
+    path1 = os.path.join(path,title+".mkv")
     if os.path.exists(path1):
         if int(time.time()-os.path.getmtime(path1))>300:
+            return "mkv"
+        else:
+            return "mkvmaking"
+    elif os.path.exists(os.path.join(path,title+".mp4")):
+        if int(time.time()-os.path.getmtime(os.path.join(path,title+".mp4")))>300:
             return "mp4"
         else:
             return "mp4making"
-    elif os.path.exists(os.path.join(path,title+".avi")):
-        if int(time.time()-os.path.getmtime(os.path.join(path,title+".avi")))>300:
-            return "avi"
+    elif os.path.exists(os.path.join(path,title+".264")):
+        if int(time.time()-os.path.getmtime(os.path.join(path,title+".264")))>300:
+            return "264"
         else:
             return "encoding"
+    elif os.path.exists(os.path.join(path,title+".ts.log")):
+        if int(time.time()-os.path.getmtime(os.path.join(path,title+".ts")))<300:
+            return "encoding"
+        else:
+            return "ts"
     elif os.path.exists(os.path.join(path,title+".ts")):
         if int(time.time()-os.path.getmtime(os.path.join(path,title+".ts")))>300:
             return "ts"