问题描述:
定义文件 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))










网友评论