69 lines
2.4 KiB
Python
69 lines
2.4 KiB
Python
import asyncio
|
|
|
|
import websockets
|
|
|
|
DEFAULT_SERVER_URL = "wss://connect4.abunchofknowitalls.com"
|
|
|
|
|
|
def calculate_move(opponent_move):
|
|
if opponent_move is not None:
|
|
print(f"Opponent played column {opponent_move}")
|
|
# TODO: Use the board variable to see and set the current state of the board
|
|
# TODO: Implement your move calculation logic here instead
|
|
return 0
|
|
|
|
|
|
async def gameloop(socket):
|
|
board = [[None] * 6 for _ in range(7)]
|
|
while True: # While game is active, continually anticipate messages
|
|
message = (await socket.recv()).split(":") # Receive message from server
|
|
|
|
match message[0]:
|
|
case "CONNECT":
|
|
await socket.send("READY")
|
|
|
|
case "GAME":
|
|
if message[1] == "START":
|
|
if message[2] == "1":
|
|
col = calculate_move(
|
|
None
|
|
) # calculate_move is some arbitrary function you have created to figure out the next move
|
|
await socket.send(f"PLAY:{col}") # Send your move to the sever
|
|
if (
|
|
(message[1] == "WINS")
|
|
| (message[1] == "LOSS")
|
|
| (message[1] == "DRAW")
|
|
| (message[1] == "TERMINATED")
|
|
): # Game has ended
|
|
print(message[0] + ":" + message[1])
|
|
# TODO: Create an empty board
|
|
await socket.send("READY")
|
|
|
|
case "OPPONENT": # Opponent has gone; calculate next move
|
|
col = calculate_move(
|
|
message[1]
|
|
) # Give your function your opponent's move
|
|
await socket.send(f"PLAY:{col}") # Send your move to the sever
|
|
|
|
case "ERROR":
|
|
print(f"{message[0]}: {':'.join(message[1:])}")
|
|
|
|
await socket.close()
|
|
|
|
|
|
async def join_server(username, server_url):
|
|
async with websockets.connect(
|
|
server_url, ping_interval=30, ping_timeout=30
|
|
) as socket: # Establish websocket connection
|
|
await socket.send(f"CONNECT:{username}")
|
|
await gameloop(socket)
|
|
|
|
|
|
if __name__ == "__main__": # Program entrypoint
|
|
server_url = (
|
|
input(f"Enter server address [{DEFAULT_SERVER_URL}]: ").strip()
|
|
or DEFAULT_SERVER_URL
|
|
)
|
|
username = input("Enter username: ")
|
|
asyncio.run(join_server(username, server_url))
|