Merge pull request #3 from JBSP-code/chapter_03

JBSP-code · web-flow · commit b2e641b7040e · 2023-01-22T17:22:20.000+01:00
Chapter 03
diff --git a/.gitignore b/.gitignore
@@ -11,7 +11,6 @@ tex2pdf*
 .coverage
 .idea
 .vscode
-03_picnic/picnic.py
 04_jump_the_five/jump.py
 05_howler/howler.py
 06_wc/wc.py
diff --git a/03_picnic/picnic.py b/03_picnic/picnic.py
@@ -0,0 +1,51 @@
+#!/usr/bin/env python3
+"""
+Author : Jeffrey Schmid-Paz
+Date   : 2023-01-05
+Purpose: Create a list of items to bring along a picnic
+"""
+
+import argparse
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Picnic game",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    parser.add_argument("item", metavar="str", help="Item(s) to bring", nargs="+")
+
+    parser.add_argument("-s", "--sorted", help="Sort the items", action="store_true")
+
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+    items = args.item
+
+    if args.sorted == True:
+        items.sort()
+
+    items_to_bring = ""
+    if len(items) >= 3:
+        items.insert(-1, "and ")
+        items_to_bring = ", ".join(items[:-1]) + items[-1]
+    elif len(items) == 2:
+        items_to_bring = " and ".join(items)
+    else:
+        items_to_bring = items[0]
+
+    print(f"You are bringing {items_to_bring}.")
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()
diff --git a/03_picnic/test.py b/03_picnic/test.py
@@ -4,7 +4,7 @@
 import os
 from subprocess import getoutput
 
-prg = './picnic.py'
+prg = "picnic.py"
 
 
 # --------------------------------------------------
@@ -18,51 +18,52 @@ def test_exists():
 def test_usage():
     """usage"""
 
-    for flag in ['', '-h', '--help']:
-        out = getoutput(f'{prg} {flag}')
-        assert out.lower().startswith('usage')
+    for flag in ["", "-h", "--help"]:
+        out = getoutput(f"{prg} {flag}")
+        assert out.lower().startswith("usage")
 
 
 # --------------------------------------------------
 def test_one():
     """one item"""
 
-    out = getoutput(f'{prg} chips')
-    assert out.strip() == 'You are bringing chips.'
+    out = getoutput(f"{prg} chips")
+    assert out.strip() == "You are bringing chips."
 
 
 # --------------------------------------------------
 def test_two():
     """two items"""
 
     out = getoutput(f'{prg} soda "french fries"')
-    assert out.strip() == 'You are bringing soda and french fries.'
+    assert out.strip() == "You are bringing soda and french fries."
 
 
 # --------------------------------------------------
 def test_more_than_two():
     """more than two items"""
 
     arg = '"potato chips" coleslaw cupcakes "French silk pie"'
-    out = getoutput(f'{prg} {arg}')
-    expected = ('You are bringing potato chips, coleslaw, '
-                'cupcakes, and French silk pie.')
+    out = getoutput(f"{prg} {arg}")
+    expected = (
+        "You are bringing potato chips, coleslaw, " "cupcakes, and French silk pie."
+    )
     assert out.strip() == expected
 
 
 # --------------------------------------------------
 def test_two_sorted():
     """two items sorted output"""
 
-    out = getoutput(f'{prg} -s soda candy')
-    assert out.strip() == 'You are bringing candy and soda.'
+    out = getoutput(f"{prg} -s soda candy")
+    assert out.strip() == "You are bringing candy and soda."
 
 
 # --------------------------------------------------
 def test_more_than_two_sorted():
     """more than two items sorted output"""
 
-    arg = 'bananas apples dates cherries'
-    out = getoutput(f'{prg} {arg} --sorted')
-    expected = ('You are bringing apples, bananas, cherries, and dates.')
+    arg = "bananas apples dates cherries"
+    out = getoutput(f"{prg} {arg} --sorted")
+    expected = "You are bringing apples, bananas, cherries, and dates."
     assert out.strip() == expected
