首页 > 脚本专栏 > python > python构建聊天室

使用python构建一个自己的聊天室

2023-04-03 09:28:11 作者：SYBH.

这篇文章主要介绍了使用python构建一个自己的聊天室,有助于帮助小伙伴们理解掌握Python的socket编程,需要的朋友可以参考下

一、简介

本篇文章旨在通过实现一个简单的多人聊天室，帮助读者掌握Python的socket编程。我们将分为以下几个部分来展开本文的内容：

二、网络编程基础概念

IP地址：网络中每个设备的唯一标识符，用于在网络中定位设备。
端口：标识设备上的不同网络应用，一个端口对应一个特定的服务或应用。
协议：数据通信的规则，如TCP和UDP。

三、Socket编程简介及原理

Socket是一种通信句柄，用于在网络中传输数据。
Python的socket库提供了丰富的函数和类，用于实现网络编程。
Socket编程主要包括创建socket对象、绑定地址和端口、监听连接、接受和发送数据等操作。

四、聊天室架构及功能需求

聊天室分为服务器端和客户端。
服务器端负责接收客户端的连接请求，转发客户端的消息。
客户端负责与服务器建立连接，并发送和接收消息。
聊天室需要支持多个客户端同时在线，并实时接收和发送消息。

五、实现聊天室服务器端

1.使用socket创建服务器端socket对象。

import socket
import threading
 
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

2.绑定IP地址和端口。

server.bind(("0.0.0.0", 8080))

3.监听客户端连接。

server.listen(5)

4.当有新的客户端连接时，将其加入已连接的客户端列表。

clients = []
 
while True:
    client_socket, client_address = server.accept()
    clients.append(client_socket)
    thread = threading.Thread(target=handle_client, args=(client_socket, client_address))
    thread.start()

5.使用多线程处理每个客户端的消息接收和发送。

def handle_client(client_socket, client_address):
    print(f"[INFO] Client {client_address} connected.")
    while True:
        try:
            message = client_socket.recv(1024).decode("utf-8")
            if not message:
                break
            print(f"[{client_address}] {message}")
            broadcast(client_socket, f"[{client_address}] {message}")
        except:
            break
    print(f"[INFO] Client {client_address} disconnected.")
    clients.remove(client_socket)
    client_socket.close()
``

6.当客户端断开连接时，从客户端列表中移除

    clients.remove(client_socket)
    client_socket.close()

7.广播消息给其他客户端。

def broadcast(from_client_socket, message):
    for client_socket in clients:
        if client_socket != from_client_socket:
            client_socket.send(message.encode("utf-8"))

8.聊天室服务器端完整代码：

import socket
import threading
 
def handle_client(client_socket, client_address):
    print(f"[INFO] Client {client_address} connected.")
    while True:
        try:
            message = client_socket.recv(1024).decode("utf-8")
            if not message:
                break
            print(f"[{client_address}] {message}")
            broadcast(client_socket, f"[{client_address}] {message}")
        except:
            break
    print(f"[INFO] Client {client_address} disconnected.")
    clients.remove(client_socket)
    client_socket.close()
 
def broadcast(from_client_socket, message):
    for client_socket in clients:
        if client_socket != from_client_socket:
            client_socket.send(message.encode("utf-8"))
 
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind(("0.0.0.0", 8080))
server.listen(5)
 
print("[INFO] Server is listening...")
 
clients = []
 
while True:
    client_socket, client_address = server.accept()
    clients.append(client_socket)
    thread = threading.Thread(target=handle_client, args=(client_socket, client_address))
    thread.start()

六、实现聊天室客户端

1.使用socket创建客户端socket对象。

import socket
import threading
 
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

2.连接到服务器端的IP地址和端口。

client.connect(("127.0.0.1", 8080))

3.使用多线程处理服务器端发送过来的消息。

def receive_message():
    while True:
        message = client.recv(1024).decode("utf-8")
        print(message)
 
thread = threading.Thread(target=receive_message)
thread.start()

4.接收用户输入的消息，并发送给服务器端。

while True:
    message = input()
    client.send(message.encode("utf-8"))

5.聊天室客户端完整代码：

import socket
import threading
 
def receive_message():
    while True:
        message = client.recv(1024).decode("utf-8")
        print(message)
 
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect(("127.0.0.1", 8080))
 
print("[INFO] Connected to the server.")
 
thread = threading.Thread(target=receive_message)
thread.start()
 
while True:
    message = input()
    client.send(message.encode("utf-8"))

七、测试聊天室

分别启动聊天室服务器端和多个客户端。在客户端发送消息，观察其他客户端是否能收到消息。尝试断开客户端，观察服务器端和其他客户端的表现。

到此这篇关于使用python构建一个自己的聊天室的文章就介绍到这了,更多相关python构建聊天室内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！