Dump the first N lines of CSV files inside a ZIP archive

ziphead.py

import sys
from zipfile import ZipFile

LINES = 10

def main(argv: list[str]) -> int:
    with ZipFile(argv[1]) as z:
        for fn in z.namelist():
            if not fn.endswith('.csv'):
                continue
            with z.open(fn) as f:
                print(f'==={fn}===')
                i = 0
                for l in f:
                    sys.stdout.write(l.decode('ascii'))
                    i += 1
                    if i >= LINES:
                        break

    return 0

if __name__ == '__main__':
    sys.exit(main(sys.argv))