dotfiles/bin/fitwidth.py

63 lines
1.4 KiB
Python
Raw Normal View History

2022-11-09 14:08:23 -07:00
#!/usr/bin/env python3
import argparse, sys
# Parse args
parser = argparse.ArgumentParser(prog="FitWidth v1.0.0",
description='Fit text (stdin|file) to a given character width');
2022-11-13 22:47:06 -07:00
parser.add_argument(
'-t', '--truncate',
action='store_true',
help='Truncate longer lines with a "..."',
);
parser.add_argument(
'-w', '--wrap',
action='store_true',
help='Wrap longer lines across two lines with a ">" at the end',
);
parser.add_argument(
'width', type=int,
help='Wrap longer lines across two lines with a ">" at the end',
);
parser.add_argument(
'file', nargs='?',
type=argparse.FileType('r'),
help="File to read from, instead of stdin",
);
2022-11-09 14:08:23 -07:00
args = parser.parse_args();
# Validate args
if not args.truncate and not args.wrap:
args.truncate = True
if args.width < 10:
print("Minimum width is 10")
exit(1);
if args.file is not None:
file = args.file
elif not sys.stdin.isatty():
file = sys.stdin
else:
print("Please provide a file or pipe in text to fit")
exit(1)
# Print out file
w = args.width
for line in file:
line = line.rstrip()
if len(line) > w:
if args.truncate:
print(line[:w-3] + "...")
else:
i = w-1
print(line[:i] + ">")
while i < len(line):
print(">" + line[i:i+w-1])
i += w-1
else:
print(line)
file.close()