zip ファイル内に含まれる特定の CSV ファイルを明示して読み取る例

zipcsv.py

#!/usr/bin/env python3

"""
zip ファイル内に含まれる特定の CSV ファイルを明示して読み取る例
"""

import argparse
import csv
import io
import zipfile


def main(args):
    with zipfile.ZipFile(args.zipfile) as zf:
        if args.csvfile not in zf.namelist():
            print(f"{args.csvfile} not found in {args.zipfile.name}")
            return
        with zf.open(args.csvfile) as f:
            reader = csv.reader(io.TextIOWrapper(f, encoding=args.csv_encoding))
            for row in reader:
                print(row)


def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("zipfile", type=argparse.FileType("rb"))
    parser.add_argument("csvfile")
    parser.add_argument("--csv-encoding", default="utf-8")
    return parser.parse_args()


if __name__ == "__main__":
    main(parse_args())