forked from Advanced_Python/advanced-python-homework-2023
62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
import cmd
|
|
import threading
|
|
from queue import Empty, Queue
|
|
from equipment.turtle_device import TurtleDevice
|
|
import argparse
|
|
import time
|
|
|
|
|
|
class TurtleDeviceThread(threading.Thread):
|
|
# TODO(Homework 4)
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.device = TurtleDevice()
|
|
self.queue = Queue()
|
|
self.event = Event()
|
|
|
|
def run(self):
|
|
while True:
|
|
try:
|
|
item = self.queue.get()
|
|
except self.queue.Empty:
|
|
continue
|
|
else:
|
|
if (item == 'exit'):
|
|
break
|
|
self.device.execute(item[0], item[1:])
|
|
self.queue.task_done()
|
|
def __message_processing__(self):
|
|
while True:
|
|
new_message = self.queue.get()
|
|
#print(type(new_message))
|
|
#print(new_message)
|
|
#print("Name and args of action are {}".format(new_message))
|
|
self.device.execute(*new_message)
|
|
#time.sleep(30)
|
|
self.queue.task_done()
|
|
if self.event.is_set():
|
|
break
|
|
def __message_thread__(self):
|
|
thread = Thread(target = self.__message_processing__, daemon=True)
|
|
thread.start()
|
|
|
|
class NoBlockingTurtleShell(cmd.Cmd):
|
|
intro = 'Welcome to the turtle shell. Type help or ? to list commands.\n'
|
|
prompt = '(turtle) '
|
|
file = None
|
|
|
|
def __init__(self, turtle_thread: TurtleDeviceThread):
|
|
super(NoBlockingTurtleShell, self).__init__()
|
|
turtle_thread.__message_thread__()
|
|
|
|
def do_execute(self, arg):
|
|
turtle_thread.queue.put(parse(arg))
|
|
|
|
def do_exit(self, arg):
|
|
turtle_thread.event.set()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
turtle_thread = TurtleDeviceThread()
|
|
turtle_thread.start()
|
|
NoBlockingTurtleShell(turtle_thread).cmdloop() |