使用python批量獲取excel的sheet名稱

這個腳本的重用是批量讀取excel並獲取每個excel的所有sheet名輸出到一個文件中。

環境：python 3.7.3

# -*- coding:utf-8 -*-

'''
本代碼的目的是獲取多個excl的sheet名，並輸出到指定文件中
'''

import sys
import xlrd
import os
import logging


# 設置logging.basicConfig()方法的參數和配置logging.basicConfig函數
FORMAT = '[%(funcName)s: %(lineno)d]: %(message)s'
LEVEL = logging.INFO
logging.basicConfig(level = LEVEL, format=FORMAT)

# 保存輸出sheet name
output_file = './SheetNameList.txt'

# init function
def get_obj_list(dir_name):
    filelist = os.listdir(dir_name)
    for item in filelist :
        item = dir_name + item
        if os.path.isfile(item) and ( item[-5:] == '.xlsx' or item[-5:] == '.xlsm'):
            if item.find("$") != -1 or item.find("csr_example") != -1 :
                continue
            GetSheetNameList(item)
        elif os.path.isdir(item):    
            item = item + '/'
            new_obj_list = []
            new_obj_list = get_obj_list(item)


# get sheet name and save in output_file
def GetSheetNameList(excelName):
    fd = open(output_file, 'a+')
    fd.write(excelName + '\n')
    excelfd = xlrd.open_workbook(excelName)
    for sheetName in excelfd.sheet_names():
        fd.write(sheetName + '\n')
    fd.close()


if __name__ == "__main__":
    if os.path.exists(output_file):
        os.remove(output_file)
    get_obj_list('./excel/')

文件結構：