Merge pull request #4 from JBSP-code/chapter_04

JBSP-code · web-flow · commit 8ad47ad36088 · 2023-01-23T19:08:36.000+01:00
Chapter 04
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,49 @@
+#!/usr/bin/env python3
+"""
+Author : Jeffrey Paz-Schmid
+Date   : 2023-01-22
+Purpose: Encode Phone-Numbers
+"""
+
+import argparse
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Jump the Five",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    parser.add_argument("text", metavar="str", help="Input text")
+
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+
+    jumper = {
+        "1": "9",
+        "2": "8",
+        "3": "7",
+        "4": "6",
+        "5": "0",
+        "6": "4",
+        "7": "3",
+        "8": "2",
+        "9": "1",
+        "0": "5",
+    }
+
+    print("".join([jumper.get(char, char) for char in args.text]))
+
+
+# --------------------------------------------------
+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."
