ezeetester
diff --git a/‎bin/compile.py‎
Lines changed: 12 additions & 23 deletions b/‎bin/compile.py‎
Lines changed: 12 additions & 23 deletions
diff --git a/‎bin/contents.txt‎
Lines changed: 1 addition & 1 deletion b/‎bin/contents.txt‎
Lines changed: 1 addition & 1 deletion
diff --git a/‎bin/tablify.py‎
Lines changed: 114 additions & 0 deletions b/‎bin/tablify.py‎
Lines changed: 114 additions & 0 deletions
@@ -21,17 +21,20 @@ def get_args():
         description='Compile my book',
         formatter_class=argparse.ArgumentDefaultsHelpFormatter)
 
-    # parser.add_argument('positional',
-    #                     metavar='str',
-    #                     help='A positional argument')
-
     parser.add_argument('-i',
                         '--dir',
                         help='Input dir',
                         metavar='str',
                         type=str,
                         default=os.getcwd())
 
+    parser.add_argument('-f',
+                        '--outfile',
+                        help='Output filename',
+                        metavar='str',
+                        type=str,
+                        default='playful_python.pdf')
+
     parser.add_argument('-o',
                         '--outdir',
                         help='Output dir',
@@ -46,18 +49,6 @@ def get_args():
                         type=str,
                         default=None)
 
-    # parser.add_argument('-i',
-    #                     '--int',
-    #                     help='A named integer argument',
-    #                     metavar='int',
-    #                     type=int,
-    #                     default=0)
-
-    # parser.add_argument('-f',
-    #                     '--flag',
-    #                     help='A boolean flag',
-    #                     action='store_true')
-
     return parser.parse_args()
 
 
@@ -80,9 +71,6 @@ def main():
     book_file = os.path.join(out_dir, 'book.md')
 
     with open(book_file, 'wt') as fh:
-        # fh.write('\n'.join(
-        #     ['\\documentclass{article}', '\\begin{document}',
-        #      '\\tableofcontents', '\\newpage', '']))
         fh.write('\\setcounter{tocdepth}{1}\\tableofcontents\n\\newpage\n\n')
 
         top_readme = 'README.md'
@@ -102,25 +90,26 @@ def main():
                 print('\tREADME')
                 chapter = 'Chapter {}: '.format(i)
                 text = open(readme).read()
-                text = re.sub('^#\s+', '# ' + chapter, text)
+                text = re.sub(r'^#\s+', '# ' + chapter, text)
                 fh.write(text + '\n\\newpage\n\n')
 
             solution = os.path.join(in_dir, dir_name, 'solution.py')
             if os.path.isfile(solution):
                 print('\tSOLUTION')
-                fh.write('## Solution\n\n'.format(dir_name))
+                fh.write('## Solution\n\n')
                 fh.write('````\n')
                 numbered = getoutput('cat -n {}'.format(solution))
                 fh.write(numbered)
                 fh.write('\n````\n')
                 fh.write('\n\\newpage\n\n')
 
-    cmd = 'pandoc {} --pdf-engine=xelatex -o book.pdf'
-    rv, out = getstatusoutput(cmd.format(book_file))
+    cmd = 'pandoc {} --pdf-engine=xelatex -o {}'
+    rv, out = getstatusoutput(cmd.format(book_file, args.outfile))
 
     if rv != 0:
         die('Error: {}'.format(out))
 
+
 # --------------------------------------------------
 if __name__ == '__main__':
     main()
@@ -18,7 +18,7 @@ family_tree
 gematria
 histy
 guess
-kentucky_fryer
+kentucky_friar
 mad_libs
 license_plates
 mad_libs
 
@@ -0,0 +1,114 @@
+#!/usr/bin/env python3
+"""
+Author : Ken Youens-Clark <kyclark@gmail.com>
+Date   : 2019-05-09
+Purpose: Diplay delimited text files as ASCII tables
+"""
+
+import argparse
+import csv
+import os
+import sys
+from tabulate import tabulate
+
+
+# --------------------------------------------------
+def get_args():
+    """get command-line arguments"""
+    parser = argparse.ArgumentParser(
+        description='Diplay delimited text files as ASCII tables',
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+
+    parser.add_argument(
+        'file',
+        metavar='FILE',
+        nargs='+',
+        type=argparse.FileType('r', encoding='UTF-8'),
+        help='Input file(s)')
+
+    parser.add_argument(
+        '-d',
+        '--delimiter',
+        help='Delimiter',
+        metavar='str',
+        type=str,
+        default='')
+
+    parser.add_argument(
+        '-s',
+        '--style',
+        help='Tabulate table style',
+        metavar='str',
+        type=str,
+        choices=[
+            'plain', 'simple', 'github', 'grid', 'fancy_grid', 'pipe',
+            'orgtbl', 'jira', 'presto', 'psql', 'rst', 'mediawiki', 'moinmoin',
+            'youtrack', 'html', 'latex', 'latex_raw', 'latex_booktabs',
+            'textile'
+        ],
+        default='psql')
+
+    parser.add_argument(
+        '-l',
+        '--limit',
+        help='Limit display to number records',
+        metavar='int',
+        type=int,
+        default=0)
+
+    parser.add_argument(
+        '-n', '--no_headers', help='No headers in file', action='store_true')
+
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def warn(msg):
+    """Print a message to STDERR"""
+    print(msg, file=sys.stderr)
+
+
+# --------------------------------------------------
+def die(msg='Something bad happened'):
+    """warn() and exit with error"""
+    warn(msg)
+    sys.exit(1)
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+    args = get_args()
+
+    file_hdr = '/ ' + ('*' * 10) + ' {} ' + ('*' * 10) + ' /'
+
+    for fh in args.file:
+        if len(args.file) > 1:
+            print(file_hdr.format(os.path.basename(fh.name)))
+
+        _, ext = os.path.splitext(fh.name)
+        delimiter = args.delimiter if args.delimiter else ',' \
+            if ext == '.csv' else '\t'
+        data = list(csv.reader(fh, delimiter=delimiter))
+
+        if not data:
+            warn('No data in "{}"'.format(fh.name))
+            continue
+
+        headers = []
+        if args.no_headers:
+            num_fields = len(data[0])
+            headers = map(lambda i: 'Field {}'.format(i + 1),
+                          range(num_fields))
+        else:
+            headers = data.pop(0)
+
+        if args.limit:
+            data = data[:args.limit]
+
+        print(tabulate(data, headers=headers, tablefmt=args.style))
+
+
+# --------------------------------------------------
+if __name__ == '__main__':
+    main()