rec10/trunk/src/install.py

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