Problem 16 solution

lukemcguire · lukemcguire · commit 47f1bb3cfd18 · 2024-07-01T10:01:55.000-07:00
diff --git a/16_scrambler/scrambler.py b/16_scrambler/scrambler.py
@@ -0,0 +1,74 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <luke.mcguire@gmail.com>
+Date   : 2024-07-01
+Purpose: Scramble the letters of words
+"""
+
+import argparse
+import os
+import random
+import re
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Scramble the letters of words",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    parser.add_argument("text", metavar="text", help="Input text or file")
+    parser.add_argument("-s", "--seed", help="Random seed", metavar="int", type=int)
+
+    args = parser.parse_args()
+    if os.path.isfile(args.text):
+        with open(args.text, "rt", encoding="utf-8") as fh:
+            args.text = fh.read().rstrip()
+
+    return args
+
+
+# --------------------------------------------------
+def shuffle(word):
+    """Scramble the inside letters of a word"""
+    if len(word) > 3 and re.match(r"\w+", word):
+        first, middle, last = word[0], list(word[1:-1]), word[-1]
+        random.shuffle(middle)
+        word = first + "".join(middle) + last
+    return word
+
+
+# --------------------------------------------------
+def test_shuffle():
+    """Test the shuffle function"""
+    state = random.getstate()
+    random.seed(1)
+
+    assert shuffle("") == ""
+    assert shuffle("a") == "a"
+    assert shuffle("at") == "at"
+    assert shuffle("hat") == "hat"
+    assert shuffle("potato") == "ptoato"
+    assert shuffle("scrambled") == "slerbmcad"
+
+    random.setstate(state)
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+    random.seed(args.seed)
+    splitter = re.compile(r"([a-zA-Z](?:[a-zA-Z']*[a-zA-Z])?)")
+
+    for line in args.text.splitlines():
+        print("".join(map(shuffle, splitter.split(line))))
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()
diff --git a/16_scrambler/test.py b/16_scrambler/test.py
@@ -5,25 +5,25 @@
 import re
 from subprocess import getstatusoutput, getoutput
 
-prg = './scrambler.py'
-fox = '../inputs/fox.txt'
-bustle = '../inputs/the-bustle.txt'
-spiders = '../inputs/spiders.txt'
+PRG = "./scrambler.py"
+FOX = "../inputs/fox.txt"
+BUSTLE = "../inputs/the-bustle.txt"
+SPIDERS = "../inputs/spiders.txt"
 
 
 # --------------------------------------------------
 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"python {PRG} {flag}")
         assert rv == 0
         assert re.match("usage", out, re.IGNORECASE)
 
@@ -32,17 +32,17 @@ def test_usage():
 def test_text1():
     """Text"""
 
-    out = getoutput(f'{prg} foobar -s 1')
-    assert out.strip() == 'faobor'
+    out = getoutput(f"python {PRG} foobar -s 1")
+    assert out.strip() == "faobor"
 
 
 # --------------------------------------------------
 def test_text2():
     """Text"""
 
-    text = 'The quick brown fox jumps over the lazy dog.'
-    expected = 'The qicuk bworn fox jpmus over the lzay dog.'
-    out = getoutput(f'{prg} "{text}" -s 2')
+    text = "The quick brown fox jumps over the lazy dog."
+    expected = "The qicuk bworn fox jpmus over the lzay dog."
+    out = getoutput(f'python {PRG} "{text}" -s 2')
     assert out.strip() == expected
 
 
@@ -62,22 +62,22 @@ def test_file_bustle():
 Unitl eettnriy.
     """.strip()
 
-    out = getoutput(f'{prg} --seed 3 {bustle}')
+    out = getoutput(f"python {PRG} --seed 3 {BUSTLE}")
     assert out.strip() == expected.strip()
 
 
 # --------------------------------------------------
 def test_file_fox():
     """File input"""
 
-    out = getoutput(f'{prg} --seed 4 {fox}')
-    assert out.strip() == 'The qciuk bworn fox jpums oevr the lzay dog.'
+    out = getoutput(f"python {PRG} --seed 4 {FOX}")
+    assert out.strip() == "The qciuk bworn fox jpums oevr the lzay dog."
 
 
 # --------------------------------------------------
 def test_file_spiders():
     """File input"""
 
-    out = getoutput(f'{prg} --seed 9 {spiders}')
+    out = getoutput(f"python {PRG} --seed 9 {SPIDERS}")
     expected = "Do'nt wrory, sedrpis,\nI keep hsoue\ncalusaly."
     assert out.strip() == expected
