135 lines
3.4 KiB
Python
135 lines
3.4 KiB
Python
import string
|
|
import sys
|
|
import os
|
|
import time
|
|
import asyncio
|
|
from Tomato import Tomato
|
|
from Cucumber import Cucumber
|
|
from Plant import Plant
|
|
from Trac import Trac
|
|
|
|
string.ascii_letters = 'oOpPx'
|
|
string.ascii_numbers = '01'
|
|
OKGREEN = '\033[92m'
|
|
OKBLUE = '\033[94m'
|
|
OKRED = '\033[91m'
|
|
ENDC = '\033[0m'
|
|
|
|
|
|
def initialize_field():
|
|
field = []
|
|
for i in range(11):
|
|
row = []
|
|
for j in range(7):
|
|
if i == 0 or i == 10 or j == 0 \
|
|
or j == 3 or j == 6:
|
|
row.append(None)
|
|
elif j < 3:
|
|
row.append(Cucumber())
|
|
else:
|
|
row.append(Tomato())
|
|
field.append(row)
|
|
return field
|
|
|
|
|
|
def print_field(field, tractor):
|
|
if sys.platform == "win32":
|
|
os.system("cls")
|
|
else:
|
|
os.system("clear")
|
|
print(tractor.get_position())
|
|
sys.stdout.write(OKBLUE)
|
|
sys.stdout.write("$")
|
|
sys.stdout.write(" ")
|
|
sys.stdout.write("\n")
|
|
|
|
for x, row in enumerate(field):
|
|
for y, i in enumerate(row):
|
|
if not i:
|
|
if tractor.get_position()[0] == x \
|
|
and tractor.get_position()[1] == y:
|
|
sys.stdout.write(OKBLUE)
|
|
sys.stdout.write(tractor.get_symbol())
|
|
sys.stdout.write(" ")
|
|
else:
|
|
sys.stdout.write(" ")
|
|
else:
|
|
symbol = i.get_symbol()
|
|
if symbol[1] == "green":
|
|
sys.stdout.write(OKGREEN)
|
|
sys.stdout.write(symbol[0])
|
|
sys.stdout.write(" ")
|
|
else:
|
|
sys.stdout.write(OKRED)
|
|
sys.stdout.write(symbol[0])
|
|
sys.stdout.write(" ")
|
|
|
|
sys.stdout.write("\n")
|
|
|
|
def update_state(field):
|
|
for row in field:
|
|
for i in row:
|
|
if i:
|
|
i.tick(1)
|
|
|
|
def try_move(field, tractor):
|
|
x = tractor.get_position()[0]
|
|
y = tractor.get_position()[1]
|
|
rotation = tractor.get_rotation()
|
|
if rotation == 'N':
|
|
x -= 1
|
|
elif rotation == 'S':
|
|
x += 1
|
|
elif rotation == 'W':
|
|
y -= 1
|
|
else:
|
|
y += 1
|
|
if x >= 0 and x < len(field) \
|
|
and y >= 0 and y < len(field[0]) \
|
|
and not field[x][y]:
|
|
# print(str(x) + " " + str(y))
|
|
return True
|
|
return False
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
field = initialize_field()
|
|
tractor = Trac('N', (0,0))
|
|
|
|
async def handle_echo(reader, writer):
|
|
data = await reader.readline()
|
|
message = data.decode().split()
|
|
|
|
if message[0] == "rotate":
|
|
tractor.set_rotation(message[1])
|
|
writer.write("OK\n".encode())
|
|
elif message[0] == "try":
|
|
if try_move(field, tractor):
|
|
writer.write("OK\n".encode())
|
|
else:
|
|
writer.write("FAIL\n".encode())
|
|
else:
|
|
tractor.move()
|
|
writer.write(("OK\n").encode())
|
|
|
|
print_field(field, tractor)
|
|
# print(tractor.get_position().get_x())
|
|
# print(tractor.get_position().get_y())
|
|
# print(tractor.get_rotation())
|
|
update_state(field)
|
|
await writer.drain()
|
|
|
|
writer.close()
|
|
|
|
async def main():
|
|
server = await asyncio.start_server(
|
|
handle_echo, '127.0.0.1', 8888)
|
|
|
|
addr = server.sockets[0].getsockname()
|
|
|
|
async with server:
|
|
await server.serve_forever()
|
|
|
|
asyncio.run(main())
|