ch04: initial setup

JBSP-code · JBSP-code · commit 489e4b7ad759 · 2023-01-22T17:51:29.000+01:00
diff --git a/.gitignore b/.gitignore
@@ -11,7 +11,6 @@ tex2pdf*
 .coverage
 .idea
 .vscode
-04_jump_the_five/jump.py
 05_howler/howler.py
 06_wc/wc.py
 07_gashlycrumb/gashlycrumb.py
diff --git a/04_jump_the_five/jump.py b/04_jump_the_five/jump.py
@@ -0,0 +1,72 @@
+#!/usr/bin/env python3
+"""
+Author : Anonymous <Anonymous@localhost>
+Date   : 2023-01-22
+Purpose: Rock the Casbah
+"""
+
+import argparse
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description='Rock the Casbah',
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+
+    parser.add_argument('positional',
+                        metavar='str',
+                        help='A positional argument')
+
+    parser.add_argument('-a',
+                        '--arg',
+                        help='A named string argument',
+                        metavar='str',
+                        type=str,
+                        default='')
+
+    parser.add_argument('-i',
+                        '--int',
+                        help='A named integer argument',
+                        metavar='int',
+                        type=int,
+                        default=0)
+
+    parser.add_argument('-f',
+                        '--file',
+                        help='A readable file',
+                        metavar='FILE',
+                        type=argparse.FileType('rt'),
+                        default=None)
+
+    parser.add_argument('-o',
+                        '--on',
+                        help='A boolean flag',
+                        action='store_true')
+
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+    str_arg = args.arg
+    int_arg = args.int
+    file_arg = args.file
+    flag_arg = args.on
+    pos_arg = args.positional
+
+    print(f'str_arg = "{str_arg}"')
+    print(f'int_arg = "{int_arg}"')
+    print('file_arg = "{}"'.format(file_arg.name if file_arg else ''))
+    print(f'flag_arg = "{flag_arg}"')
+    print(f'positional = "{pos_arg}"')
+
+
+# --------------------------------------------------
+if __name__ == '__main__':
+    main()
diff --git a/04_jump_the_five/test.py b/04_jump_the_five/test.py
@@ -4,7 +4,7 @@
 import os
 from subprocess import getstatusoutput
 
-prg = './jump.py'
+prg = "jump.py"
 
 
 # --------------------------------------------------
@@ -18,19 +18,19 @@ def test_exists():
 def test_usage():
     """usage"""
 
-    for flag in ['-h', '--help']:
-        rv, out = getstatusoutput(f'{prg} {flag}')
+    for flag in ["-h", "--help"]:
+        rv, out = getstatusoutput(f"{prg} {flag}")
         assert rv == 0
-        assert out.lower().startswith('usage')
+        assert out.lower().startswith("usage")
 
 
 # --------------------------------------------------
 def test_01():
     """test"""
 
-    rv, out = getstatusoutput(f'{prg} 123-456-7890')
+    rv, out = getstatusoutput(f"{prg} 123-456-7890")
     assert rv == 0
-    assert out == '987-604-3215'
+    assert out == "987-604-3215"
 
 
 # --------------------------------------------------
@@ -39,4 +39,4 @@ def test_02():
 
     rv, out = getstatusoutput(f'{prg} "That number to call is 098-765-4321."')
     assert rv == 0
-    assert out.rstrip() == 'That number to call is 512-340-6789.'
+    assert out.rstrip() == "That number to call is 512-340-6789."
