Problem 09 solution

lukemcguire · lukemcguire · commit 2997b7255903 · 2024-06-29T14:47:40.000-07:00
diff --git a/09_abuse/abuse.py b/09_abuse/abuse.py
@@ -0,0 +1,83 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <luke.mcguire@gmail.com>
+Date   : 2024-06-26
+Purpose: A Shakesperean insult generator
+"""
+
+import argparse
+import random
+
+
+# --------------------------------------------------
+def positive_int(string):
+    """check for positive integer"""
+    try:
+        value = int(string)
+    except ValueError as e:
+        raise argparse.ArgumentTypeError(f"invalid int value: '{string}'") from e
+    if value < 1:
+        raise argparse.ArgumentTypeError(f'"{value}" must be > 0')
+    return value
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="A Shakesperean insult generator",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    parser.add_argument(
+        "-a",
+        "--adjectives",
+        help="Number of adjectives",
+        metavar="int",
+        type=positive_int,
+        default=2,
+    )
+
+    parser.add_argument(
+        "-n",
+        "--number",
+        help="Number of insults",
+        metavar="int",
+        type=positive_int,
+        default=3,
+    )
+
+    parser.add_argument("-s", "--seed", help="Random seed", metavar="int", type=int)
+
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    adjectives = """
+bankrupt base caterwauling corrupt cullionly detestable dishonest false
+filthsome filthy foolish foul gross heedless indistinguishable infected
+insatiate irksome lascivious lecherous loathsome lubbery old peevish
+rascaly rotten ruinous scurilous scurvy slanderous sodden-witted
+thin-faced toad-spotted unmannered vile wall-eyed""".split()
+
+    nouns = """
+Judas Satan ape ass barbermonger beggar block boy braggart butt
+carbuncle coward coxcomb cur dandy degenerate fiend fishmonger fool
+gull harpy jack jolthead knave liar lunatic maw milksop minion
+ratcatcher recreant rogue scold slave swine traitor varlet villain worm""".split()
+
+    args = get_args()
+    random.seed(args.seed)
+
+    for _ in range(args.number):
+        descriptors = ", ".join(random.sample(adjectives, args.adjectives))
+        print(f"You {descriptors} {random.choice(nouns)}!")
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()
diff --git a/09_abuse/test.py b/09_abuse/test.py
@@ -7,22 +7,22 @@
 import string
 from subprocess import getstatusoutput, getoutput
 
-prg = './abuse.py'
+PRG = "./abuse.py"
 
 
 # --------------------------------------------------
 def test_exists():
     """exists"""
 
-    assert os.path.isfile(prg)
+    assert os.path.isfile(PRG)
 
 
 # --------------------------------------------------
 def test_usage():
     """usage"""
 
-    for flag in ['-h', '--help']:
-        rv, out = getstatusoutput(f'{prg} {flag}')
+    for flag in ["-h", "--help"]:
+        rv, out = getstatusoutput(f"python3 {PRG} {flag}")
         assert rv == 0
         assert re.match("usage", out, re.IGNORECASE)
 
@@ -32,7 +32,7 @@ def test_bad_adjective_str():
     """bad_adjectives"""
 
     bad = random_string()
-    rv, out = getstatusoutput(f'{prg} -a {bad}')
+    rv, out = getstatusoutput(f"python3 {PRG} -a {bad}")
     assert rv != 0
     assert re.search(f"invalid int value: '{bad}'", out)
 
@@ -42,18 +42,18 @@ def test_bad_adjective_num():
     """bad_adjectives"""
 
     n = random.choice(range(-10, 0))
-    rv, out = getstatusoutput(f'{prg} -a {n}')
+    rv, out = getstatusoutput(f"python3 {PRG} -a {n}")
     print(out)
     assert rv != 0
-    assert re.search(f'--adjectives "{n}" must be > 0', out)
+    assert re.search(f'--adjectives: "{n}" must be > 0', out)
 
 
 # --------------------------------------------------
 def test_bad_number_str():
     """bad_number"""
 
     bad = random_string()
-    rv, out = getstatusoutput(f'{prg} -n {bad}')
+    rv, out = getstatusoutput(f"python3 {PRG} -n {bad}")
     assert rv != 0
     assert re.search(f"invalid int value: '{bad}'", out)
 
@@ -63,17 +63,17 @@ def test_bad_number_int():
     """bad_number"""
 
     n = random.choice(range(-10, 0))
-    rv, out = getstatusoutput(f'{prg} -n {n}')
+    rv, out = getstatusoutput(f"python3 {PRG} -n {n}")
     assert rv != 0
-    assert re.search(f'--number "{n}" must be > 0', out)
+    assert re.search(f'--number: "{n}" must be > 0', out)
 
 
 # --------------------------------------------------
 def test_bad_seed():
     """bad seed"""
 
     bad = random_string()
-    rv, out = getstatusoutput(f'{prg} -s {bad}')
+    rv, out = getstatusoutput(f"python3 {PRG} -s {bad}")
     assert rv != 0
     assert re.search(f"invalid int value: '{bad}'", out)
 
@@ -82,15 +82,15 @@ def test_bad_seed():
 def test_01():
     """test"""
 
-    out = getoutput(f'{prg} -s 1 -n 1')
-    assert out.strip() == 'You filthsome, cullionly fiend!'
+    out = getoutput(f"python3 {PRG} -s 1 -n 1")
+    assert out.strip() == "You filthsome, cullionly fiend!"
 
 
 # --------------------------------------------------
 def test_02():
     """test"""
 
-    out = getoutput(f'{prg} --seed 2')
+    out = getoutput(f"python3 {PRG} --seed 2")
     expected = """
 You corrupt, detestable beggar!
 You peevish, foolish gull!
@@ -103,7 +103,7 @@ def test_02():
 def test_03():
     """test"""
 
-    out = getoutput(f'{prg} -s 3 -n 5 -a 1')
+    out = getoutput(f"python3 {PRG} -s 3 -n 5 -a 1")
     expected = """
 You infected villain!
 You vile braggart!
@@ -118,7 +118,7 @@ def test_03():
 def test_04():
     """test"""
 
-    out = getoutput(f'{prg} --seed 4 --number 2 --adjectives 4')
+    out = getoutput(f"python3 {PRG} --seed 4 --number 2 --adjectives 4")
     expected = """
 You infected, lecherous, dishonest, rotten recreant!
 You filthy, detestable, cullionly, base lunatic!
@@ -130,4 +130,4 @@ def test_04():
 def random_string():
     """generate a random filename"""
 
-    return ''.join(random.choices(string.ascii_lowercase + string.digits, k=5))
+    return "".join(random.choices(string.ascii_lowercase + string.digits, k=5))
