#!/usr/bin/env python
""" 
Extract database from installation CD
"""
import os
import sys
import os.path
import shutil
from subprocess import call
import zipfile

'''Check files'''
if not os.path.exists('./i12comp.exe') or not os.path.exists('./data2.cab') or not os.path.exists('./data1.hdr'):

  print  "######################################################"
  print  "#                                                    #"
  print  "#  First place me and i12comp.exe in one folder with #"
  print  "#  data2.cab and data1.hdr like in example below.    #"
  print  "#  Then run me again and I extract them for you      #"
  print  "#                                                    #"
  print  "#    <DIR>          ..                               #"
  print  "#    <DIR>          pyren                            #"
  print  "#    <DIR>          .                                #"
  print  "#             1,190 extrdata.bat                     #"
  print  "#           135,168 i12comp.exe                      #"
  print  "#     1,009,293,010 data2.cab                        #"
  print  "#         2,989,301 data1.hdr                        #"
  print  "#                                                    #"
  print  "######################################################"

 
''' call extract tool  '''
print "Unpacking files. It may take some minutes"
call('i12comp.exe x -o -d -f data2.cab DiagOnCa*'.split(' '))  
call('i12comp.exe x -o -d -f data2.cab *.xml'.split(' '))  
call('i12comp.exe x -o -d -f data2.cab *.bqm'.split(' '))  
call('i12comp.exe x -o -d -f data2.cab *.dat'.split(' '))  
call('i12comp.exe x -o -d -f data2.cab *.zip'.split(' '))  


for root, dirs, files in os.walk("[Applicatif BORNEO_CLIP]NS_NSR_PL_1___[TARGETDIR]\CLIP\Data\GenAppli"):
  targ = root.replace("[Applicatif BORNEO_CLIP]NS_NSR_PL_1___[TARGETDIR]\CLIP\Data\GenAppli",".")
  if   targ[-3:-1]=='__': targ=targ[:-3]
  elif targ[-4:-2]=='__': targ=targ[:-4]

  if len(targ)>0 and not os.path.exists(targ):
    os.makedirs(targ)  

  for fil in files:
    src = root+'\\'+fil
    dst = targ+'\\'+fil[:-4]+fil[-4:].lower()
    shutil.copyfile(src,dst)
    print dst

for root, dirs, files in os.walk("[MTC_CLIP_X91]NS_NSR_NPL_1___[TARGETDIR]__1"):
  targ = root.replace("[MTC_CLIP_X91]NS_NSR_NPL_1___[TARGETDIR]__1",".")

  for fil in files:
    src = root+'\\'+fil
    dst = targ+'\\'+fil[:-4]+fil[-4:].lower()
    shutil.copyfile(src,dst)
    print dst

for root, dirs, files in os.walk("[MTC_CLIP_X91]NS_NSR_NPL_1___[TARGETDIR]"):
  targ = root.replace("[MTC_CLIP_X91]NS_NSR_NPL_1___[TARGETDIR]",".\BVMEXTRACTION")
  print targ
  if   targ[-3:-1]=='__': targ=targ[:-3]
  elif targ[-4:-2]=='__': targ=targ[:-4]

  if len(targ)>0 and not os.path.exists(targ):
    os.makedirs(targ)  

  for fil in files:
    src = root+'\\'+fil
    dst = targ+'\\'+fil[:-4]+fil[-4:].lower()
    shutil.copyfile(src,dst)
    print dst

print "Extracting BVM"
zipfile.ZipFile(".\BvmConfig.zip").extractall('.\\')
shutil.move(".\BVM_CONFIG", ".\BVMEXTRACTION")


print "Deliting unused files"
os.system('rmdir "./[Applicatif BORNEO_CLIP_RSM]NS_NSR_PL_1___[TARGETDIR]" /S /Q') 
os.system('rmdir "./[Applicatif BORNEO_CLIP_X91]NS_NSR_PL_2___[WINDIR]__1" /S /Q') 
os.system('rmdir "./[Applicatif BORNEO_CLIP]NS_NSR_NPL_544___[TARGETDIR]" /S /Q') 
os.system('rmdir "./[Applicatif BORNEO_CLIP_X91]NS_NSR_PL_1___[TARGETDIR]" /S /Q') 
os.system('rmdir "./[CONFIG_AUTO_FCT_X91]NS_NSR_NPL_544___[TARGETDIR]" /S /Q') 
os.system('rmdir "./[DocDiag_CLIP_X91]NS_NSR_PL_1___[TARGETDIR]__1" /S /Q') 
os.system('rmdir "./[Update Agent]NS_NSR_PL_1___[TARGETDIR]" /S /Q') 
os.system('rmdir "./[Applicatif BORNEO_CLIP]NS_NSR_PL_1___[TARGETDIR]" /S /Q') 
os.system('rmdir "./[MTC_CLIP_X91]NS_NSR_NPL_1___[TARGETDIR]" /S /Q')
os.system('rmdir "./[MTC_CLIP_X91]NS_NSR_NPL_1___[TARGETDIR]__1" /S /Q')
os.system('rmdir "./BL" /S /Q')
os.system('del "./BvmConfig.zip"')

print "Done"