python 使用socket制作聊天室-(附带源码可以直接运行)

admin • 2023-12-17 19:46 • Python

python 使用socket制作聊天室

整体流程:必拿下!!!

python 使用socket制作聊天室

1.基本前提

为什么socket能做聊天室呢

socket提供了通信的服务比如你可以创建tcp udp的请求向某个主机发起通信
就像是你想和某人说话你得会发音

就是提供了服务端的方式和客户端的方式
服务端处理用户发起的请求
客户端发送数据和接收服务端的响应数据
就像你打开浏览器看视频你作为客户端发起请求在某处的服务端(web服务器)会把请求接受处理
返回相应的数据通过浏览器(客户端)接受解析就变成了看的见的网页数据

想搞定聊天室
1.需要一个服务端提供下服务

时刻接受链接进来的用户

接受用户的发送信息

对所有的用户发送广播这样都能收到

2.需要多个客户端

1.链接进入聊天室
2.发送数据
3.接受广播数据

基本关系

2.服务端的搭建

1.导入socket包创建通讯tcp等请求
导入threading包使用线程来处理那些请求数据

import socket
import threading
#全局变量
SERVER_HOST = 'localhost'
SERVER_PORT = 8000
BUFFER_SIZE = 1024

2.创建socket的实例tcp通信对象
socket.AF_INET, socket.SOCK_STREAM 是创建tcp的参数
百度可以了解更多比如udp等

 server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

3.绑定端口服务的套接字也就是IP地址+端口号

 server_socket.bind((SERVER_HOST, SERVER_PORT))
 server_socket.listen(5) # 监听 最大可以链接的数量

4.进入循环中等待客户端的链接

    while True:
        client_socket, addr = server_socket.accept()

5.把链接的客户存放起来

# 存放链接进来聊天的用户
clients = []

    while True:
        client_socket, addr = server_socket.accept()
        clients.append(client_socket)

使用线程给进入的用户分配处理函数

# 开启线程
handle_client处理函数
client_socket 参数
thread = threading.Thread(target=handle_client, args=(client_socket,))
thread.start()

handle_client的定义
1.客户端第一次进来需要输入自己的名称

def handle_client(client_socket):
    """处理客户端连接"""
    name = client_socket.recv(BUFFER_SIZE).decode()#接受客户端发来的名称
    print(f'[*] {name} 已连接')#服务端打印查看
    welcome = f'欢迎来到聊天室, {name}!n'.encode()
    client_socket.send(welcome)#返回给客户端 欢迎的信息

2.循环等待用户的连天数据发送

    while True:
        msg = client_socket.recv(BUFFER_SIZE)#接受用户发来的消息 比如 你好
        #处理用户的退出
        if msg.decode() == 'quit':
            print(f'[*] {name} 断开连接')
            client_socket.close()
            break
        #向聊天室的所有人广播
        broadcast(msg, name)

broadcast定义
clients 所有的链接客户
发送谁说了什么
send(f’{name}: {msg.decode()}'.encode())
encode 是转码的意思
decode 是解码的意思

def broadcast(msg, name):
    """向所有已连接的客户端广播消息"""
    for client in clients:
        client.send(f'{name}: {msg.decode()}'.encode())

服务端总结:

创建一个tcp的服务端
把链接的用户存放起来
第一次接受用户名字
while循环接受发送的信息并广播出去

3. 客户端的搭建

引入socket包
引入threading包

import socket
import threading
#全局变量
SERVER_HOST = 'localhost'
SERVER_PORT = 8000
BUFFER_SIZE = 1024

创建一个tcp通信

client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

发起连接

 client_socket.connect((SERVER_HOST, SERVER_PORT))

发送自己的昵称

name = input('请输入您的昵称: ')
  # 发送昵称
name_bytes = name.encode()
client_socket.send(name_bytes)

开启线程进行聊天的处理

# 开启线程用来接收服务器发送的消息
    thread_output = threading.Thread(target=handle_output, args=(client_socket,))
    thread_output.start()

    # 开启线程用来处理用户输入的消息
    thread_input = threading.Thread(target=handle_input, args=(client_socket,))
    thread_input.start()

handle_input的定义
输入要发送的信息

def handle_input(client_socket):
    """处理用户输入"""
    while True:
        msg = input('')#输入要发送的信息
        client_socket.send(msg.encode())
        if msg == 'quit':#退出处理
            break

handle_output的定义
接受广播的数据展示出来

def handle_output(client_socket):
    """处理消息输出"""
    while True:
        msg = client_socket.recv(BUFFER_SIZE)#接收数据
        if msg.decode() == 'quit':
            break
        print(msg.decode())#展示

客户端总结

发起连接
进行昵称发送
发送信息
接受广播信息

4 完整的源码分享

服务端

import socket
import threading

SERVER_HOST = '0.0.0.0'
SERVER_PORT = 8000
BUFFER_SIZE = 1024

def handle_client(client_socket):
    """处理客户端连接"""
    name = client_socket.recv(BUFFER_SIZE).decode()
    print(f'[*] {name} 已连接')
    welcome = f'欢迎来到聊天室, {name}!n'.encode()
    client_socket.send(welcome)

    while True:
        msg = client_socket.recv(BUFFER_SIZE)
        if msg.decode() == 'quit':
            print(f'[*] {name} 断开连接')
            client_socket.close()
            break
        broadcast(msg, name)

def broadcast(msg, name):
    """向所有已连接的客户端广播消息"""
    for client in clients:
        client.send(f'{name}: {msg.decode()}'.encode())

if __name__ == '__main__':
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_socket.bind((SERVER_HOST, SERVER_PORT))
    server_socket.listen(5)
    print(f'[*] 正在听取端口 {SERVER_PORT}')

    clients = []

    while True:
        client_socket, addr = server_socket.accept()
        clients.append(client_socket)

        # 开启线程
        thread = threading.Thread(target=handle_client, args=(client_socket,))
        thread.start()

客户端的代码

import socket
import threading

SERVER_HOST = 'localhost'
SERVER_PORT = 8000
BUFFER_SIZE = 1024

def handle_input(client_socket):
    """处理用户输入"""
    while True:
        msg = input('')
        client_socket.send(msg.encode())
        if msg == 'quit':
            break

def handle_output(client_socket):
    """处理消息输出"""
    while True:
        msg = client_socket.recv(BUFFER_SIZE)
        if msg.decode() == 'quit':
            break
        print(msg.decode())

if __name__ == '__main__':
    name = input('请输入您的昵称: ')
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    client_socket.connect((SERVER_HOST, SERVER_PORT))

    # 发送昵称
    name_bytes = name.encode()
    client_socket.send(name_bytes)

    # 开启线程用来接收服务器发送的消息
    thread_output = threading.Thread(target=handle_output, args=(client_socket,))
    thread_output.start()

    # 开启线程用来处理用户输入的消息
    thread_input = threading.Thread(target=handle_input, args=(client_socket,))
    thread_input.start()