#!/usr/bin/env python
#coding=utf-8
'''
自動識別未轉(zhuǎn)換的UI文件并轉(zhuǎn)換為對應版本的py文件
    —— by Joy
'''
import os
 
# 此處的path為UI文件存放的路徑
path = r'D:\SVNzhangy\fast-transfer\src'
 
# 記錄轉(zhuǎn)換成功的文件數(shù)量
count = 0
for root, dirs, files in os.walk(path):
  filename = []
  for file in files:
    if file.endswith('.py'):
      filename2 = file.split('.')[0]
      filename.append(filename2)
 
  version = input("請輸入1 or 2 確認pyqt轉(zhuǎn)換版本(1 - pyqt4,2 - pyqt5):")
 
  for file in files:
    if file.endswith('.ui'):
      filename1 = file.split('.')[0]
      filename4 = filename1 + "_qt4"
      filename5 = filename1 + "_qt5"
      try:
        if "1" == str(version):
          if filename.count(filename4) == 0:
            os.system('pyuic4 -o %s.py %s.ui -d'%(path + '\\' + filename4, path+'\\'+filename1))
            count+=1
        if "2" == str(version):
          if filename.count(filename5) == 0:
            os.system('pyuic5 -o %s.py %s.ui -d' % (path + '\\' + filename5, path + '\\' + filename1))
            count += 1
      except Exception as e:
        print "文件轉(zhuǎn)化錯誤，請檢查是否正確安裝指令對應PyQt"
        raise e
 
  if count == 0:
    print "暫無需要轉(zhuǎn)換的UI文件"
  else:
    print '轉(zhuǎn)換完畢，請查收！本次共轉(zhuǎn)換 ' + str(count) + ' 個文件'