snail.py

from controller import Controller
import os
from get_move_from_server import get_server_move
import turtle

AI_IP = "replace me!"

def user_loop(control=Controller):
    w = control()
    w.create_temp_photo()
    w.start_turtle()
    w.window.onkey(w.up, 'Up')
    w.window.onkey(w.down, 'Down')
    w.window.onkey(w.right, 'r')
    w.window.onkey(w.left, 'l')
    w.window.onkey(w.piv_right, 'Right')
    w.window.onkey(w.piv_left, 'Left')
    w.window.onkey(w.exit_turtle, 'x')
    w.window.listen()
    turtle.mainloop()

def user_supervision():
    get_mode = raw_input("Continue AI driving? 1 = Yes, 2 = No:  ")
    if get_mode == "1": AI_loop()

def get_img_path():
    return "./images/current_image/" + os.listdir("./images/current_image")[0]

def set_server_ip():
    return raw_input("Enter AI server IP: ")

def AI_loop(
        counter=100,
        ai=get_server_move,
        user=user_supervision,
        img_path=get_img_path,
        control=Controller
    ):
    ip = set_server_ip()
    con = control() # BE NICE NOT TO MAKE A NEW CONTROLLER EVERY LOOP?
    con.create_temp_photo()
    while counter>0:
        ai_url = "http://" + ip + "/upload"
        move = ai(img_path(), ai_url)
        print("move ======== ", move)
        if move == 'forward':
            con.up()
        elif move == 'pivot right':
            con.piv_right()
        else:
            con.piv_left()
        counter -= 1
    return user()

def set_mode():
    get_mode = raw_input("Choose mode: 1 = User, 2 = AI: ")
    if get_mode == "1": user_loop()
    elif get_mode == "2":
        set_server_ip()
        AI_loop()

if __name__ == "__main__":
   set_mode()