Moved stuff from client to server, fixed bugs and renamed main.py to client.py

This commit is contained in:
Bruno Rybársky 2021-08-18 11:50:56 +02:00
parent 2ce2ce50f2
commit 61900cfb8b
3 changed files with 164 additions and 145 deletions

60
client.py Normal file

@ -0,0 +1,60 @@
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())

84
main.py

@ -1,84 +0,0 @@
import aioconsole
import websockets
import asyncio
import json
IP = "localhost:8765"
player_id = 0
ws = websockets.connect("ws://" + IP)
connected = False
got_message = 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_to_server():
while not connected:
await asyncio.sleep(1)
global ws
global player_id
global got_message
while True:
async for message in ws:
try:
data = json.loads(message)
if data["action"] == "redraw":
print(data["data"])
got_message = True
elif data["action"] == "info":
print(data["data"])
got_message = True
elif data["action"] == "id":
player_id = data["id"]
if got_message:
print("Enter command:\n")
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("Enter command:\n")
if cmd == "reset":
await ws.send(json.dumps({"action": "reset"}))
elif cmd == "redraw":
await ws.send(json.dumps({"action": "redraw"}))
elif cmd.startswith("resize"):
cmd_args = cmd.split(" ")
if len(cmd_args) >= 3:
await ws.send(json.dumps({"action": "resize", "width": cmd_args[1], "height": cmd_args[2]}))
else:
print("Incorrect argument number.")
elif cmd.startswith("add"):
cmd_args = cmd.split(" ")
if len(cmd_args) >= 2:
await ws.send(json.dumps({"action": "add", "col": cmd_args[1], "id": player_id}))
elif cmd == "help":
await ws.send(json.dumps({"action": "help"}))
else:
print("Unknown command. Type 'help' to list all commands.")
loop = asyncio.get_event_loop()
async def run_stuff():
await asyncio.gather(connect(), recieve_data_to_server(), send_data_to_server())
asyncio.run(run_stuff())

129
server.py

@ -4,7 +4,7 @@ import random
import websockets
import threading
width = 9
width = 10
height = 6
player_on_turn = 0
@ -27,71 +27,105 @@ async def hello(ws, _):
global width
global height
global player_on_turn
i = 0
exitit = 0
connected.add(ws)
this_id = random.randint(0, 100000)
i = 0
players.append({"ws": ws, "id": this_id})
if len(players) >= 3:
await socket_broadcast(json.dumps({"action": "info", "data": f"Warning {len(players)} players are connected"}))
await socket_broadcast(json.dumps({"action": "print", "data": f"Warning {len(players)} players are connected"}))
await ws.send(json.dumps({"action": "id", "id": this_id}))
try:
while i == 0:
await ws.send(await draw(grid, True))
for player in players:
if player["ws"] == ws:
await ws.send(json.dumps({"action": "print", "data": f"It is your turn, type 'add (number of column from 0 to {width - 1})' :"}))
else:
ws.send(json.dumps({"action": "print", "data": f"It is player{player_on_turn} 's turn."}))
await ws.send(json.dumps({"action": "print", "data": "Enter command:"}))
while exitit == 0:
try:
x = await ws.recv()
try:
temp = json.loads(x)
print(str(temp) + " data")
if temp["action"] == "reset":
temp2 = json.loads(x)
print(str(temp2))
cmd_args = str(temp2["action"]).split(" ")
temp = {"action": cmd_args[0], "args": [], "id": temp2["id"]}
i = 0
for cmd_arg in cmd_args:
if i != 0:
temp["args"].append(cmd_arg)
i = i + 1
print(str(temp))
if str(temp["action"]) == "reset":
await reset()
await draw(grid)
await socket_broadcast(
json.dumps({"action": "info", "data": f"It is player{player_on_turn} 's turn."}))
json.dumps({"action": "print", "data": f"It is player{player_on_turn} 's turn."}))
await players[player_on_turn]['ws'].send(json.dumps(
{"action": "info", "data": f"It is your turn, input column number from 0 to {width - 1}:"}))
elif temp["action"] == "redraw":
{"action": "print",
"data": f"It is your turn, input column number from 0 to {width - 1}:"}))
elif str(temp["action"]) == "redraw":
await ws.send(await draw(grid, True))
elif temp["action"] == "resize":
elif str(temp["action"]) == "resize":
try:
width = int(temp["width"])
height = int(temp["height"])
if len(temp["args"]) >= 2:
width = int(temp["args"][0])
height = int(temp["args"][1])
await reset()
await ws.send(await draw(grid, True))
await socket_broadcast(
json.dumps({"action": "info", "data": f"It is player{player_on_turn} 's turn."}))
await players[player_on_turn]['ws'].send(json.dumps({"action": "info",
"data": f"It is your turn, type 'add (number of column from 0 to {width - 1}' :"}))
json.dumps({"action": "print", "data": f"It is player{player_on_turn} 's turn."}))
await players[player_on_turn]['ws'].send(json.dumps({"action": "print",
"data": f"It is your turn, type 'add (number of column from 0 to {width - 1})' :"}))
except ValueError:
await ws.send(json.dumps({"action": "info", "data": "Invalid arguments to resize."}))
elif temp["action"] == "add":
await ws.send(json.dumps({"action": "print", "data": "Invalid arguments to resize."}))
elif str(temp["action"]) == "add":
if len(temp["args"]) >= 1:
plx = 0
for player in players:
if temp["id"] == player["id"]:
if int(temp["id"]) == int(player["id"]):
if plx == player_on_turn:
res = await add(grid, temp["col"])
res = await add(grid, temp["args"][0])
if res == 1:
await apply_gravity(grid)
await socket_broadcast(await draw(grid, True))
await socket_broadcast(json.dumps(
{"action": "info", "data": f"It is player{player_on_turn} 's turn."}))
await players[player_on_turn]['ws'].send(json.dumps({"action": "info",
"data": f"It is your turn, input column number from 0 to {width - 1}:"}))
{"action": "print", "data": f"It is player{player_on_turn} 's turn."}))
await players[player_on_turn]['ws'].send(json.dumps({"action": "print",
"data": f"It is your turn, input 'add (0 to {width - 1})':"}))
elif res == 2:
await players[player_on_turn]['ws'].send(json.dumps({"action": "info",
await players[player_on_turn]['ws'].send(json.dumps({"action": "print",
"data": f"Oops you have entered '{temp['col']}' which is not a valid input. Please try again player{player_on_turn}:"}))
elif res == 0:
await players[player_on_turn]['ws'].send(json.dumps({"action": "info",
await players[player_on_turn]['ws'].send(json.dumps({"action": "print",
"data": f"Oops you have entered '{temp['col']}' which is full. Please try again player{player_on_turn}:"}))
plx = plx + 1
elif temp["action"] == "help":
await ws.send(json.dumps({"action": "info",
"data": f"\n'reset' - Resets the game, also you must start the game with this\n'redraw' - Sends you the current state of th board\n'resize' - Resizes the playing board and resets it usage: 'resize (width) (height)'\n'add' - Adds your character to a column, usage: 'add (column number from 0 to {width - 1})\n'help' - Prints this help message"}))
elif str(temp["action"]) == "version":
await ws.send(json.dumps(
{"action": "print",
"data": "Bendžove Piškvorky\n'Jeminé, Bruno nemáš tam algorytmus' edition"}))
elif str(temp["action"]) == "help":
await ws.send(json.dumps({"action": "print",
"data": f"\n'reset' - Resets the game\n'redraw' - Sends you the current state of the board\n'resize' - Resizes the playing board and resets it usage: 'resize (width) (height)'\n'add' - Adds your character to a column, usage: 'add (column number from 0 to {width - 1})\n'version' - Prints version of the server\n'help' - Prints this help message"}))
else:
await ws.send(json.dumps(
{"action": "print", "data": "Unknown command. Type 'help' to list all commands."}))
await ws.send(json.dumps({"action": "print", "data": "Enter command:"}))
except Exception as e:
print(e)
except Exception as e:
print(e)
connected.remove(ws)
i = 1
exitit = 1
finally:
# Unregister.
try:
@ -155,30 +189,39 @@ async def add(gridx, col3):
async def draw(gridx, return_data=False):
temporary_storage = ""
for _ in range((width * 2) + 1):
temporary_storage = temporary_storage + "_"
temporary_storage = temporary_storage + "\n|"
tmp = ""
for i in range((width * 4) - 7):
tmp = tmp + ""
tmp = tmp + "\n/"
for i in range((width * 2) - 1):
tmp = tmp + " "
tmp = tmp + "\\"
tmp = tmp + "\n"
for i in range(0, width):
temporary_storage = temporary_storage + f"{i}|"
temporary_storage = temporary_storage + "\n|"
tmp = tmp + f"{i}"
tmp = tmp + "\n"
for y in range(0, height):
for x in range(0, width):
temporary_storage = temporary_storage + gridx[x][y] + "|"
temporary_storage = temporary_storage + "\n|"
temporary_storage = temporary_storage[:-1]
for _ in range((width * 2) + 1):
temporary_storage = temporary_storage + "¯"
print(temporary_storage)
tmp = tmp + gridx[x][y] + ""
tmp = tmp + "\n"
tmp = tmp[:-1]
tmp = tmp + "\\"
for i in range((width * 2) - 1):
tmp = tmp + " "
tmp = tmp + "/\n"
for i in range((width * 2) + 1):
tmp = tmp + ""
print(tmp)
if return_data:
return json.dumps({"action": "redraw", "data": temporary_storage})
return json.dumps({"action": "print", "data": tmp})
else:
await socket_broadcast(json.dumps({"action": "redraw", "data": temporary_storage}))
await socket_broadcast(json.dumps({"action": "print", "data": tmp}))
start_server = websockets.serve(hello, "localhost", 8765)
loop = asyncio.get_event_loop()
loop.run_until_complete(reset())
def loop_in_thread():