61 lines
1.3 KiB
Python
61 lines
1.3 KiB
Python
|
import aioconsole
|
||
|
|
import websockets
|
||
|
|
import asyncio
|
||
|
|
import json
|
||
|
|
|
||
|
|
IP = "localhost:8765"
|
||
|
|
|
||
|
|
player_id = 0
|
||
|
|
ws = websockets.connect("ws://" + IP)
|
||
|
|
connected = False
|
||
|
|
|
||
|
|
|
||
|
|
async def connect():
|
||
|
|
global ws
|
||
|
|
global connected
|
||
|
|
async with websockets.connect("ws://localhost:8765") as wsx:
|
||
|
|
ws = wsx
|
||
|
|
connected = True
|
||
|
|
while True:
|
||
|
|
await asyncio.sleep(10)
|
||
|
|
|
||
|
|
|
||
|
|
async def recieve_data_from_server():
|
||
|
|
while not connected:
|
||
|
|
await asyncio.sleep(1)
|
||
|
|
global ws
|
||
|
|
global player_id
|
||
|
|
while True:
|
||
|
|
async for message in ws:
|
||
|
|
try:
|
||
|
|
data = json.loads(message)
|
||
|
|
if data["action"] == "print":
|
||
|
|
print(data["data"])
|
||
|
|
elif data["action"] == "id":
|
||
|
|
player_id = data["id"]
|
||
|
|
except Exception as e:
|
||
|
|
print(e)
|
||
|
|
print(message)
|
||
|
|
|
||
|
|
|
||
|
|
async def send_data_to_server():
|
||
|
|
while not connected:
|
||
|
|
await asyncio.sleep(1)
|
||
|
|
global ws
|
||
|
|
global player_id
|
||
|
|
while True:
|
||
|
|
cmd = await aioconsole.ainput()
|
||
|
|
data = {"action": cmd, "id": player_id}
|
||
|
|
print(data)
|
||
|
|
await ws.send(json.dumps(data))
|
||
|
|
|
||
|
|
|
||
|
|
loop = asyncio.get_event_loop()
|
||
|
|
|
||
|
|
|
||
|
|
async def run_stuff():
|
||
|
|
await asyncio.gather(connect(), recieve_data_from_server(), send_data_to_server())
|
||
|
|
|
||
|
|
|
||
|
|
asyncio.run(run_stuff())
|