Python-統計目錄（資料夾）中Excel檔案個數和資料量

#!/usr/bin/env python

# coding:utf-8

"""

@File Name: zwc_count_file.py

@Version: 1.0

@Python Version:  3.7

@Author: liguanbin

@Created Time: 2021/6/18 15:10

@Software: PyCharm

@Desc：

"""

import os

import xlrd

import time

global file_list

file_list = []

def count_filenum():

    input_path = input("請輸入目錄：")

    if not input_path.endswith('\\'):

        input_path = input_path + "\\"

    else:

        input_path = input_path

    start_time = time.strftime("%Y%m%d%H%M%S", time.localtime())

    log_file = os.path.join(input_path,"檔案統計_"+start_time[0:8]+".txt")

    for parent, dirtory, files in os.walk(input_path):

        for file in files:

            if file.endswith("xlsx") or file.endswith("xls"):

                file_path = os.path.join(parent,file)

                file_list.append(file_path)

    write_file(log_file, "【統計目錄】：" + input_path)

    write_file(log_file, "-" * 100)

    all_num = 0

    for excel in file_list:

        fh = xlrd.open_workbook(excel)

        # 開啟檔案

        sheet = fh.sheet_by_index(0)

        num = sheet.nrows

        all_num += num

        # 將每個檔案的記錄數下入檔案

        line_string = excel[len(input_path):len(excel)] + " : " + str(num)

        write_file(log_file,line_string)

    write_file(log_file, '-' * 100)

    write_file(log_file, "【目錄中的總檔案數】：" + str(len(file_list)))

    write_file(log_file, "【所有檔案的總記錄數】：" + str(all_num))

    write_file(log_file, '*' * 100)

def write_file(filename,content):

    file=open(filename,'a',encoding='utf-8')            #以追加方式開啟日誌檔案

    print(content)                                      #控制檯列印日誌

    file.writelines(content+'\n')                       #寫入內容

    file.close()

if __name__ == '__main__':

    count_filenum()