Finished Chapter three.

Gabriel Soares Martins · Gabriel Soares Martins · commit 61df97d9ad8c · 2024-12-20T18:31:46.000-03:00
diff --git a/03_picnic/picnic.py b/03_picnic/picnic.py
@@ -0,0 +1,79 @@
+#!/usr/bin/env python3
+"""
+Author : gmartins <gmartins@localhost>
+Date   : 2024-12-18
+Purpose: Rock the Casbah
+"""
+
+import argparse
+
+TEMPLATE = 'You are bringing {}.'
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description='Picnic game',
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+
+    parser.add_argument('positional',
+                        metavar='str',
+                        nargs='+',
+                        help='Item(s) to bring')
+
+    parser.add_argument('-s',
+                        '--sorted',
+                        help='Sort the items',
+                        action='store_true')
+
+    parser.add_argument('-n',
+                        '--not-oxford',
+                        help='Do not print Oxford comma',
+                        action='store_true')
+
+    parser.add_argument('-r',
+                        '--sep',
+                        metavar='separator',
+                        default=',',
+                        help='Separator')
+
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+    sorted_arg = args.sorted
+
+    pos_arg = args.positional
+    oxford_arg = args.not_oxford
+    sep_arg = args.sep + ' '
+
+    if sep_arg != ', ':
+        oxford_arg = True
+
+    if sorted_arg:
+        pos_arg.sort()
+
+    if len(pos_arg) < 1:
+        print('Error')
+    elif len(pos_arg) == 1:
+        print(TEMPLATE.format(pos_arg[0]))
+    else:
+        if len(pos_arg) == 2:
+            s = ' and '.join(pos_arg)
+        else:
+            pos_arg[-1] = 'and ' + pos_arg[-1]
+            s = sep_arg.join(pos_arg)
+            if oxford_arg:
+                s = s.replace(sep_arg + 'and', ' and')
+        print(TEMPLATE.format(s))
+
+
+# --------------------------------------------------
+if __name__ == '__main__':
+    main()
diff --git a/03_picnic/test.py b/03_picnic/test.py
@@ -66,3 +66,24 @@ def test_more_than_two_sorted():
     out = getoutput(f'{prg} {arg} --sorted')
     expected = ('You are bringing apples, bananas, cherries, and dates.')
     assert out.strip() == expected
+
+
+# --------------------------------------------------
+def test_more_than_two_not_oxford():
+    """more than two items that do not contain oxford comma"""
+
+    arg = '"potato chips" coleslaw cupcakes "French silk pie" -n'
+    out = getoutput(f'{prg} {arg}')
+    expected = ('You are bringing potato chips, coleslaw, '
+                'cupcakes and French silk pie.')
+    assert out.strip() == expected
+
+
+def test_more_than_two_sep():
+    """more than two items with custom separator"""
+
+    arg = '"potato chips" coleslaw cupcakes "French silk pie" -r ";"'
+    out = getoutput(f'{prg} {arg}')
+    expected = ('You are bringing potato chips; coleslaw; '
+                'cupcakes and French silk pie.')
+    assert out.strip() == expected
