client.py

import pygame
import socket
import pickle

class GameClient:
    def __init__(self, host='localhost', port=5555):
        self.client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.host = host
        self.port = port
        self.player_id = None
        self.players = {}
        
    def connect(self):
        try:
            self.client.connect((self.host, self.port))
            # Receive player ID from server
            self.player_id = pickle.loads(self.client.recv(4096))
            print(f"Connected as Player {self.player_id}")
            return True
        except Exception as e:
            print(f"Connection failed: {e}")
            return False
    
    def send(self, data):
        try:
            self.client.send(pickle.dumps(data))
            return pickle.loads(self.client.recv(4096))
        except Exception as e:
            print(f"Send error: {e}")
            return None

class Game:
    def __init__(self):
        pygame.init()
        self.width = 800
        self.height = 600
        self.screen = pygame.display.set_mode((self.width, self.height))
        pygame.display.set_caption("Multiplayer Game")
        self.clock = pygame.time.Clock()
        self.running = True
        
        self.client = GameClient()
        if not self.client.connect():
            self.running = False
            return
        
        self.player_x = 400
        self.player_y = 300
        self.player_speed = 5
    
    def handle_events(self):
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                self.running = False
    
    def update(self):
        keys = pygame.key.get_pressed()
        
        if keys[pygame.K_LEFT] or keys[pygame.K_a]:
            self.player_x -= self.player_speed
        if keys[pygame.K_RIGHT] or keys[pygame.K_d]:
            self.player_x += self.player_speed
        if keys[pygame.K_UP] or keys[pygame.K_w]:
            self.player_y -= self.player_speed
        if keys[pygame.K_DOWN] or keys[pygame.K_s]:
            self.player_y += self.player_speed
        
        # Keep player in bounds
        self.player_x = max(20, min(self.width - 20, self.player_x))
        self.player_y = max(20, min(self.height - 20, self.player_y))
        
        # Send position to server and get all players
        player_data = {'x': self.player_x, 'y': self.player_y}
        self.players = self.client.send(player_data)
        
        if self.players is None:
            self.running = False
    
    def draw(self):
        self.screen.fill((30, 30, 30))
        
        # Draw all players
        if self.players:
            for pid, pdata in self.players.items():
                color = pdata['color']
                x, y = int(pdata['x']), int(pdata['y'])
                
                # Draw player circle
                pygame.draw.circle(self.screen, color, (x, y), 20)
                
                # Draw player ID
                font = pygame.font.Font(None, 24)
                if pid == self.client.player_id:
                    text = font.render(f"You (P{pid})", True, (255, 255, 255))
                else:
                    text = font.render(f"P{pid}", True, (255, 255, 255))
                self.screen.blit(text, (x - 30, y - 40))
        
        # Draw instructions
        font = pygame.font.Font(None, 20)
        text = font.render("Use WASD or Arrow Keys to move", True, (200, 200, 200))
        self.screen.blit(text, (10, 10))
        
        pygame.display.flip()
    
    def run(self):
        while self.running:
            self.handle_events()
            self.update()
            self.draw()
            self.clock.tick(60)
        
        pygame.quit()

if __name__ == "__main__":
    game = Game()
    game.run()