Python scripts for communication over serial and scramble generation

1 files changed, 22 insertions, 0 deletions

diff --git a/gen-scramble.py b/gen-scramble.py
new file mode 100644
index 0000000..d1482cd
--- /dev/null
+++ b/gen-scramble.py
@@ -0,0 +1,22 @@
+import sys
+import random
+
+faces = "RLUDFB"
+rot = " '2"
+
+def random_rotation():
+    r = random.choice(faces) + random.choice(rot)
+    return r.strip()
+
+def gen_scramble():
+    scramble = ''
+    for _ in range(int(sys.argv[2])):
+        scramble += random_rotation() + ' '
+    return scramble.strip()
+
+if len(sys.argv) != 4:
+    print(f"Usage: python3 {sys.argv[0]} seed length scrambles")
+    exit(1)
+
+for i in range(int(sys.argv[3])):
+    print(gen_scramble())