Lập trình mạng với Python: Tạo chat room

import socket
import threading

# Cấu hình server
HOST = "127.0.0.1"  # Địa chỉ localhost
PORT = 12345        # Cổng để giao tiếp

server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind((HOST, PORT))
server.listen()

clients = []
nicknames = []

# Gửi tin nhắn đến tất cả client
def broadcast(message):
    for client in clients:
        client.send(message)

# Xử lý tin nhắn từ client
def handle(client):
    while True:
        try:
            message = client.recv(1024)
            broadcast(message)
        except:
            index = clients.index(client)
            clients.remove(client)
            client.close()
            nickname = nicknames[index]
            broadcast(f"{nickname} đã rời khỏi phòng!".encode("utf-8"))
            nicknames.remove(nickname)
            break

# Chấp nhận kết nối từ client
def receive():
    print("Server đang chạy...")
    while True:
        client, address = server.accept()
        print(f"Kết nối từ {str(address)}")

        client.send("NICK".encode("utf-8"))
        nickname = client.recv(1024).decode("utf-8")
        nicknames.append(nickname)
        clients.append(client)

        print(f"Nickname của client là {nickname}")
        broadcast(f"{nickname} đã tham gia phòng chat!".encode("utf-8"))
        client.send("Chào mừng đến với phòng chat!".encode("utf-8"))

        thread = threading.Thread(target=handle, args=(client,))
        thread.start()

receive()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

import socket
import threading

# Cấu hình client
HOST = "127.0.0.1"  # Địa chỉ server (phải khớp với HOST ở server)
PORT = 12345        # Cổng server (phải khớp với PORT ở server)

nickname = input("Nhập nickname của bạn: ")

client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect((HOST, PORT))

# Nhận tin nhắn từ server
def receive():
    while True:
        try:
            message = client.recv(1024).decode("utf-8")
            if message == "NICK":
                client.send(nickname.encode("utf-8"))
            else:
                print(message)
        except:
            print("Có lỗi xảy ra!")
            client.close()
            break

# Gửi tin nhắn đến server
def write():
    while True:
        message = f"{nickname}: {input('')}"
        client.send(message.encode("utf-8"))

# Chạy các luồng
receive_thread = threading.Thread(target=receive)
receive_thread.start()

write_thread = threading.Thread(target=write)
write_thread.start()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40