Problem 03 Solution

Author: lukemcguire

03_picnic/picnic.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <[email protected]>
+Date   : 2024-06-17
+Purpose: List of items to bring on the picnic
+"""
+
+import argparse
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Get items to bring on the picnic",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    parser.add_argument(
+        "items",
+        nargs="+",
+        metavar="item",
+        help="Item(s) to bring",
+    )
+
+    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 = sorted(args.items) if args.sorted else args.items
+    supplies = ""
+    match len(items):
+        case 1:
+            supplies = items[0]
+        case 2:
+            supplies = " and ".join(items)
+        case _:
+            supplies = ", ".join(items[:-1]) + ", and " + items[-1]
+    print(f"You are bringing {supplies}.")
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()

03_picnic/test.py

@@ -4,65 +4,64 @@
 import os
 from subprocess import getoutput
 
-prg = './picnic.py'
+PRG = "./picnic.py"
 
 
 # --------------------------------------------------
 def test_exists():
     """exists"""
 
-    assert os.path.isfile(prg)
+    assert os.path.isfile(PRG)
 
 
 # --------------------------------------------------
 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"python3 {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"python3 {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.'
+    out = getoutput(f'python3 {PRG} soda "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"python3 {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"python3 {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"python3 {PRG} {arg} --sorted")
+    expected = "You are bringing apples, bananas, cherries, and dates."
     assert out.strip() == expected