-#!/usr/bin/python
-# coding: UTF-8
-# Rec10 TS Recording Tools
-# Copyright (C) 2009 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
-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
- 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)
- rT=re.compile("(.+)_(\d*)\Z")
- tT=rT.match(title)
- tT2=rT.match(cmpp)
- if tT:
- ntitle=tT.group(0)
- if tT2:
- cmpp=cmpp.group(0)
- 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 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"
- 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)")
- (opts,args)=parser.parse_args(sys.argv)
- if opts.ltitle!="":
- #print opts.ltitle
- 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)
- for t in sa:
- recdblist.printutf8(u"自動推測実行中")
- sf=searchFolder(t,recordedpath)
- if sf!="":
- recdblist.printutf8(u"移動先")
- recdblist.printutf8(t+" : "+os.path.join(sf, destNameMKV(t, recpath,sf)+".mkv"))
- recdblist.printutf8(u"実行中")
- execMove(t, recpath, recordedpath)
- else:
- recdblist.printutf8(t+" can't find matching folder.")
- elif opts.list:
- sa=search_mkv(recpath, recordedpath)
- for t in sa:
- sf=searchFolder(t,recordedpath)
- if sf!="":
- recdblist.printutf8(t+u" : "+os.path.join(sf, destNameMKV(t, recpath,sf)+".mkv"))
- elif opts.etitle!="":
- etitle=unicode(opts.etitle,'utf-8')
- execMove(etitle,recpath, recordedpath)
-
+#!/usr/bin/python\r
+# coding: UTF-8\r
+# Rec10 TS Recording Tools\r
+# Copyright (C) 2009-2011 Yukikaze\r
+\r
+import os\r
+import os.path\r
+import optparse\r
+import sys\r
+\r
+import configreader\r
+import recdblist\r
+import auto_move\r
+import guess\r
+recordedpath=unicode(configreader.getConfPath("recorded"),'utf-8')\r
+recpath=unicode(configreader.getConfPath("recpath"),'utf-8')\r
+\r
+if __name__ == "__main__":\r
+ usage="usage: %prog read -h"\r
+ version="%prog 0.9.4"\r
+ parser=optparse.OptionParser(usage=usage,version=version)\r
+ 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)")\r
+ parser.add_option("-A","--Auto",action="store_true",dest="auto",default=False,help="auto classifying mode(not a test)")\r
+ parser.add_option("-D","--Delete",action="store_true",dest="delete",default=False,help="auto delete tempfile mode(not a test)")\r
+ parser.add_option("-e","--exec",action="store",type="string",dest="etitle",default="",metavar="TITLE",help="exec move(not a test)")\r
+ parser.add_option("-l","--list",action="store_true",dest="list",default=False,help="File listing mode(test for -A)")\r
+ parser.add_option("-t","--tssearch",action="store_true",dest="tssearch",default=False,help="Ts move auto search")\r
+ parser.add_option("-T","--Tsmove",action="store_true",dest="tsmove",default=False,help="Ts auto move.")\r
+ parser.add_option("-S","--SeriesNum",action="store",type="string",dest="seriespath",default="",metavar="TITLE",help="Search Series Number in the path.")\r
+ #parser.add_option("-f","--filenumlist",action="store_true",dest="filenumlist",default=False,help="Guess Program Number.")\r
+\r
+ (opts,args)=parser.parse_args(sys.argv)\r
+ if opts.ltitle!="":##-sの場合\r
+ ltitle=unicode(opts.ltitle,'utf-8')\r
+ sf=guess.searchFolder(ltitle, recordedpath)\r
+ print "###MKV###"\r
+ recdblist.printutf8(os.path.join(sf, auto_move.getMoveDestpath(ltitle,recpath,sf,".mkv")+".mkv"),verbose_level=100)\r
+ print "\n"\r
+ print "###MP4###"\r
+ recdblist.printutf8(os.path.join(sf, auto_move.getMoveDestpath(ltitle,recpath,sf,".mp4")+".mp4"),verbose_level=100)\r
+ print "\n"\r
+ elif opts.auto:##-A の場合\r
+ sa=auto_move.searchFile(recpath, recordedpath, ".mkv")\r
+ for t in sa:\r
+ recdblist.printutf8(u"自動推測実行中-MKV",verbose_level=100)\r
+ sf=guess.searchFolder(t,recordedpath)\r
+ if sf!="":\r
+ recdblist.printutf8(u"移動先",verbose_level=100)\r
+ recdblist.printutf8(t+" : "+os.path.join(sf, auto_move.getMoveDestpath(t, recpath,sf,".mkv")+".mkv"),verbose_level=100)\r
+ recdblist.printutf8(u"実行中",verbose_level=100)\r
+ auto_move.execMove(t, recpath, recordedpath,".mkv",1)\r
+ else:\r
+ recdblist.printutf8(t+" can't find matching folder",verbose_level=100)\r
+ sa=auto_move.searchFile(recpath, recordedpath, ".mp4")\r
+ for t in sa:\r
+ recdblist.printutf8(u"自動推測実行中-MP4",verbose_level=100)\r
+ sf=guess.searchFolder(t,recordedpath)\r
+ if sf!="":\r
+ recdblist.printutf8(u"移動先",verbose_level=100)\r
+ recdblist.printutf8(t+" : "+os.path.join(sf, auto_move.getMoveDestpath(t, recpath,sf,".mp4")+".mp4"),verbose_level=100)\r
+ recdblist.printutf8(u"実行中",verbose_level=100)\r
+ auto_move.execMove(t, recpath, recordedpath,".mp4",1)\r
+ else:\r
+ recdblist.printutf8(t+" can't find matching folder",verbose_level=100)\r
+ elif opts.delete:##-Dの場合\r
+ sa=auto_move.searchFile(recpath, recordedpath,".mkv")\r
+ for t in sa:\r
+ auto_move.execDelete(t, recpath)\r
+ sa=auto_move.searchFile(recpath, recordedpath,".mp4")\r
+ for t in sa:\r
+ auto_move.execDelete(t, recpath)\r
+ elif opts.list:##-lの場合\r
+ sa=auto_move.searchFile(recpath, recordedpath,".mkv")\r
+ for t in sa:\r
+ sf=guess.searchFolder(t,recordedpath)\r
+ if sf!="":\r
+ recdblist.printutf8(t+u" : "+os.path.join(sf, auto_move.getMoveDestpath(t, recpath,sf,".mkv")+".mkv"),verbose_level=100)\r
+ sa=auto_move.searchFile(recpath, recordedpath,".mp4")\r
+ for t in sa:\r
+ sf=guess.searchFolder(t,recordedpath)\r
+ if sf!="":\r
+ recdblist.printutf8(t+u" : "+os.path.join(sf, auto_move.getMoveDestpath(t, recpath,sf,".mp4")+".mp4"),verbose_level=100)\r
+ elif opts.etitle!="":\r
+ if os.path.exists(os.path.join(recpath, etitle+".mkv")):\r
+ auto_move.execMove(etitle,recpath, recordedpath,".mkv",1)\r
+ elif os.path.exists(os.path.join(recpath, etitle+".mp4")):\r
+ auto_move.execMove(etitle,recpath, recordedpath,".mp4",1)\r
+ elif opts.tssearch:## -tの場合\r
+ tsmovepath=""\r
+ try:\r
+ tsmovepath=unicode(configreader.getConfPath("ts_movepath"),'utf-8')\r
+ except:\r
+ ""\r
+ if tsmovepath!="":\r
+ sa=auto_move.searchFile(recpath, tsmovepath,".ts")\r
+ for t in sa:\r
+ sf=guess.searchFolder(t,tsmovepath,700)\r
+ if sf!="":\r
+ recdblist.printutf8(t+u" : "+os.path.join(sf, auto_move.getMoveDestpath(t, recpath,sf,".ts")+".ts"))\r
+ print "\n"\r
+ sa=auto_move.searchFile(recpath, tsmovepath,".ts")\r
+ else:\r
+ recdblist.printutf8(u"ts_movepathが設定されていません。/etc/rec10.confを設定してください。",verbose_level=100)\r
+ elif opts.tsmove:##-Tの場合\r
+ tsmovepath=""\r
+ try:\r
+ tsmovepath=unicode(configreader.getConfPath("ts_movepath"),'utf-8')\r
+ except:\r
+ ""\r
+ if tsmovepath!="":\r
+ sa=auto_move.searchFile(recpath, tsmovepath,".ts")\r
+ for t in sa:\r
+ sf=guess.searchFolder(t,tsmovepath,700)\r
+ if sf!="":\r
+ recdblist.printutf8(u"移動先",verbose_level=100)\r
+ recdblist.printutf8(t+" : "+os.path.join(sf, auto_move.getMoveDestpath(t, recpath,sf,".ts")+".ts"),verbose_level=100)\r
+ recdblist.printutf8(u"実行中",verbose_level=100)\r
+ auto_move.execMove(t, recpath, tsmovepath,".ts",0)\r
+ sa=auto_move.searchFile(recpath, tsmovepath,".ts")\r
+ else:\r
+ recdblist.printutf8(u"ts_movepathが設定されていません。/etc/rec10.confを設定してください。",verbose_level=100)\r
+ elif opts.seriespath!="":##-Sの場合##与えられたパスにシリーズがそろっているかを調べる。\r
+ ss=guess.detSeriesNum(opts.seriespath)\r
+ for sstitle, ssv in ss.iteritems():\r
+ if len(ssv)>0:\r
+ maxt=0\r
+ bt=0\r
+ alr=[]\r
+ for ssi,ssiv in ssv.iteritems():\r
+ alr.append(ssi)\r
+ if maxt<ssi:\r
+ maxt=ssi\r
+ alr=list(set(alr))\r
+ alr.sort()\r
+ alr.reverse()\r
+ nuke=""\r
+ for i in xrange(1,maxt+1,1):\r
+ if alr.count(i)==0:\r
+ nuke=nuke+str(i)+","\r
+ if len(nuke)==0:\r
+ recdblist.printutf8(sstitle+" "+str(maxt))\r
+ else:\r
+ recdblist.printutf8(u"**"+sstitle+" |"+str(maxt)+"| "+nuke)\r
OSDN Git Service
