Add lines_to_files.py.

This commit is contained in:
voussoir 2020-11-15 18:00:09 -08:00
parent 3c061f8daa
commit 9ba6511050

35
lines_to_files.py Normal file
View file

@ -0,0 +1,35 @@
import argparse
import sys
from voussoirkit import gentools
from voussoirkit import pathclass
def linestofiles(file, lines_per_file):
file = pathclass.Path(file)
basename = file.replace_extension('').basename
basename = basename + '_{index}'
folder = file.parent
handle = file.open('r', encoding='utf-8')
chunks = gentools.chunk_generator(handle, lines_per_file)
for (index, chunk) in enumerate(chunks):
chunk = ''.join(chunk)
chunk_file = folder.with_child(basename.format(index=index)).add_extension(file.extension)
print(chunk_file)
chunk_file.open('w', encoding='utf-8').write(chunk)
handle.close()
def linestofiles_argparse(args):
return linestofiles(args.file, args.lines_per_file)
def main(argv):
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('file')
parser.add_argument('lines_per_file', type=int)
parser.set_defaults(func=linestofiles_argparse)
args = parser.parse_args(argv)
return args.func(args)
if __name__ == '__main__':
raise SystemExit(main(sys.argv[1:]))