Python系列：摸鱼小利器，使用Python创建一个聊天室

知世不是芝士

于 2024-04-19 15:00:00 发布

阅读量842

点赞数 21

分类专栏： Python python脚本 Python教程文章标签： python 开发语言数据分析

本文链接：https://blog.csdn.net/python122_/article/details/137953370

版权

Python 同时被 3 个专栏收录

35 篇文章 0 订阅

订阅专栏

python脚本

22 篇文章 0 订阅

订阅专栏

Python教程

15 篇文章 0 订阅

订阅专栏

本文介绍了如何使用Python的socket库创建一个简单的聊天室，包括服务器端和客户端的实现，以及如何通过多线程保持长连接。作者还提供了学习Python的资源链接，包括教程、实战案例和面试资料。

摘要由CSDN通过智能技术生成

引言

上班是帮老板赚钱，摸鱼才是赚老板的钱。

日常摸鱼跟小伙伴聊天是我们经常干的事儿，但是对于有些位置不好的同学，背对领导或者坐在过道中的，就需要时刻盯防经过的人，以防自己上班聊天划水摸鱼被抓现行，那么今天就来介绍如何使用Python创建一个聊天室。

socket介绍

在网络编程中，Socket 是一种通信机制，允许计算机在网络上相互通信。Python 中的 socket 库提供了一种简单而强大的方式来创建网络应用程序,本文将介绍socket库的基本概念、创建和使用套接字（Socket）的方法以及一些常见的网络编程用例。

基本概念

Socket 是一种通信的端点，可以通过网络发送和接收数据。在网络编程中，通常使用两种类型的Socket：服务器端套接字和客户端套接字。服务器端套接字等待来自客户端的连接请求，而客户端套接字尝试与服务器端建立连接。

创建服务器

import socket


# 创建服务器端套接字
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 绑定地址和端口
server_address = ('localhost', 8888)
server_socket.bind(server_address)

# 监听连接请求
server_socket.listen(5)

print('Waiting for client to connect...')

# 接受客户端连接
client_socket, client_address = server_socket.accept()
print('Connection from:', client_address)

# 接收数据
data = client_socket.recv(1024)
print('Received:', data.decode())

# 发送数据
message = 'Hello, client!'
client_socket.send(message.encode())

# 关闭连接
client_socket.close()
server_socket.close()

通过上面的代码我们成功出创建了一个聊天室的服务器，其中的ip可以替换成你自己电脑的ip地址，这样比人就可以通过IP:Port来链接你的电脑。

运行后我们可以看到服务器目前正在等待连接中，接下来我们就创建客户端来链接服务器。

创建客户端

import socket

# 创建客户端套接字
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 连接到服务器
server_address = ('localhost', 8888)
client_socket.connect(server_address)

# 发送数据
message = 'Hello, server!'
client_socket.send(message.encode())

# 接收数据
data = client_socket.recv(1024)
print('Received:', data.decode())

# 关闭连接
client_socket.close()

通过上面的代码我们已经成功创建好了客户端，链接服务器地址哪里修改成服务器的对应IP和端口就行了，接下来我们就来链接服务器。

可以看到运行客户端后我们可以看到服务端和客户端都有反应，服务端收到客户端的连接后发送了一天消息，客户端同时也收到了服务端发来的消息，这两条消息是我们指定的，并且发送一条后聊天室就关闭，接下来我们来优化一下

保持聊天室长连接

这里我们使用多线程来创建一个持久性连接的聊天室

服务器

import socket
import threading

def handle_client(client_socket):
    while True:
        # 接收客户端消息
        data = client_socket.recv(1024)
        if not data:
            break  # 如果客户端断开连接，退出循环
        message = data.decode()
        print(f"Received from {client_socket.getpeername()}: {message}")

        # 广播消息给所有客户端
        broadcast(message, client_socket)

    # 关闭连接
    print(f"Connection from {client_socket.getpeername()} closed.")
    clients.remove(client_socket)
    client_socket.close()

def broadcast(message, sender_socket):
    for client in clients:
        if client != sender_socket:
            try:
                client.send(message.encode())
            except:
                # 如果发送失败，说明客户端已断开连接，移除该客户端
                clients.remove(client)

# 创建服务器端套接字
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 绑定地址和端口
server_address = ('localhost', 8888)
server_socket.bind(server_address)

# 监听连接请求
server_socket.listen(5)

print('Waiting for clients to connect...')

# 存储所有客户端套接字的列表
clients = []

while True:
    # 接受客户端连接
    client_socket, client_address = server_socket.accept()
    print(f"Connection from {client_address} established.")

    # 存储客户端套接字到列表
    clients.append(client_socket)

    # 创建一个线程来处理客户端
    client_handler = threading.Thread(target=handle_client, args=(client_socket,))
    client_handler.start()

客户端

import socket
import threading

def receive_messages():
    while True:
        try:
            # 接收服务器端消息
            data = client_socket.recv(1024)
            if not data:
                break  # 如果服务器端断开连接，退出循环
            message = data.decode()
            print("Received:", message)
        except:
            break

# 创建客户端套接字
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 连接到服务器
server_address = ('localhost', 8888)
client_socket.connect(server_address)

# 创建一个线程来接收服务器端消息
receive_thread = threading.Thread(target=receive_messages)
receive_thread.start()

# 发送消息到服务器
while True:
    message = input("我是客户端1: ")
    client_socket.send(message.encode())