统计文件个数及非负数之和

问题描述:

定义文件 xx.tar.gz 的产生方式如下:
• 以 xx 为文件名的文件通过 tar 和 gzip 打包压缩产生,该文件中以字符串的方式记录
了一个非负整数;
• 或者以 xx 为名的目录通过 tar 和 gzip 打包压缩产生,该目录中包含若干 xx.tar.gz。
其中,x ∈ [0, 9]。现给定一个根据上述定义生成的文件 00.tar.gz ,请确定其中包含的以 xx 为文件名的文件个数以及这些文件中所记录的非负整数之和。

import os
import tarfile

root_dir = "/Users/chelsea/Desktop/00"
result = 0
Num = 0

def unpack_path_file(parentdir, file_name):
    archive = tarfile.open(parentdir + "/" + file_name, "r:gz",encoding='gbk')
    for tarinfo in archive:
        archive.extract(tarinfo, parentdir)
    archive.close()

def search_folder(parent_dir):
    global result,Num
    for file_name in os.listdir(parent_dir):
        if os.path.isfile(parent_dir + "/" + file_name) and file_name.find(".tar.gz") != -1:
            unpack_path_file(parent_dir, file_name)
    for file_name in os.listdir(parent_dir):
        if os.path.isfile(parent_dir + "/" + file_name) and file_name.find(".tar.gz") == -1:
            temp_file = open(parent_dir + "/" + file_name,encoding='gbk',errors='ignore')
            number = temp_file.read()
            temp_file.close()
            result = result + int(number)
            Num = Num + 1
    for dir_name in os.listdir(parent_dir):
        if os.path.isdir(parent_dir + "/" + dir_name):
            search_folder(parent_dir + "/" + dir_name)
search_folder(root_dir)
print("Sum:" + str(result))
print("fileCount:" + str(Num))


推荐阅读更多精彩内容