# Rec10 TS Recording Tools
# Copyright (C) 2009-2010 Yukikaze
-import glob
-import n_gram
import os
import os.path
-import re
-import time
import optparse
import sys
-import shutil
import configreader
import recdblist
+import auto_move
+import guess
recordedpath=unicode(configreader.getpath("recorded"),'utf-8')
recpath=unicode(configreader.getpath("recpath"),'utf-8')
-def search_b25(path):
- """
- 録画フォルダを検索し、各QUEの実行状況を表示する
- """
- b25list = glob.glob(path + "/*.b25")
- for b25f in b25list:
- ##b25f is title.ts.b25 avi is title.avi
- dir = os.path.split(b25f)[0]
- title = os.path.split(b25f)[1]
- title = title.replace(".ts.b25", "")
- avipath = os.path.join(dir, title + ".avi")
- mp4path = os.path.join(dir, title + ".mp4")
- tspath = os.path.join(dir, title + ".ts")
- b25f = b25f.replace(".ts.b25", ".avi")
- mode = "ts"
- if os.path.isfile(tspath):##tsファイルが存在している
- dtime = time.time()-os.path.getmtime(tspath)
- dtime = int(dtime)
- if dtime > 120:
- if os.path.getsize(tspath) > 1 * 1000 * 1000:##最終更新から22分以上経過かつ1MB以上
- mode = "avi"
- else:
- mode = "tsmiss"
- else:
- mode = "ts"
- if os.path.isfile(mp4path):##tsファイルが存在している
- dtime = time.time()-os.path.getmtime(mkvpath)
- dtime = int(dtime)
- if dtime > 120:
- if os.path.getsize(avipath) > 1 * 1000 * 1000:##最終更新から22分以上経過かつ1MB以上
- mode = "fin"
- else:
- mode = "avimiss"
- else:
- mode = "avi"
- recdblist.printutf8(title + ":" + mode)
-def search_avi(temppath,recpath):
- """
- 録画一時フォルダ内のaviファイルを検索
- """
- avilist = glob.glob(temppath + "/*.avi")
- ret=[]
- for avif in avilist:
- ##b25f is title.ts.b25 avi is title.avi
- dir = os.path.split(avif)[0]
- title = os.path.split(avif)[1]
- title = title.replace(".avi", "")
- avipath = os.path.join(dir, title + ".avi")
- dtime = time.time()-os.path.getmtime(avipath)
- dtime = int(dtime)
- if dtime > 300:
- if veryfySize(avipath):
- ret.append(title)
- return ret
-def search_mkv(temppath,recpath):
- """
- 録画一時フォルダ内mkvファイルを検索
- """
- avilist = glob.glob(temppath + "/*.mkv")
- ret=[]
- for avif in avilist:
- ##b25f is title.ts.b25 avi is title.avi
- dir = os.path.split(avif)[0]
- title = os.path.split(avif)[1]
- title = title.replace(".mkv", "")
- avipath = os.path.join(dir, title + ".mkv")
- dtime = time.time()-os.path.getmtime(avipath)
- dtime = int(dtime)
- if dtime > 300:
- if veryfySize(avipath):
- ret.append(title)
- return ret
-def search_mp4(temppath,recpath):
- """
- 録画一時フォルダ内mp4ファイルを検索
- """
- avilist = glob.glob(temppath + "/*.mp4")
- ret=[]
- for avif in avilist:
- ##b25f is title.ts.b25 avi is title.avi
- dir = os.path.split(avif)[0]
- title = os.path.split(avif)[1]
- title = title.replace(".mp4", "")
- avipath = os.path.join(dir, title + ".mp4")
- dtime = time.time()-os.path.getmtime(avipath)
- dtime = int(dtime)
- if dtime > 300:
- if veryfySize(avipath):
- ret.append(title)
- return ret
-def veryfySize(path):
- #vsize=[297,497,596,1196]#SD 30m 1h 1.5h 2h
- vsize=[245,275,295,591,830]
- vsize=vsize+[325,449,560,590,602,690,805,860,1014,1138,1237]
- vsize=vsize+[261,535,540,616,740]#HD 30m 1h
- #vsize=vsize+[381,895,447]
- ret = 0
- for size in vsize:
- if os.path.getsize(path)>(size-10)*1024*1024 and os.path.getsize(path)<(size+10)*1024*1024:
- ret=1
- if os.path.getsize(path)>270*1024*1024:
- ret=1
- return ret
-def searchFolder(title,path):
- """
- titleにマッチするフォルダを探し出す。
- """
- folderpath=os.listdir(path)
- lfpath=[]
- ngram=[]
- for ft in folderpath:
- fullpath=os.path.join(path.encode('utf-8'), ft)
- if os.path.isdir(fullpath):
- lfpath.append(fullpath)
- ftt=os.listdir(fullpath)
- if len(ftt)>0:
- for ft2 in ftt:
- try:
- folderpath.append(os.path.join(fullpath, ft2))
- except Exception, inst:
- #print type(inst)
- #print inst
- ""
- else:
- lfpath.append(fullpath)
- for dirp in lfpath:
- cmpp=""
- appp=""
- ntitle=title
- if os.path.isdir(dirp):
- cmpp=os.path.dirname(dirp)
- appp=dirp
- else:
- cmpp=os.path.basename(dirp)
- appp=os.path.dirname(dirp)
- ntitle=getTitle(title)
- cmpp=getTitle(cmpp)
- p=n_gram.trigram(ntitle,cmpp)
- if p>0:
- ngram.append((p,appp))
- ngram=list(set(ngram))
- ngram.sort()
- ngram.reverse()
- if len(ngram)>0:
- #recdblist.printutf8(title + ngram[0][1] + " : "+str(ngram[0][0]))
- if ngram[0][0]>400:
- return ngram[0][1]
- else:
- return ""
- else:
- return ""
-def getTitle(title):
- rT=re.compile("(.+)_(\d*)\Z")
- tT=rT.match(title)
- rT2=re.compile("(.+)_(.+)_(\d*)\Z")#_(aichi)_2010-02-06T01:59:00.mkv
- tT2=rT2.match(title)
- rT3=re.compile("(.+)_(.+)_\d-\d-\dT\d:\d:\d\Z")#_(aichi)_2010-02-06T01:59:00.mkv
- tT3=rT3.match(title)
- ntitle=title
- if tT :
- ntitle=tT.group(0)
- elif tT2:
- ntitle=tT3.group(0)
- elif tT3:
- ntitle=tT4.group(0)
- return ntitle
-def execMove(title,temppath,recpath):
- srcpath=os.path.join(temppath,title+".mkv")
- #desttitle=destName(title, temppath, recpath)
-
- sf=searchFolder(title, recpath)
- if sf!="":
- destpath=os.path.join(sf,destNameMKV(title, temppath, sf)+".mkv")
- #os.path.join(sf,desttitle+".mkv")
- #recdblist.printutf8(srcpath)
- #recdblist.printutf8(destpath)
- recdblist.printutf8("moving now..")
- recdblist.printutf8(srcpath+" : "+destpath)
- print srcpath
- shutil.move(srcpath, destpath)
- #shutil.copy(srcpath, destpath)
- delpath=[os.path.join(temppath,title+".ts")]
- delpath.append(os.path.join(temppath,title+".avi"))
- delpath.append(os.path.join(temppath,title+".120.avi"))
- delpath.append(os.path.join(temppath,title+".timecode.txt"))
- delpath.append(os.path.join(temppath,title+".aac"))
- delpath.append(os.path.join(temppath,title+".ts.b25"))
- delpath.append(os.path.join(temppath,title+".ts.tsmix"))
- delpath.append(os.path.join(temppath,title+".ts.log"))
- delpath.append(os.path.join(temppath,title+".sa.avi"))
- delpath.append(os.path.join(temppath,title+".sa.avi.log"))
- delpath.append(os.path.join(temppath,title+".log"))
- for dp in delpath:
- try:
- os.remove(dp)
- ""
- except:
- ""
-def destName(title,temppath,recpath):
- """
- titleから移動先に作るべきファイル名を作り出す。
- """
- dstpath=os.path.join(recpath,title+".avi")
- srcpath=os.path.join(temppath,title+".avi")
- if os.path.exists(dstpath):
- gmtime=time.gmtime(os.path.getmtime(srcpath))
- times=time.strftime("%y%m%d",gmtime)
- title=title+"_"+times
- return title
-def destNameMKV(title,temppath,recpath):
- """
- titleから移動先に作るべきファイル名を作り出す。
- """
- dstpath=os.path.join(recpath,title+".mkv")
- srcpath=os.path.join(temppath,title+".mkv")
- if os.path.exists(dstpath):
- gmtime=time.gmtime(os.path.getmtime(srcpath))
- times=time.strftime("%y%m%d",gmtime)
- title=title+"_"+times
- return title
-def destNameMP4(title,temppath,recpath):
- """
- titleから移動先に作るべきファイル名を作り出す。
- """
- dstpath=os.path.join(recpath,title+".mp4")
- srcpath=os.path.join(temppath,title+".mp4")
- if os.path.exists(dstpath):
- gmtime=time.gmtime(os.path.getmtime(srcpath))
- times=time.strftime("%y%m%d",gmtime)
- title=title+"_"+times
- return title
-def detName(path,title):
- """
- type A ---title#<number>
- type B ---title#<number>subtitle
- type C ---title subtitle
- type D ---title(without number)
- """
- #if re.match("#\d\s[0,10]\z|#[0-9]\s[0,10]\z", title)
- # recdblist.printutf8("typeA")
- #elif re.match("#\d\s[0,10].|#[0-9]\s[0,10].", title)
- # recdblist.printutf8("typeB")
-def detNameType(title):
- """
- type A ---title#<number>
- type B ---title#<number>subtitle
- type C ---title subtitle
- type D ---title(without number)
- """
- recdblist.printutf8(title)
- #rA=re.compile(".+(?P<title>)#\d(?P<num>)\s[0,10]\z")
- rA=re.compile("(.+)#(\d*)\s*\Z")
- tA=rA.match(title)
- rB=re.compile("(.+)#(\d*)\s*(\D*)")
- tB=rB.match(title)
- if tA:
- recdblist.printutf8("typeA")
- recdblist.printutf8("title="+tA.group(1))
- recdblist.printutf8("num="+tA.group(2))
- elif tB:
- recdblist.printutf8("typeB")
- recdblist.printutf8("title="+tB.group(1))
- recdblist.printutf8("num="+tB.group(2))
- recdblist.printutf8("subtitle="+tB.group(3))
if __name__ == "__main__":
usage="usage: %prog read -h"
- version="%prog 0.7.0beta1"
+ version="%prog 0.9.4"
parser=optparse.OptionParser(usage=usage,version=version)
parser.add_option("-s","--search",action="store",type="string",dest="ltitle",default="",metavar="TITLE",help="test to search where the title should be contained(test for -e)")
parser.add_option("-A","--Auto",action="store_true",dest="auto",default=False,help="auto classifying mode(not a test)")
parser.add_option("-e","--exec",action="store",type="string",dest="etitle",default="",metavar="TITLE",help="exec move(not a test)")
- parser.add_option("-l","--list",action="store_true",dest="list",default=False,help="File listing mode(test for -a)")
+ parser.add_option("-l","--list",action="store_true",dest="list",default=False,help="File listing mode(test for -A)")
+ parser.add_option("-t","--tssearch",action="store_true",dest="tssearch",default=False,help="Ts move auto search")
+ parser.add_option("-T","--Tsmove",action="store_true",dest="tsmove",default=False,help="Ts auto move.")
+ #parser.add_option("-f","--filenumlist",action="store_true",dest="filenumlist",default=False,help="Guess Program Number.")
+
(opts,args)=parser.parse_args(sys.argv)
- if opts.ltitle!="":
- #print opts.ltitle
+ if opts.ltitle!="":##-sの場合
ltitle=unicode(opts.ltitle,'utf-8')
- sf=searchFolder(ltitle, recordedpath)
- recdblist.printutf8(os.path.join(sf, destNameMKV(ltitle,recpath,sf)+".mkv"))
- #print os.path.join(sf, destNameMP4(ltitle,recpath,sf)+".mp4")
- elif opts.auto:
- sa=search_mkv(recpath, recordedpath)
+ sf=guess.searchFolder(ltitle, recordedpath)
+ print "###MKV###"
+ recdblist.printutf8(os.path.join(sf, auto_move.get_move_dest_path(ltitle,recpath,sf,".mkv")+".mkv"))
+ print "\n"
+ print "###MP4###"
+ recdblist.printutf8(os.path.join(sf, auto_move.get_move_dest_path(ltitle,recpath,sf,".mp4")+".mp4"))
+ print "\n"
+ elif opts.auto:##-A の場合
+ sa=auto_move.search_file(recpath, recordedpath, ".mkv")
+ for t in sa:
+ recdblist.printutf8(u"自動推測実行中-MKV")
+ sf=guess.searchFolder(t,recordedpath)
+ if sf!="":
+ recdblist.printutf8(u"移動先")
+ recdblist.printutf8(t+" : "+os.path.join(sf, auto_move.get_move_dest_path(t, recpath,sf,".mkv")+".mkv"))
+ recdblist.printutf8(u"実行中")
+ auto_move.execMove(t, recpath, recordedpath,".mkv",1)
+ else:
+ recdblist.printutf8(t+" can't find matching folder")
+ sa=auto_move.search_file(recpath, recordedpath, ".mp4")
for t in sa:
- recdblist.printutf8(u"自動推測実行中")
- sf=searchFolder(t,recordedpath)
+ recdblist.printutf8(u"自動推測実行中-MP4")
+ sf=guess.searchFolder(t,recordedpath)
if sf!="":
recdblist.printutf8(u"移動先")
- recdblist.printutf8(t+" : "+os.path.join(sf, destNameMKV(t, recpath,sf)+".mkv"))
+ recdblist.printutf8(t+" : "+os.path.join(sf, auto_move.get_move_dest_path(t, recpath,sf,".mp4")+".mp4"))
recdblist.printutf8(u"実行中")
- execMove(t, recpath, recordedpath)
+ auto_move.execMove(t, recpath, recordedpath,".mp4",1)
else:
- recdblist.printutf8(t+" can't find matching folder.")
- elif opts.list:
- sa=search_mkv(recpath, recordedpath)
+ recdblist.printutf8(t+" can't find matching folder")
+ elif opts.list:##-lの場合
+ sa=auto_move.search_file(recpath, recordedpath,".mkv")
+ for t in sa:
+ sf=guess.searchFolder(t,recordedpath)
+ if sf!="":
+ recdblist.printutf8(t+u" : "+os.path.join(sf, auto_move.get_move_dest_path(t, recpath,sf,".mkv")+".mkv"))
+ sa=auto_move.search_file(recpath, recordedpath,".mp4")
for t in sa:
- sf=searchFolder(t,recordedpath)
+ sf=guess.searchFolder(t,recordedpath)
if sf!="":
- recdblist.printutf8(t+u" : "+os.path.join(sf, destNameMKV(t, recpath,sf)+".mkv"))
+ recdblist.printutf8(t+u" : "+os.path.join(sf, auto_move.get_move_dest_path(t, recpath,sf,".mp4")+".mp4"))
elif opts.etitle!="":
- etitle=unicode(opts.etitle,'utf-8')
- execMove(etitle,recpath, recordedpath)
-
+ if os.path.exists(os.path.join(recpath, etitle+".mkv")):
+ auto_move.execMove(etitle,recpath, recordedpath,".mkv",1)
+ elif os.path.exists(os.path.join(recpath, etitle+".mp4")):
+ auto_move.execMove(etitle,recpath, recordedpath,".mp4",1)
+ elif opts.tssearch:## -tの場合
+ tsmovepath=""
+ try:
+ tsmovepath=unicode(configreader.getpath("ts_movepath"),'utf-8')
+ except:
+ ""
+ if tsmovepath!="":
+ sa=auto_move.search_file(recpath, tsmovepath,".ts")
+ for t in sa:
+ sf=guess.searchFolder(t,tsmovepath,700)
+ if sf!="":
+ recdblist.printutf8(t+u" : "+os.path.join(sf, auto_move.get_move_dest_path(t, recpath,sf,".ts")+".ts"))
+ print "\n"
+ sa=auto_move.search_file(recpath, tsmovepath,".ts")
+ else:
+ recdblist.printutf8(u"ts_movepathが設定されていません。/etc/rec10.confを設定してください。")
+ elif opts.tsmove:##-Tの場合
+ tsmovepath=""
+ try:
+ tsmovepath=unicode(configreader.getpath("ts_movepath"),'utf-8')
+ except:
+ ""
+ if tsmovepath!="":
+ sa=auto_move.search_file(recpath, tsmovepath,".ts")
+ for t in sa:
+ sf=guess.searchFolder(t,tsmovepath,700)
+ if sf!="":
+ recdblist.printutf8(u"移動先")
+ recdblist.printutf8(t+" : "+os.path.join(sf, auto_move.get_move_dest_path(t, recpath,sf,".ts")+".ts"))
+ recdblist.printutf8(u"実行中")
+ auto_move.execMove(t, recpath, tsmovepath,".ts",0)
+ sa=auto_move.search_file(recpath, tsmovepath,".ts")
+ else:
+ recdblist.printutf8(u"ts_movepathが設定されていません。/etc/rec10.confを設定してください。")
\ No newline at end of file