python_Excel_xlwt

xlwt

创建excel，向excel写入数据，并保存数据

安装

推荐方法：

通过pip 安装，方便简洁，如下图所示：

导入

import xlrd

创建workbook（即excel）

book = Workbook(encoding='utf-8')#create a workbook

创建sheet

sheet = book.add_sheet('Sheet1')#create a sheet

设置样式

#set style
font = xlwt.Font() # 字体
font.name = 'Times New Roman'
font.bold = True
font.underline = False
font.italic = False
style = xlwt.XFStyle() # 创建一个格式
style.font = font # 设置格式字体

往单元格写入

sheet.write(0, 0, "no",style)#第1行，第1列
sheet.write(0, 1, "file_name",style)#第1行，第2列
sheet.write(0, 2, "file_version",style)#第1行，第3列

保存

book.save(save_path)

#coding=utf-8
'''
Created on 2018年11月12日

@author: yanerfree

get the version number of the file (.dll/.exe)  in bulk
批量获取文件版本号，适合量少的，因为是单线程
'''
import os
import sys
import win32api
import xlwt
from xlwt import *
import time
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *


#测试调试输出开关,正式发布需调整为False
mytest = True
mytest = False

class GetFileVersionNo(QWidget):
    def __init__(self):
        super().__init__()
        self.setupUi()
        self.myList = []#save result
        self.status = 0#记录目前是否在获取版本状态下
        
    def setupUi(self):
        
        glayout = QGridLayout()
        glayout.setSpacing(10)
        self.button1 = QPushButton("需要获取版本号的文件路径")
        self.button1.clicked.connect(self.open_dir)
        self.button2 = QPushButton("选择文件保存位置")
        self.button2.clicked.connect(self.save_file_dialog)
        self.lineEdit1 = QLineEdit()
        self.lineEdit2 = QLineEdit()
        self.button3 = QPushButton("获取版本号")
        #self.button3.setCheckable(True)
        self.button3.clicked.connect(self.getVer)
        self.textEdit1 = QTextEdit()
        
        self.label1 = QLabel(" 选择文件类型：")
        self.combobox1 = QComboBox()
        self.combobox1.addItem("exe")
        self.combobox1.addItem("dll")
        self.combobox1.currentTextChanged.connect(self.textEdit1.clear)
        glayout.addWidget(self.button1,1,1)
        glayout.addWidget(self.button2,2,1)
        glayout.addWidget(self.label1,3,1)
        glayout.addWidget(self.combobox1,3,2)
        glayout.addWidget(self.button3,4,1)
        glayout.addWidget(self.lineEdit1,1,2,1,2)
        glayout.addWidget(self.lineEdit2,2,2,1,2)
        glayout.addWidget(self.textEdit1,4,2,10,2)
        
        self.setGeometry(20,50,500,450)
        self.setWindowTitle("GetFileVersionNo")#设置窗体标题
        
        self.setLayout(glayout)
        
    def getVer(self):
        if self.status == 0:
            self.textEdit1.clear()
            self.status = 1#状态设置为1，获取版本中
            self.changestatus()
            file_path = self.lineEdit1.text()
            if not os.path.exists(file_path):
                self.showMsg("warning","请选择正确的路径")
                return
            
            save_path = self.lineEdit2.text()
            if save_path == "":
                self.showMsg("warning","请选择结果保存路径")
                return
            tmp1 = str(save_path).split("/")[-1]
            #print("tmp:",tmp1)
            tmp2 = str(save_path).split(tmp1)[0]
            #print("tmp:",tmp2)
            if not os.path.exists(tmp2):
                self.showMsg("warning","请选择正确的保存路径")
                return
            
            self.writeToExcel(file_path,save_path)
            self.status = 0
            self.changestatus()
        else:
            self.showMsg("warning", "正在获取版本号，请等待...")
            
    def changestatus(self):
        if self.status == 0:
            self.myList=[]
            self.button1.setEnabled(True)
            self.button2.setEnabled(True)
            self.button3.setEnabled(True)
            
        else:
            self.button1.setEnabled(False)
            self.button2.setEnabled(False)
            self.button3.setEnabled(False)
            
    def open_dir(self):
        dir_path=QFileDialog.getExistingDirectory(self,"choose directory",r"C:\")
        if not os.path.exists(dir_path):
            return -1
        #dir_path = self.dir_path.replace('/','\')#windows下需要进行文件分隔符转换
        
        #将获取的路径写入lineedit中
        self.lineEdit1.setText(dir_path)
        return (dir_path)
        
    def save_file_dialog(self):
        save_fileName, ok2 = QFileDialog.getSaveFileName(self,
                            "文件保存",
                            r"C:\",
                            "Text Files (*.xls);;All Files (*)")
        #print("save_fileName，ok2:",save_fileName,ok2)
        #将获取的路径写入lineedit中
        self.lineEdit2.setText(save_fileName)
        
        return(save_fileName)
    
    def showMsg(self,t,msg):
        if(t=="warning"):
            QMessageBox.information(self,"warning",msg,QMessageBox.Yes,QMessageBox.Yes)
            self.status = 0#状态设置为1，获取版本中
            self.changestatus()
            
    def traverse_dir(self,file_path):
        #traverse the directory of file_path（the file are .dll）
        #myList=[]#save result,如果该函数递归调用，则不能在此处保存结果，
        #否则每次递归都会清楚之前保存的数据
        try:
            filelist = os.listdir(file_path)
            for i in range(0,len(filelist)):
                print(i,filelist[i])
                tmp_path = os.path.join(file_path,filelist[i])
                tmp_path = tmp_path.replace('/','\')#windows下需要进行文件分隔符转换
                print("tmp_path:",tmp_path)
                if os.path.isfile(tmp_path):
                    #if str(list[i]).split(".")[1] =="dll":
                    #if str(filelist[i])[-3:] == "exe":
                    print("self.combobox1.currentText():",self.combobox1.currentText())
                    #print("str(filelist[i])[-3:]:",str(filelist[i])[-3:])
                    if str(filelist[i])[-3:] == self.combobox1.currentText():
                        #judge if the filename ended with  ".dll"
                        #print tmp_path,getFileVersion(tmp_path)
                        print ("add to mylis",(filelist[i],self.getFileVersion(tmp_path)))
                        self.myList.append((filelist[i],self.getFileVersion(tmp_path)))
                        
                        #将内容更新到textedit中去
                        content = ""
                        for item in self.myList:
                            content = content + "%s %s
"%(item[0],item[1])
                        self.textEdit1.setText(content)
                else:
                    self.traverse_dir(tmp_path)
        except:
            self.showMsg("warning","未知错误")
            self.myList = []
             
        return self.myList
    
    def getFileVersion(self,file_name):
        #get the version of file
        try:
            info = win32api.GetFileVersionInfo(file_name, os.sep)
            ms = info['FileVersionMS']
            ls = info['FileVersionLS']
            version = '%d.%d.%d.%04d' % (win32api.HIWORD(ms), win32api.LOWORD(ms), win32api.HIWORD(ls), win32api.LOWORD(ls))
            print("version:",version)
        except:
            version = " "
        
        return version
    

    def writeToExcel(self,file_path,save_path):
        #file_path = self.file_path
        #save_path = self.save_path
        #write to excel
        print("create a workbook")
        book = Workbook(encoding='utf-8')#create a workbook
        sheet = book.add_sheet('Sheet1')#create a sheet
        #set style
        font = xlwt.Font() # 字体
        font.name = 'Times New Roman'
        font.bold = True
        font.underline = False
        font.italic = False
        style = xlwt.XFStyle() # 创建一个格式
        style.font = font # 设置格式字体
        
        #sheet.write(0, 0, label = 'Formatted value', style) # Apply the Style to the Cell
        sheet.write(0, 0, "no",style)
        sheet.write(0, 1, "file_name",style)
        sheet.write(0, 2, "file_version",style)
        list = self.traverse_dir(file_path)
        
        row=0
        num=0
        for item in list:
            row += 1
            num += 1
            print(item[0] ,item[1])
            sheet.write(row, 0, num, style)
            sheet.write(row, 1,item[0] , style)
            sheet.write(row, 2,item[1] , style)
            
        book.save(save_path)

if __name__ == '__main__':
    app = QApplication(sys.argv)
    demo = GetFileVersionNo()
    demo.show()
    sys.exit(app.exec_())