rec10/trunk/src/install.py

   1 #!/usr/bin/python
   2 # coding: UTF-8
   3 # Rec10 TS Recording Tools
   4 # Copyright (C) 2009-2010 Yukikaze
   5 import time
   6 import os.path
   7 import ConfigParser
   8 import sys
   9 import commands
  10 import configwriter
  11 import shutil
  12
  13 import recdblist
  14 def install():
  15     version = recdblist.version
  16     recdblist.printutf8(u"rec10の環境インストールを開始します。")
  17     time.sleep(1)
  18     recdblist.printutf8(u"環境インストール処理中・・・")
  19     path = str(os.path.dirname(os.path.abspath(__file__)))
  20     confp = ConfigParser.SafeConfigParser()
  21     Conf = 'rec10.conf'
  22     confpath=""
  23     if os.path.exists(os.path.join(path,Conf)):
  24         confpath=os.path.join(path,Conf)
  25     elif os.path.exists(os.path.join("/etc","rec10.conf")):
  26         confpath=os.path.join("/etc","rec10.conf")
  27     elif os.path.exists(os.path.join("/etc/rec10","rec10.conf")):
  28         confpath=os.path.join("/etc/rec10","rec10.conf")
  29     if confpath!="":
  30         confp.read(confpath)
  31     else:
  32         recdblist.printutf8(u"rec10.confが見つかりません。")
  33         recdblist.printutf8(u"このinstall.pyと同じフォルダに置くか、/etc/rec10.confもしくは/etc/rec10/rec10.confにおいてください。")
  34         sys.exit(1)
  35     recdblist.printutf8(u"設定ファイルの読み込みを確認//設定ファイルのパスは"+confpath)
  36     if confp.get('path',"recpath")=="/path of /recording":
  37         recdblist.printutf8(u"録画先のフォルダを設定してください(recpath=)")
  38         sys.exit(1)
  39     recdblist.printutf8(u"DB処理に入ります")
  40     if os.path.exists(os.path.join(path,"rec10d.py")):
  41         try:
  42             update_db_all()
  43         except Exception, inst:
  44             recdblist.printutf8(u"DB処理中にエラーが出ました。configファイルのDB設定を見直してください。")
  45             recdblist.printutf8(str(type(inst)))
  46             recdblist.printutf8(str(inst))
  47     recdblist.printutf8(u"チャンネル設定に入ります")
  48     if os.path.exists(os.path.join(path,"chlist.xml")):
  49         recdblist.printutf8(u"チャンネルリストが見つかりました。||chlist.xml")
  50     else:
  51         recdblist.printutf8(u"チャンネルリストが見つかりません。"+os.path.join(path,"chlist.xml")+u"を作成してください。")
  52         recdblist.printutf8(u"東京/名古屋/神戸のサンプルがchlist_sample_*.xmlにあるため、参考にしてください。")
  53         sys.exit(1)
  54     bsok=int(confp.get('env',"bs"))
  55     csok=int(confp.get('env',"cs"))
  56     import chdata
  57     chdata.new_chdata(bsok,csok)
  58     recdblist.printutf8(u"おめでとうございます　初期設定は完了しました。")
  59     recdblist.printutf8(u"rec10を実行するユーザーのcrontabに"+os.path.join(path,"rec10")+u"を追加してください(5分周期が目安)")
  60     import rec10d
  61     rec10d.rec10db.change_installed_in_status()
  62 def getpath(cmd):
  63     recdblist.printutf8(cmd+u"の場所を確認中")
  64     txt=commands.getoutput(u"which "+cmd)
  65     if len(txt.split(" "))>3:
  66         recdblist.printutf8(cmd+u"の場所を確認できませんでした。終了します")
  67         sys.exit(1)
  68     else:
  69         recdblist.printutf8(cmd+u" : "+txt)
  70         if len(txt.splitlines()):
  71             txt=txt.splitlines()[0]
  72         return txt.strip()
  73 def move():
  74     useB=0
  75     path = str(os.path.dirname(os.path.abspath(__file__)))
  76     recdblist.printutf8(u"rec10 ver 0.9.4 installer.")
  77     recdblist.printutf8(u"rec10のインストールを開始します。")
  78     maxn=14
  79     configwriter.setpath(u"wine", getpath(u"wine"))
  80     configwriter.setpath(u"recpt1", getpath(u"recpt1"))
  81     configwriter.setpath(u"x264", getpath(u"x264"))
  82     configwriter.setpath(u"mp4box", getpath(u"MP4Box"))
  83     configwriter.setpath(u"mencoder", getpath(u"mencoder"))
  84     configwriter.setpath(u"ffmpeg", getpath(u"ffmpeg"))
  85     #configwriter.setpath(u"xvfb-run", getpath(u"xvfb-run"))
  86     configwriter.setpath(u"mkvmerge", getpath(u"mkvmerge"))
  87     configwriter.setpath(u"b25", getpath(u"b25"))
  88     recdblist.printutf8(u"必須環境の設定が完了しました")
  89     recdblist.printutf8(u"個人設定に入ります")
  90     if os.path.exists("/etc/rec10.conf"):
  91         recdblist.printutf8(u"以前の設定ファイルが見つかりました。")
  92         recdblist.printutf8(u"インポートしますか？")
  93         useB=raw_input("[Y/n]:")
  94         if useB=="N" or useB == "n":
  95             useB=0
  96         else:
  97             useB=1
  98             recdblist.printutf8(u"設定ファイルのインポーを行います。")
  99     recdblist.printutf8(u"rec10の録画ファイルが置かれる場所を入力してください(1/"+str(maxn)+")")
 100     recdblist.printutf8(u"100GB程度は確保されている必要があります")
 101     recpath=getConf(useB,"/etc/rec10.conf","path","recpath","path : ")
 102     configwriter.setpath(u"recpath",recpath)
 103     recdblist.printutf8(u"DBの設定に入ります。")
 104     recdblist.printutf8(u"rec10が使用するMySQLのユーザーを設定してください。(2/"+str(maxn)+")")
 105     mysql_user=getConf(useB,"/etc/rec10.conf","db","mysql_user","mysql_user : ")
 106     configwriter.setdbpath("mysql_user", mysql_user)
 107     recdblist.printutf8(u"パスワードを設定してください(3/"+str(maxn)+")")
 108     mysql_passwd=getConf(useB,"/etc/rec10.conf","db","mysql_passwd","mysql_passwd : ")
 109     configwriter.setdbpath("mysql_passwd", mysql_passwd)
 110     recdblist.printutf8(u"同時録画可能数の設定に入ります")
 111     recdblist.printutf8(u"TE(地デジ)録画可能数(PT*だと2 白Friioだと1)(4/"+str(maxn)+")")
 112     te_max=str(int(getConf(useB,"/etc/rec10.conf","env","te_max","te_max : ")))
 113     configwriter.setenv("te_max", te_max)
 114     recdblist.printutf8(u"BS/CS110録画可能数(PT*だと2 黒Friioだと1)(5/"+str(maxn)+")")
 115     bscs_max=str(int(getConf(useB,"/etc/rec10.conf","env","bscs_max","bscs_max : ")))
 116     configwriter.setenv("bscs_max", bscs_max)
 117     recdblist.printutf8(u"同時エンコード最大数を設定してください(6/"+str(maxn)+")")
 118     recdblist.printutf8(u"CPUのコア数が目安です")
 119     enc_max=raw_input("[2]:")
 120     if enc_max=="":
 121         enc_max="2"
 122     else:
 123         enc_max=str(int(enc_max))
 124     configwriter.setenv("enc_max", enc_max)
 125     recdblist.printutf8(u"二カ国語放送/5.1ch放送の音声変換の設定です。\nNeroAACエンコーダーを使用しますか?(NeroAACEncを別途入手してtstoolsに入れてください。))")
 126     recdblist.printutf8(u"入手先:http://www.nero.com/jpn/technologies-aac-codec.html\n (7/"+str(maxn)+")")
 127     audioenc=raw_input("[y/N]:  ")
 128     if audioenc=="y" or audioenc=="Y":
 129         configwriter.setpath("useNeroAAC", "1")
 130         recdblist.printutf8(u"NeroAACEncのパスを指定してください")
 131         aacpath=getConf(useB,"/etc/rec10.conf","path","NeroAAC","[/usr/local/bin/neroAacEnc] : ")
 132         if aacpath.replace(" ","")=="":
 133             aacpath="/usr/local/bin/neroAacEnc"
 134         configwriter.setpath("NeroAAC",aacpath)
 135     else:
 136         configwriter.setpath("useNeroAAC", "0")
 137         recdblist.printutf8(u"lameを使用します。")
 138     recdblist.printutf8(u"保存する画質を教えてください(7/"+str(maxn)+")")
 139     recdblist.printutf8(u"x264のcrfの値です。標準だと30分で800MBほどになります")
 140     crf=getConf(useB,"/etc/rec10.conf","env","crf","[24] : ")
 141     if crf.replace(" ","")=="":
 142         crf="24"
 143     configwriter.setenv("crf", crf)
 144     recdblist.printutf8(u"アニメーションを保存する画質を教えてください(8/"+str(maxn)+")")
 145     recdblist.printutf8(u"x264のcrfの値です。標準だと30分で800MBほどになります")
 146     a_crf=getConf(useB,"/etc/rec10.conf","env","animation_crf","[20] : ")
 147     if a_crf.replace(" ","")=="":
 148         a_crf="20"
 149     configwriter.setenv("animation_crf", a_crf)
 150     recdblist.printutf8(u"x264の圧縮率を設定してください(8/"+str(maxn)+")")
 151     recdblist.printutf8(u"x264のpresetの値です。標準だとmedium、小さければ小さいほど高圧縮になります")
 152     preset=getConf(useB,"/etc/rec10.conf","env","x264_preset","[4] : ")
 153     if preset.replace(" ","")=="":
 154         preset="4"
 155     configwriter.setenv("x264_preset", preset)
 156     recdblist.printutf8(u"保存するコンテナの設定(9/"+str(maxn)+")")
 157     recdblist.printutf8(u"MP4を標準にしますか(もしくはMKVが使えます)")
 158     useMP4=raw_input("[y/N]:")
 159     if useMP4=="Y" or useMP4 == "y" :
 160         useMP4="1"
 161     else:
 162         useMP4="0"
 163     configwriter.setenv("make_mp4", useMP4)
 164     recdblist.printutf8(u"TSファイルの自動削除(10/"+str(maxn)+")")
 165     recdblist.printutf8(u"中間体のtsファイルを削除しますか")
 166     recdblist.printutf8(u"(削除するとやり直しができないため残しておくことを推奨します。)")
 167     removeTS=raw_input("[y/N]:")
 168     if removeTS=="Y" or removeTS == "y" :
 169         removeTS="1"
 170     else:
 171         removeTS="0"
 172     configwriter.setenv("remove_ts", removeTS)
 173     recdblist.printutf8(u"チャンネルの設定に入ります(11/"+str(maxn)+")")
 174     recdblist.printutf8(u"rec10を設置する地域を選んでください。")
 175     recdblist.printutf8(u"1:東京 2:神戸 3:名古屋 4:その他")
 176     ch=raw_input()
 177     ch=int(ch)
 178     if ch==1:
 179         try:
 180             shutil.copy(os.path.join(path,"chlist_sample_tokyo.xml"),os.path.join(path,"chlist.xml"))
 181         except:
 182             recdblist.printutf8(u"既存のchlis.xmlを上書きできないため、前のものを使います。")
 183     elif ch==2:
 184         try:
 185             shutil.copy(os.path.join(path,"chlist_sample_kobe.xml"),os.path.join(path,"chlist.xml"))
 186         except:
 187             recdblist.printutf8(u"既存のchlis.xmlを上書きできないため、前のものを使います。")
 188     elif ch==3:
 189         try:
 190             shutil.copy(os.path.join(path,"chlist_sample_nagoya.xml"),os.path.join(path,"chlist.xml"))
 191         except:
 192             recdblist.printutf8(u"既存のchlis.xmlを上書きできないため、前のものを使います。")
 193     elif ch==4:
 194         if not os.path.exists(os.path.join(path,"chlist.xml")):
 195             recdblist.printutf8(u"チャンネルの設定が必要です。")
 196             recdblist.printutf8(u"chlist_sample_*を参考にchlist.xmlを作成した後に次のステップに進んでください。")
 197             time.sleep(5)
 198     recdblist.printutf8(u"BS放送は受信可能ですか？(12/"+str(maxn)+")")
 199     bsok=raw_input("y/N:")
 200     if bsok=="Y" or bsok == "y" :
 201         bsok="1"
 202     else:
 203         bsok="0"
 204     configwriter.setenv("bs", bsok)
 205     recdblist.printutf8(u"CS放送(スカパーe2)は受信可能ですか？(13/"+str(maxn)+")")
 206     csok=raw_input("y/N:")
 207     if csok=="Y" or csok == "y" :
 208         csok="1"
 209         recdblist.printutf8(u"標準パックの番組は入力してありますがそれ以外は追加でchlist_cs.xmlに追加する必要があります。")
 210     else:
 211         csok="0"
 212     configwriter.setenv("cs", csok)
 213
 214     recdblist.printutf8(u"自機識別オプション(14/"+str(maxn)+")")
 215     recdblist.printutf8(u"本rec10の識別名を入力してください")
 216     recdblist.printutf8(u"複数台のrec10を運用する場合に効果的です。空白でもかまいません。")
 217     recdblist.printutf8(u"例 kobe01")
 218     iff=getConf(useB,"/etc/rec10.conf","env","iff","[] : ")
 219     configwriter.setenv("iff", iff)
 220     confp = ConfigParser.SafeConfigParser()
 221     confp.read("/etc/rec10.conf")
 222     if useB:
 223         configwriter.setpath("recorded",confp.get("path","recorded"))
 224         configwriter.setpath("move_destpath",confp.get("path","move_destpath"))
 225         configwriter.setpath("tmp",confp.get("path","tmp"))
 226         try:
 227             configwriter.setenv("x264_addline",confp.get("env","x264_addline"))
 228         except:
 229             configwriter.setenv("x264_addline","")
 230         if confp.get("path","b25_remote")=="":
 231             configwriter.setpath("b25_remote", "0")
 232         elif confp.get("path","b25_remote")=="1":
 233             configwriter.setpath("b25_remote", confp.get("path","b25_remote"))
 234             configwriter.setpath("b25", confp.get("path","b25"))
 235             configwriter.setpath("b25_env", confp.get("path","b25_env"))
 236         else:
 237             configwriter.setpath("b25_remote", confp.get("path","b25_remote"))
 238
 239     configwriter.writeconf()
 240     recdblist.printutf8(u"初期設定が終了しました。")
 241     recdblist.printutf8(u"makeを実行した後にrootにてmake installを実行するとインストールされます。")
 242 def getConf(useBeforeConf,Confpath,dbsection,key,defaultstr):
 243     tstr=defaultstr
 244     if os.path.exists(Confpath) and useBeforeConf==1:
 245         try:
 246             confp = ConfigParser.SafeConfigParser()
 247             confp.read("/etc/rec10.conf")
 248             tstr=u"["+confp.get(dbsection,key)+u"]:"
 249         except:
 250             tstr=defaultstr
 251     tmpconf=raw_input(tstr)
 252     if tmpconf.replace(" ","")=="" and os.path.exists(Confpath) and useBeforeConf==1 :
 253         confp = ConfigParser.SafeConfigParser()
 254         confp.read("/etc/rec10.conf")
 255         try:
 256             tmpconf=confp.get(dbsection,key)
 257         except:
 258             tmpconf=""
 259     return tmpconf
 260 def update_db_all():
 261     import rec10d
 262     tversion=0
 263     if rec10d.rec10db.select_installed_in_status()==1:
 264         if rec10d.rec10db.select_version_in_status()==0:
 265             tversion=0
 266             recdblist.printutf8(u"既存のDBが見つかりました。0.9.1と仮定してアップデート処理を行います。")
 267             update_db(0)
 268             time.sleep(1)
 269         else:
 270             recdblist.printutf8(u"既存のDBが見つかりました。アップデート処理を行います。")
 271         tversion=int(rec10d.rec10db.select_version_in_status())
 272         while (recdblist.version>tversion):
 273             update_db(tversion)
 274             tversion=int(rec10d.rec10db.select_version_in_status())
 275     else:
 276         rec10d.rec10db.drop_in_status()
 277         rec10d.rec10db.drop_in_settings()
 278         time.sleep(2)
 279         rec10d.rec10db.new_in_status()
 280         rec10d.rec10db.new_in_settings()
 281         rec10d.rec10db.change_version_in_status(recdblist.version)
 282 def update_db(version):
 283     import rec10d
 284     if version==0:
 285         rec10d.rec10db.update_db_to93()
 286     elif version==93:
 287         rec10d.rec10db.update_db_93to94()
 288 if __name__ == "__main__":
 289     move()