Improve function use, main and release

Author: Gfcello

NetHack_AI.py

@@ -3,60 +3,64 @@
 import time
 import sys
 
-print('Press Space to begin')
-keyboard.wait("space")
+def main():
+    print('Press Space to begin')
+    keyboard.wait("space")
 
-while (True):
-    if keyboard.is_pressed('Esc'):
-            print("Exiting...")
-            sys.exit(0)
+    while (True):
+        if keyboard.is_pressed('Esc'):
+                print("Exiting...")
+                sys.exit(0)
 
-    # Generate Int for Decision Making
-    randInt = random.randint(0, 181)
-    # Lowercase Letter Inputs
-    if randInt < 52 :
-        randInt = randInt % 26
-        if randInt == 7 : # Keys to avoid (history)
-            randInt = randInt + 1
-        key = ord('a') + randInt
-        keyboard.press(chr(key))
-        time.sleep(0.1)
-        keyboard.release(chr(key))
-    # Uppercase Letter Inputs
-    elif randInt < 104 :
-        randInt = randInt % 26
-        if randInt == 14 or randInt == 18 or randInt == 21: # Keys to avoid (save, options)
-            randInt = randInt + 1
-        key = ord('a') + randInt
-        keyboard.press('shift')
-        keyboard.press(chr(key))
-        time.sleep(0.1)
-        keyboard.release('shift')
-        keyboard.release(chr(key))
-    # Directions (numbers)
-    elif randInt < 154 :
-        randInt = randInt % 10 #Doesn't put them in order, but gets 0-9 at 5 randInt chances
-        key = ord('0') + randInt
-        keyboard.press(chr(key))
-        time.sleep(0.1)
-        keyboard.release(chr(key))
-    # Exit Menus
-    elif randInt < 169 :
-        keyboard.press('space')
-        time.sleep(0.1)
-        keyboard.release('space')
-    # Pick up Item
-    elif randInt < 174 :
-        keyboard.press(',')
-        time.sleep(0.1)
-        keyboard.release(',')
-    # Direction at Self
-    elif randInt < 176 :
-        keyboard.press('.')
-        time.sleep(0.1)
-        keyboard.release('.')
-    else :
-        keyboard.press('enter')
-        time.sleep(0.1)
-        keyboard.release('enter')
-    time.sleep(0.01)
+        # Generate Int for Decision Making
+        randInt = random.randint(0, 181)
+        # Lowercase Letter Inputs
+        if randInt < 52 :
+            randInt = randInt % 26
+            if randInt == 7 : # Keys to avoid (history)
+                randInt = randInt + 1
+            key = ord('a') + randInt
+            keyboard.press(chr(key))
+            time.sleep(0.1)
+            keyboard.release(chr(key))
+        # Uppercase Letter Inputs
+        elif randInt < 104 :
+            randInt = randInt % 26
+            if randInt == 14 or randInt == 18 or randInt == 21: # Keys to avoid (save, options)
+                randInt = randInt + 1
+            key = ord('a') + randInt
+            keyboard.press('shift')
+            keyboard.press(chr(key))
+            time.sleep(0.1)
+            keyboard.release('shift')
+            keyboard.release(chr(key))
+        # Directions (numbers)
+        elif randInt < 154 :
+            randInt = randInt % 10 #Doesn't put them in order, but gets 0-9 at 5 randInt chances
+            key = ord('0') + randInt
+            keyboard.press(chr(key))
+            time.sleep(0.1)
+            keyboard.release(chr(key))
+        # Exit Menus
+        elif randInt < 169 :
+            keyboard.press('space')
+            time.sleep(0.1)
+            keyboard.release('space')
+        # Pick up Item
+        elif randInt < 174 :
+            keyboard.press(',')
+            time.sleep(0.1)
+            keyboard.release(',')
+        # Direction at Self
+        elif randInt < 176 :
+            keyboard.press('.')
+            time.sleep(0.1)
+            keyboard.release('.')
+        else :
+            keyboard.press('enter')
+            time.sleep(0.1)
+            keyboard.release('enter')
+        time.sleep(0.01)
+
+if __name__ == '__main__':
+    main()

Pokemon_AI.py

@@ -3,41 +3,41 @@
 import time
 import sys
 
-print('Press Space to begin')
-keyboard.wait("space")
+used_keys = [
+    'w', 'a', 's', 'd', 'x', 'z', 'enter'
+]
 
-while (True):
-    if keyboard.is_pressed('Esc'):
-            print("Exiting...")
-            sys.exit(0)
+def release_all():
+    for key in used_keys:
+        keyboard.release(key)
 
-    randInt = random.randint(0, 360)
-    if(randInt < 100):
-        keyboard.press('z')
-        time.sleep(0.2)
-        keyboard.release('z')
-    elif randInt < 150 :
-        keyboard.press('x')
-        time.sleep(0.2)
-        keyboard.release('x')
-    elif randInt < 200 :
-        keyboard.press('w')
-        time.sleep(0.2)
-        keyboard.release('w')
-    elif randInt < 250 :
-        keyboard.press('a')
-        time.sleep(0.2)
-        keyboard.release('a')
-    elif randInt < 300 :
-        keyboard.press('s')
-        time.sleep(0.2)
-        keyboard.release('s')
-    elif randInt < 350 :
-        keyboard.press('d')
-        time.sleep(0.2)
-        keyboard.release('d')
-    else :
-        keyboard.press('enter')
+def main():
+    print('Press Space to begin')
+    keyboard.wait("space")
+
+    while (True):
+        if keyboard.is_pressed('Esc'):
+                print("Exiting...")
+                release_all()
+                sys.exit(0)
+
+        randInt = random.randint(0, 360)
+        if(randInt < 100):
+            keyboard.press('z')
+        elif randInt < 150 :
+            keyboard.press('x')
+        elif randInt < 200 :
+            keyboard.press('w')
+        elif randInt < 250 :
+            keyboard.press('a')
+        elif randInt < 300 :
+            keyboard.press('s')
+        elif randInt < 350 :
+            keyboard.press('d')
+        else :
+            keyboard.press('enter')
         time.sleep(0.2)
-        keyboard.release('enter')
-    time.sleep(0.01)
+        release_all()
+
+if __name__ == '__main__':
+    main()

Spelunky_AI.py

@@ -3,68 +3,69 @@
 import time
 import sys
 
+used_keys = [
+    'i', 'j', 'k', 'l', 'x', 'z', 's', 'a', 'space'
+]
+
 def release_all():
-    keyboard.release('i')
-    keyboard.release('j')
-    keyboard.release('k')
-    keyboard.release('l')
-    keyboard.release('x')
-    keyboard.release('z')
-    keyboard.release('s')
-    keyboard.release('a')
-    keyboard.release('space')
+    for key in used_keys:
+        keyboard.release(key)
+
+def main():
+    print('Press Space to begin')
+    keyboard.wait("space")
 
-print('Press Space to begin')
-keyboard.wait("space")
+    while (True):
+        if keyboard.is_pressed('Esc'):
+                print("Exiting...")
+                release_all()
+                sys.exit(0)
 
-while (True):
-    if keyboard.is_pressed('Esc'):
-            print("Exiting...")
-            release_all()
-            sys.exit(0)
+        # First key press
+        randInt = random.randint(0, 100)
+        if(randInt < 10):
+            keyboard.press('i')
+        elif randInt < 20 :
+            keyboard.press('j')
+        elif randInt < 30 :
+            keyboard.press('k')
+        elif randInt < 40 :
+            keyboard.press('l')
+        elif randInt < 60 :
+            keyboard.press('x')
+        elif randInt < 80 :
+            keyboard.press('z')
+        elif randInt < 82 :
+            keyboard.press('s')
+        elif randInt < 85 :
+            keyboard.press('a')
+        else :
+            keyboard.press('space')
 
-    # First key press
-    randInt = random.randint(0, 100)
-    if(randInt < 10):
-        keyboard.press('i')
-    elif randInt < 20 :
-        keyboard.press('j')
-    elif randInt < 30 :
-        keyboard.press('k')
-    elif randInt < 40 :
-        keyboard.press('l')
-    elif randInt < 60 :
-        keyboard.press('x')
-    elif randInt < 80 :
-        keyboard.press('z')
-    elif randInt < 82 :
-        keyboard.press('s')
-    elif randInt < 85 :
-        keyboard.press('a')
-    else :
-        keyboard.press('space')
+        # Second key press
+        randInt = random.randint(0, 100)
+        if(randInt < 10):
+            keyboard.press('i')
+        elif randInt < 20 :
+            keyboard.press('j')
+        elif randInt < 30 :
+            keyboard.press('k')
+        elif randInt < 40 :
+            keyboard.press('l')
+        elif randInt < 60 :
+            keyboard.press('x')
+        elif randInt < 80 :
+            keyboard.press('z')
+        elif randInt < 82 :
+            keyboard.press('s')
+        elif randInt < 85 :
+            keyboard.press('a')
+        else :
+            keyboard.press('space')
 
-    # Second key press
-    randInt = random.randint(0, 100)
-    if(randInt < 10):
-        keyboard.press('i')
-    elif randInt < 20 :
-        keyboard.press('j')
-    elif randInt < 30 :
-        keyboard.press('k')
-    elif randInt < 40 :
-        keyboard.press('l')
-    elif randInt < 60 :
-        keyboard.press('x')
-    elif randInt < 80 :
-        keyboard.press('z')
-    elif randInt < 82 :
-        keyboard.press('s')
-    elif randInt < 85 :
-        keyboard.press('a')
-    else :
-        keyboard.press('space')
+        time.sleep(0.1)
 
-    time.sleep(0.1)
+        release_all()
 
-    release_all()
+if __name__ == '__main__':
+    main()