#!/usr/bin/python\r
# coding: UTF-8\r
# Rec10 TS Recording Tools\r
-# Copyright (C) 2009-2010 Yukikaze\r
+# Copyright (C) 2009-2011 Yukikaze\r
import commands\r
import shutil\r
-import tv2avi\r
+import auto_process\r
import os\r
import re\r
import os.path\r
import string\r
import base64\r
import time\r
+import subprocess\r
+import traceback\r
+import zip\r
+\r
+import tv2avi\r
import recdblist\r
import configreader\r
import status\r
-\r
+import tv2audio\r
+path = str(os.path.dirname(os.path.abspath(__file__))) + "/"\r
+tmppath = configreader.getConfPath("tmp")+"/"\r
+if tmppath=="/":\r
+ tmppath=path\r
+if not os.path.exists(tmppath):\r
+ os.mkdir(tmppath)\r
def ts2mp4(pin, pout, opt):\r
dir=os.path.split(pout)[0]\r
title=os.path.split(pout)[1]\r
if os.path.isfile(tpraw) and os.path.getsize(tpraw)>10*1000:\r
raw2mp4(tpraw, tpmp4, opt)\r
time.sleep(10)\r
- if os.path.exists(tpraw):\r
+ if os.path.exists(tpraw) and not re.search("B",opt):\r
os.remove(tpraw)\r
+ zip.addFile2FileZip(recdblist.getLogTitle(pin)+".command.log", recdblist.getLogTitle(pin)+".log.zip")\r
+ if os.path.exists(recdblist.getLogTitle(pin)+".command.log"):\r
+ os.remove(recdblist.getLogTitle(pin)+".command.log")\r
def raw2mp4(pin,pout,opt):\r
dir=os.path.split(pout)[0]\r
title=os.path.split(pout)[1]\r
title=os.path.splitext(title)[0]\r
duration="-fps 29.970030 "\r
- ffmpeg=configreader.getpath("ffmpeg")\r
- isdualaac=0\r
- ispentaaudio=0\r
if re.search("a",opt):\r
duration="-fps 23.976023 "\r
- elif re.search("v",opt):\r
- duration="-fps 23.976023 "\r
- if re.search("d",opt):\r
- isdualaac=1\r
- duration="-fps 29.970030 "##ffmpegが24fpsに対応していないための措置\r
- elif re.search("5",opt):\r
- ispentaaudio=1\r
- duration="-fps 29.970030 "##ffmpegが24fpsに対応していないための措置\r
- exe = configreader.getpath("mp4box")\r
+ if re.search("I",opt):\r
+ duration="-fps 29.970030 "\r
+ exe = configreader.getConfPath("mp4box")+u" -tmp "+tmppath\r
txt=""\r
os.environ['LANG']="ja_JP.UTF-8"\r
- if isdualaac==1:\r
- wineexe=configreader.getpath("wine")\r
- pincap=pin.replace(".264",".srt")\r
- try:\r
- cap2ass=configreader.getpath("caption2ass")\r
- except:\r
- cap2ass=""\r
- pints=pin.replace(".264",".ts")\r
- paac1=pin.replace(".264","_1.aac")\r
- paac2=pin.replace(".264","_2.aac")\r
- recdblist.printutf8(paac1)\r
- if not os.path.exists(paac1):\r
- paac1=pin.replace(".264","_1.mp3")\r
- if not os.path.exists(paac2):\r
- paac2=pin.replace(".264","_2.mp3")\r
- e0=wineexe+u" "+cap2ass+u" -format srt \"Z:\\"+pints+"\" \"Z:\\"+pincap+"\""\r
- e1=exe +u" "+duration+" -add \""+pin+"\" -new \""+pout+"\""\r
- e1a1=exe +u" -add \""+paac1+"\" \""+pout+"\""\r
- e1a2=exe +u" -add \""+paac2+"\" \""+pout+"\""\r
- e1s=exe +u" -add \""+pincap+"\" \""+pout+"\""\r
- if os.path.isfile(cap2ass):\r
- txt=txt+execcomd(e0)\r
- execmp4box(pin, pout, e1)\r
- addmp4(paac1, pout, e1a1)\r
- addmp4(paac2, pout, e1a2)\r
- if os.path.exists(pincap) and (os.path.getsize(pincap)>1000):\r
- addmp4(pincap, pout, e1s)\r
- elif ispentaaudio==1:\r
- wineexe=configreader.getpath("wine")\r
- pincap=pin.replace(".264",".srt")\r
- try:\r
- cap2ass=configreader.getpath("caption2ass")\r
- except:\r
- cap2ass=""\r
- pints=pin.replace(".264",".ts")\r
- paac1=pin.replace(".264","_1.aac")\r
- paac2=pin.replace(".264","_2.aac")\r
- recdblist.printutf8(paac1)\r
- if not os.path.exists(paac1):\r
- paac1=pin.replace(".264","_1.mp3")\r
- if not os.path.exists(paac2):\r
- paac2=pin.replace(".264","_2.mp3")\r
- e0=wineexe+u" "+cap2ass+u" -format srt \"Z:\\"+pints+"\" \"Z:\\"+pincap+"\""\r
- e1=exe +u" "+duration+" -add \""+pin+"\" -new \""+pout+"\""\r
- e1a1=exe +u" -add \""+paac1+"\" \""+pout+"\""\r
- e1a2=exe +u" -add \""+paac2+"\" \""+pout+"\""\r
- e1s=exe +u" -add \""+pincap+"\" \""+pout+"\""\r
- if os.path.isfile(cap2ass):\r
- txt=txt+execcomd(e0)\r
- execmp4box(pin, pout, e1)\r
- addmp4(paac1, pout, e1a1)\r
- addmp4(paac2, pout, e1a2)\r
- if os.path.exists(pincap) and (os.path.getsize(pincap)>1000):\r
- addmp4(pincap, pout, e1s)\r
- else:\r
- wineexe=configreader.getpath("wine")\r
- pincap=pin.replace(".264",".srt")\r
- try:\r
- cap2ass=configreader.getpath("caption2ass")\r
- except:\r
- cap2ass=""\r
- pints=pin.replace(".264",".ts")\r
- pinaac=pin.replace(".264",".aac")\r
- e0=wineexe+u" "+cap2ass+u" -format srt \"Z:\\"+pints+"\" \"Z:\\"+pincap+"\""\r
- e1=ffmpeg+" -i \""+pints+"\" -vn -f aac -acodec copy \""+pinaac+"\""\r
- e2=exe +u" "+duration+" -add \""+pin+"\" -new \""+pout+"\""\r
- e2a=exe +u" -add \""+pinaac+"\" \""+pout+"\""\r
- e2s=exe +u" -add \""+pincap+"\" \""+pout+"\""\r
- if os.path.isfile(cap2ass):\r
- txt=txt+execcomd(e0)\r
- txt=txt+execcomd(e1)\r
- execmp4box(pin, pout, e2)\r
- addmp4(pinaac, pout, e2a)\r
- if os.path.exists(pincap) and (os.path.getsize(pincap)>1000):\r
- addmp4(pincap, pout, e2s)\r
- recdblist.addlog(pout, txt, "mp4box-log")\r
+ pints=pin.replace(".264",".ts")\r
+ e1=exe +u" "+duration+" -add \""+pin+"\" -new \""+pout+"\""\r
+ execmp4box(pin, pout, e1)\r
+ addAudio(pints, pout, opt)\r
+ addCaption(pints, pout)\r
if status.getSettings_auto_del_tmp()==1:\r
if os.path.exists(pout):\r
- if re.search(opt,"MW8") or re.search(opt,"MW9"):\r
- if os.path.getsize(pout)>50*1000*1000:\r
- deltitle(dir,title)\r
- elif re.search(opt,"8") or re.search(opt,"9"):\r
- ""\r
+ if re.search("t",opt):\r
+ auto_process.deleteTmpFile(dir, title, ".264")\r
+ elif re.search("k",opt):\r
+ ""#削除しない\r
else:\r
- if os.path.getsize(pout)>100*1000*1000:\r
- deltitle(dir,title)\r
+ auto_process.deleteTmpFile(dir, title, ".mp4")\r
def mkv2mp4(pin,pout):\r
- exeb = configreader.getpath(u"mkvextract")\r
- exe = configreader.getpath(u"mp4Box")\r
- #dtsedit=configreader.getpath("DtsEdit")\r
- wineexe=configreader.getpath("wine")\r
+ exeb = configreader.getConfPath(u"mkvextract")\r
+ exe = configreader.getConfPath(u"mp4Box")\r
+ #dtsedit=configreader.getConfPath("DtsEdit")\r
+ wineexe=configreader.getConfPath("wine")\r
dir=os.path.split(pin)[0]\r
title=os.path.split(pin)[1]\r
title=os.path.splitext(title)[0]\r
txt=txt+execcomd(exe3)+"\n"\r
except:\r
""\r
- recdblist.addlog(pin, txt, "MKV2MP4-log")\r
+ recdblist.addLog(pin, txt, u"MKV2MP4-log")\r
txt = "\n####MKV2MP4-log####\n"+txt\r
time.sleep(10)\r
if status.getSettings_auto_del_tmp()==1:\r
if os.path.exists(pout):\r
- if os.path.getsize(pout)>100*1000*1000:\r
- deltitle(dir,title)\r
+ auto_process.deleteTmpFile(dir, title, ".mp4")\r
+def addCaption(pts,pmp4):##字幕の追加を試みる。\r
+ wineexe=configreader.getConfPath("wine")\r
+ pincap=pts.replace(".ts",".srt")\r
+ try:\r
+ cap2ass=configreader.getConfPath("caption2ass")\r
+ except:\r
+ cap2ass=""\r
+ if os.path.isfile(cap2ass):\r
+ e0=wineexe+u" "+cap2ass+u" -format srt \"Z:\\"+pts+"\" \"Z:\\"+pincap+"\""\r
+ recdblist.printutf8(e0)\r
+ p0=subprocess.Popen(e0,shell=True,stdout=subprocess.PIPE)\r
+ time.sleep(100)\r
+ if p0.poll==None:#実行中\r
+ #if os.path.getsize(pincap)<1000:#2mで1kb以下の場合自動で終了\r
+ # os.kill(p0.pid,signal.SIGKILL)\r
+ #else:\r
+ os.waitpid(p0.pid, 0)\r
+ logt=unicode(p0.communicate()[0], "UTF-8")\r
+ recdblist.addLog(pts,e0, u"Captionログ-コマンド")\r
+ recdblist.addLog(pts,logt, u"Captionログ-詳細")\r
+ recdblist.addCommandLogZip(pts, "mp4box_caption", "mp4box_caption", e0, logt)\r
+ if os.path.exists(pincap):\r
+ if os.path.getsize(pincap)>1000:\r
+ exe = configreader.getConfPath("mp4box")+u" -tmp "+tmppath\r
+ e1s=exe +u" -add \""+pincap+"\" \""+pmp4+"\""\r
+ addmp4(pincap,pmp4,e1s)\r
+def addAudio(pts,pmp4,opts):#オプションに応じた音声の追加を行う\r
+ exe = configreader.getConfPath("mp4box")+u" -tmp "+tmppath\r
+ if re.search("d",opts) or re.search("5",opts):#二カ国語放送/5.1ch放送の場合\r
+ paac1=pts.replace(".ts","_1.aac")\r
+ paac2=pts.replace(".ts","_2.aac")\r
+ recdblist.printutf8(paac1)\r
+ if not os.path.exists(paac1):\r
+ paac1=pts.replace(".ts","_1.mp3")\r
+ if not os.path.exists(paac2):\r
+ paac2=pts.replace(".ts","_2.mp3")\r
+ e1a1=exe +u" -add \""+paac1+"\" \""+pmp4+"\""\r
+ e1a2=exe +u" -add \""+paac2+"\" \""+pmp4+"\""\r
+ if os.path.exists(paac1):\r
+ addmp4(paac1, pmp4, e1a1)\r
+ if os.path.exists(paac2):\r
+ addmp4(paac2, pmp4, e1a2)\r
+ elif re.search("b",opts):#BonTsDemuxを使って音声をスプリットした場合\r
+ paac=pts.replace(".ts",".aac")\r
+ if not os.path.exists(paac):\r
+ paac=pts.replace(".ts",".mp3")\r
+ e1a1=exe +u" -add \""+paac+"\" \""+pmp4+"\""\r
+ if os.path.exists(paac):\r
+ addmp4(paac, pmp4, e1a1)\r
+ else:\r
+ tv2audio.ts2single_audio(pts,opts)\r
+ pinaac=pts.replace(".ts",".aac")\r
+ if not os.path.exists(pinaac):\r
+ pinaac=pinaac.replace(".aac",".mp3")\r
+ e1a=exe +u" -add \""+pinaac+"\" \""+pmp4+"\""\r
+ if os.path.exists(pinaac):\r
+ addmp4(pinaac, pmp4, e1a)\r
def execmp4box(pin,pout,cmd):\r
title=os.path.splitext(os.path.split(pin)[1])[0]\r
- ptin=os.path.join(os.path.dirname(pin),base64.b16encode(title.encode('utf-8'))+".264")\r
+ nt=base64.b16encode(title.encode('utf-8'))\r
+ if len(nt)>200:\r
+ nt=nt[:180]\r
+ ptin=os.path.join(os.path.dirname(pin),nt+".264")\r
recdblist.printutf8(ptin)\r
shutil.move(pin,ptin)\r
time.sleep(10)\r
- ptout=os.path.join(os.path.dirname(pout),base64.b16encode(title.encode('utf-8'))+".mp4")\r
+ ptout=os.path.join(os.path.dirname(pout),nt+".mp4")\r
cmdn=string.replace(cmd,pin,ptin)\r
cmdn=string.replace(cmdn,pout,ptout)\r
recdblist.printutf8(cmdn)\r
+ recdblist.addCommandSelfLog(pin, cmdn)\r
txt=""\r
try:\r
txt=execcomd(cmdn)\r
txt= "error occures in execmp4box\n"\r
txt=txt+ str(type(inst))+"\n"\r
txt=txt+str(inst)\r
- recdblist.addlog(pin, txt, "MP4Box-log")\r
+ recdblist.addCommonlogEX("Error", "excecmp4box(tv2mp4.py)", str(type(inst)), str(inst)+traceback.format_exc(),verbose_level=200,log_level=200)\r
+ recdblist.addLog(pin, cmdn, u"MP4Boxログ-コマンド")\r
+ recdblist.addLog(pin, txt, u"MP4Boxログ-詳細")\r
+ recdblist.addCommandLogZip(pin, "MP4Box", "mp4box", cmdn,txt)\r
time.sleep(5)\r
shutil.move(ptin,pin)\r
shutil.move(ptout,pout)\r
def addmp4(padd,pout,cmd):#without video\r
title=os.path.splitext(os.path.split(padd)[1])[0]\r
ext=os.path.splitext(os.path.split(padd)[1])[1]\r
- ptadd=os.path.join(os.path.dirname(padd),base64.b16encode(title.encode('utf-8'))+ext)\r
- ptoutb=os.path.join(os.path.dirname(pout),base64.b16encode(title.encode('utf-8'))+"_b.mp4")\r
- ptout=os.path.join(os.path.dirname(pout),base64.b16encode(title.encode('utf-8'))+".mp4")\r
+ nt=base64.b16encode(title.encode('utf-8'))\r
+ if len(nt)>200:\r
+ nt=nt[:180]\r
+ ptadd=os.path.join(os.path.dirname(padd),nt+ext)\r
+ ptoutb=os.path.join(os.path.dirname(pout),nt+"_b.mp4")\r
+ ptout=os.path.join(os.path.dirname(pout),nt+".mp4")\r
shutil.move(padd,ptadd)\r
if os.path.isfile(pout):\r
shutil.move(pout,ptoutb)\r
txt= "error occures in addmp4\n"\r
txt=txt+ str(type(inst))+"\n"\r
txt=txt+str(inst)\r
- recdblist.addlog(pout, txt, "MP4Box-log-add")\r
+ recdblist.addLog(pout, cmdn, u"MP4Box追加ログ-コマンド")\r
+ recdblist.addLog(pout, txt, u"MP4Box追加ログ-詳細")\r
+ recdblist.addCommandLogZip(pout, "MP4Box_add", "mp4box_add", cmdn, txt)\r
time.sleep(5)\r
- os.remove(ptoutb)\r
shutil.move(ptadd,padd)\r
- shutil.move(ptout,pout)\r
+ if os.path.exists(ptout):\r
+ shutil.move(ptout,pout)\r
+ os.remove(ptoutb)\r
+ else:\r
+ txtt=padd+u"のインポートエラー"\r
+ recdblist.addLog(pout, txtt, u"MP4Box追加ログ-コマンド")\r
+ shutil.move(ptoutb,pout)\r
time.sleep(5)\r
def execcomd(cmd):\r
txt=""\r
except:\r
""\r
return txt\r
-\r
-def deltitle(path,title):\r
- dp=[]\r
- dp.append(os.path.join(path,title+".avi"))\r
- dp.append(os.path.join(path,title+".264"))\r
- dp.append(os.path.join(path,title+".120.avi"))\r
- dp.append(os.path.join(path,title+".noodml.avi"))\r
- dp.append(os.path.join(path,title+".aac"))\r
- dp.append(os.path.join(path,title+".m2v"))\r
- dp.append(os.path.join(path,title+"_1.aac"))\r
- dp.append(os.path.join(path,title+"_2.aac"))\r
- dp.append(os.path.join(path,title+"_1.mp3"))\r
- dp.append(os.path.join(path,title+"_2.mp3"))\r
- dp.append(os.path.join(path,title+".srt"))\r
- if configreader.getenv("remove_ts")=="1":\r
- dp.append(os.path.join(path,title+".ts"))\r
- if os.path.exists(os.path.join(path,title+".ts"))and os.path.exists(os.path.join(path,title+".ts.b25")):\r
- if os.path.getsize(os.path.join(path,title+".ts"))*10>os.path.getsize(os.path.join(path,title+".ts.b25")):\r
- if os.path.getsize(os.path.join(path,title+".ts"))>1000*1000*100:\r
- dp.append(os.path.join(path,title+".ts.b25"))\r
- if os.path.exists(os.path.join(path,title+".mp4")):\r
- if os.path.getsize(os.path.join(path,title+".mp4"))>1000*1000*10:\r
- for ip in dp:\r
- try:\r
- os.remove(ip)\r
- except:\r
- ""\r