python unzipping files code example

Example 1: python extract zip file without directory structure

import os
import shutil
import zipfile

my_dir = r"D:\Download"
my_zip = r"D:\Download\my_file.zip"

with zipfile.ZipFile(my_zip) as zip_file:
    for member in zip_file.namelist():
        filename = os.path.basename(member)
        # skip directories
        if not filename:
            continue

        # copy file (taken from zipfile's extract)
        source = zip_file.open(member)
        target = open(os.path.join(my_dir, filename), "wb")
        with source, target:
            shutil.copyfileobj(source, target)

Example 2: extracting zip files python

import os
import optparse
import zipfile

def get_arguments():
    parser = optparse.OptionParser()
    parser.add_option("--src","--source",dest="paths", help="Removing file with extension .SRT and .VTT")
    (options,arguments)=parser.parse_args()
    if not options.paths:
        parser.error("[-] Please specify source, use --help for more info.")
    return options


def Source_folder(folder_path):
    print('[+] Extracting zip file')
    for path, dir_list, file_list in os.walk(folder_path):
        try:
            for file_name in file_list:
                if file_name.endswith(".zip"):
                    abs_file_path = os.path.join(path, file_name)
                    parent_path = os.path.split(abs_file_path)[0]
                    output_folder_name = os.path.splitext(abs_file_path)[0]
                    output_path = os.path.join(parent_path, output_folder_name)

                    zip_obj = zipfile.ZipFile(abs_file_path, 'r')
                    zip_obj.extractall(output_path)
                    zip_obj.close()

        except FileNotFoundError:
            print('Error', file_name)


options = get_arguments()
Source_folder(options.paths)