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.进入循环中等待客户端的链接

whileTrue:
        client_socket, addr = server_socket.accept()

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

# 存放链接进来聊天的用户
clients =[]whileTrue:
        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.客户端第一次进来需要输入自己的名称

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

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

whileTrue:
        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 是解码的意思

defbroadcast(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的定义
输入要发送的信息

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

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

defhandle_output(client_socket):"""处理消息输出"""whileTrue:
        msg = client_socket.recv(BUFFER_SIZE)#接收数据if msg.decode()=='quit':breakprint(msg.decode())#展示

客户端总结

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

4 完整的源码分享

服务端

import socket
import threading

SERVER_HOST ='0.0.0.0'
SERVER_PORT =8000
BUFFER_SIZE =1024defhandle_client(client_socket):"""处理客户端连接"""
    name = client_socket.recv(BUFFER_SIZE).decode()print(f'[*] {name} 已连接')
    welcome =f'欢迎来到聊天室, {name}!\n'.encode()
    client_socket.send(welcome)whileTrue:
        msg = client_socket.recv(BUFFER_SIZE)if msg.decode()=='quit':print(f'[*] {name} 断开连接')
            client_socket.close()break
        broadcast(msg, name)defbroadcast(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 =[]whileTrue:
        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 =1024defhandle_input(client_socket):"""处理用户输入"""whileTrue:
        msg =input('')
        client_socket.send(msg.encode())if msg =='quit':breakdefhandle_output(client_socket):"""处理消息输出"""whileTrue:
        msg = client_socket.recv(BUFFER_SIZE)if msg.decode()=='quit':breakprint(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()