cmd/repeat.py
Ethan Dalool 27463a1711 Update repeat.py, delete forever.py.
forever can be achieved by repeat x inf | do_cmd.
2020-12-09 07:26:36 -08:00

49 lines
1.1 KiB
Python

'''
Repeat the input as many times as you want.
> repeat "hello" 8
> echo hi | repeat !i 4
'''
import argparse
import sys
from voussoirkit import clipext
from voussoirkit import pipeable
def repeat_argparse(args):
text = clipext.resolve(args.text)
if args.times == 'inf':
try:
while True:
print(text)
except KeyboardInterrupt:
return 0
else:
try:
times = int(args.times)
except ValueError:
pipeable.stderr('times should be an integer >= 1.')
return 1
if times < 1:
pipeable.stderr('times should be >= 1.')
return 1
try:
for t in range(times):
print(text)
except KeyboardInterrupt:
return 1
def main(argv):
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('text')
parser.add_argument('times')
parser.set_defaults(func=repeat_argparse)
args = parser.parse_args(argv)
return args.func(args)
if __name__ == '__main__':
raise SystemExit(main(sys.argv[1:]))