Update linenumbers.py.

linenumbers.py

@@ -1,12 +1,28 @@
+import argparse
 import sys
 
-from voussoirkit import clipext
+from voussoirkit import pipeable
 
-if len(sys.argv) == 1:
+def linenumbers_argparse(args):
-    sys.argv.append('!i')
+    lines = pipeable.input(args.source, read_files=True)
-text = clipext.resolve(sys.argv[1])
+    if args.lazy:
-lines = text.splitlines()
+        form = '{no} | {line}'
-digits = len(str(len(lines)))
+    else:
-form = '{no:>0%d} | {line}' % digits
+        lines = list(lines)
-for (index, line) in enumerate(lines):
+        digits = len(str(len(lines)))
-    print(form.format(no=index+1, line=line))
+        form = '{no:>0%d} | {line}' % digits
+    for (index, line) in enumerate(lines):
+        print(form.format(no=index+1, line=line))
+
+def main(argv):
+    parser = argparse.ArgumentParser(description=__doc__)
+
+    parser.add_argument('source')
+    parser.add_argument('--lazy', dest='lazy', action='store_true')
+    parser.set_defaults(func=linenumbers_argparse)
+
+    args = parser.parse_args(argv)
+    return args.func(args)
+
+if __name__ == '__main__':
+    raise SystemExit(main(sys.argv[1:]))