07

overtune · overtune · commit 5ec7566a4770 · 2020-08-04T12:51:09.000+02:00
diff --git a/06_wc/wc.py b/06_wc/wc.py
@@ -1 +1,52 @@
-The quick brown fox jumps over the lazy dog.
+#!/usr/bin/env python3
+"""Emulate wc (word count)"""
+
+import argparse
+import sys
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description='Emulate wc (word count)',
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+
+    parser.add_argument('file',
+                        metavar='FILE',
+                        nargs='*',
+                        default=[sys.stdin],
+                        type=argparse.FileType('rt'),
+                        help='Input file(s)')
+
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def main():
+    """Start things up"""
+
+    args = get_args()
+
+    total_lines, total_bytes, total_words = 0, 0, 0
+    for fh in args.file:
+        num_lines, num_words, num_bytes = 0, 0, 0
+        for line in fh:
+            num_lines += 1
+            num_bytes += len(line)
+            num_words += len(line.split())
+
+        total_lines += num_lines
+        total_bytes += num_bytes
+        total_words += num_words
+
+        print(f'{num_lines:8}{num_words:8}{num_bytes:8} {fh.name}')
+
+    if len(args.file) > 1:
+        print(f'{total_lines:8}{total_words:8}{total_bytes:8} total')
+
+
+# --------------------------------------------------
+if __name__ == '__main__':
+    main()
diff --git a/07_gashlycrumb/gashlycrumb.py b/07_gashlycrumb/gashlycrumb.py
@@ -0,0 +1,51 @@
+#!/usr/bin/env python3
+"""
+Author : Johan Runesson <info@johanrunesson.se>
+Date   : 2020-08-04
+Purpose: Gashlycrumb
+"""
+
+import argparse
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description='Gashlycrumb',
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+
+    parser.add_argument('letter',
+                        metavar='letter',
+                        nargs='+',
+                        help='Letter(s)')
+
+    parser.add_argument('-f',
+                        '--file',
+                        help='Input file',
+                        metavar='FILE',
+                        type=argparse.FileType('rt'),
+                        default='gashlycrumb.txt')
+
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def main():
+    """Start things up"""
+
+    args = get_args()
+    dictionary = { line[0].lower(): line.rstrip() for line in args.file }
+    # dictionary = {}
+
+    # for line in args.file:
+    #     dictionary[line[0].lower()] = line.rstrip()
+
+    for letter in args.letter:
+        print(dictionary.get(letter.lower(), f'I do not know "{letter}".'))
+
+
+# --------------------------------------------------
+if __name__ == '__main__':
+    main()
diff --git a/NOTES.md b/NOTES.md
@@ -100,3 +100,9 @@ Handle output to file if specified else to print:
 ```
 out_fh = open(args.outfile, 'wt') if args.outfile else sys.stdout
 ```
+
+Create a dictionary with a loop:
+
+```
+lookup = { line[0].upper(): line.rstrip() for line in fh }
+```
