Python——gzip(待完善)

'''
解压单个文件夹的单个.gz文件
'''
#! /bin/env python3
import gzip

def un_gz(filename):
    #获取文件名称，去掉后缀名
    f_name = filename.replace(".gz","")
    #开始解压
    g_file = gzip.GzipFile(filename)
    #读取解压后的文件，并写入去掉后缀名的同名文件（即得到解压后的文件）
    open(f_name,"wb+").write(g_file.read())
    g_file.close()

un_gz('path') #单个文件夹中某一个压缩文件的地址

'''
解压某个文件夹下的所有.gz文件
'''
'''
#path = ''
def process_dataset(path):
    if os.path.exists(path):
        # os.system('cd path')
        # path = os.getcwd()
        dirs = os.listdir(path)
        # print(dirs) ['.gz','.gz','.gz'......]
        for dir in dirs:
            if '.gz' in dir:
                # print(dir)   ['.gz']
                filename = dir.replace(".gz","")
                gzip_file = gzip.GzipFile(path+dir)
                # print(gzip_file)  <...>
                # print(filename)   .ent .ent.gz
                with open(path+filename,'wb+') as f:
                    f.write(gzip_file.read())

if __name__ == '__main__':
    process_dataset(path='')
'''
import glob
import os
import gzip

path='/home/yhm/D-protein/pdb'
file_lst=glob.glob(path+'/*')
filename_lst=[os.path.basename(i) for i in file_lst]
print(filename_lst)

#! /bin/env python3
import os
import gzip
path ='/home/yhm/D-protein/pdb'                     #包含压缩文件的文件夹路径
files = os.listdir(path)                            #通过os读入一系列文件夹
for filename in files:                              #循环读入一系列文件夹
    if '.gz' in filename:                           #如果'.gz'存在于文件夹中
        print('.gz')
    def un_gz(filename):                            #定义解压文件的函数
        f_name = filename.replace(".gz","")
        g_file = gzip.GzipFile(filename)
        open(f_name,"wb+").write(g_file,read())
        g_file.close()
    un_gz()