聊天室
需求分析: 建立产品的使用雏形(客户端) visio
根据需求,规划要实现的功能
技术分析: 根据功能确定实现技术
网络通信 : udp 网络技术
消息收发模型
* 转发 客户端--> 服务端-->其他客户端
* 用户进入聊天室时要存储用户地址 怎么存
{
name:address}
[(name,address),....]
class Person:
def __init__(self,name,address):
self.name = name
self.address = address
* 收发消息互相不干扰
父进程 发送消息
子进程 接收消息
- 功能模块分析(封装) : 拆
实现一个功能测一个功能
- 函数封装
进入聊天室
聊天
退出聊天室
通信协议设计
请求类型 数据
进入 L name
聊天 C name content
退出 E name
- 分模块具体逻辑设计
搭建基本网络结构
服务端: 1. 创建udp网络通信
2. 准备循环接受客户端消息
3. 根据请求调用合适的函数处理
客户端: 1.创建udp套接字
进入聊天室
客户端: 1. 输入姓名
2. 发送请求
3. 接受服务端的通知结果
4. Y 则进入 N 则重新输入姓名
服务端: 1. 接收请求
- 判断用户是否存在
3. Y 告知客户端不能进入 over
4. N 告知客户端进入聊天室
通知其他用户
存储用户信息
聊天
客户端: 1. 创建子进程
2. 子进程循环接收消息
3. 父进程循环输入发送消息
服务端: 1. 接收消息
2. 将消息转发给其他人
退出聊天室 输入exit表示退出
客户端 : 1. exit表示退出
2. 发送消息
3. 结束进程服务端 : 1. 接收消息
2. 告知其他用户
3. 将该用户删除
- 优化完善
功能 : 类似qq群功能
【1】 有人进入聊天室需要输入姓名,姓名不能重复
【2】 有人进入聊天室时,其他人会收到通知:xxx 进入了聊天室
【3】 一个人发消息,其他人会收到:xxx : xxxxxxxxxxx
【4】 有人退出聊天室,则其他人也会收到通知:xxx退出了聊天室
【5】 扩展功能:服务器可以向所有用户发送公告:管理员消息: xxxxxxxxx
"""
author: xx
email: [email protected]
time: 2020-8-10
env: Python 3.6
socket and Process
"""
from socket import *
from multiprocessing import Process
服务端
# 服务器地址
HOST = "0.0.0.0"
PORT = 8000
ADDR = (HOST, PORT)
# 存储结构用于存储用户信息 {name:address }
user = {
}
# 处理进入聊天室
def login(sock, name, addr):
# 如果用户存在
if name in user or "管理" in name:
sock.sendto(b"FAIL", addr)
return
# 告知用户进入
sock.sendto(b"OK", addr)
# 告知其他用户
msg = "欢迎 %s 进入聊天室" % name
for i in user:
sock.sendto(msg.encode(), user[i])
# 增加用户
user[name] = addr
# print("测试:",user)
# 聊天
def chat(sock, name, content):
msg = "%s : %s" % (name, content)
# 循环发送
for i in user:
if i == name:
continue
sock.sendto(msg.encode(), user[i])
# 退出
def exit(sock, name):
del user[name] # 删除用户
msg = "%s 退出了聊天室" % name
# 循环发送
for i in user:
sock.sendto(msg.encode(), user[i])
# 子进程处理请求
def request(sock):
# 循环接受客户端消息
while True:
# 所有请求都在这接受
data, addr = sock.recvfrom(1024 * 10)
# 对data做基本的解析
tmp = data.decode().split(' ', 2)
# 根据请求选择函数处理
if tmp[0] == "L":
# tmp --> ['L','name']
login(sock, tmp[1], addr)
elif tmp[0] == "C":
# tmp --> [C,name,content]
chat(sock, tmp[1], tmp[2])
elif tmp[0] == "E":
# tmp --> [E,name]
exit(sock, tmp[1])
# 启动函数用于搭建网络
def main():
# UDP套接字
sock = socket(AF_INET, SOCK_DGRAM)
sock.bind(ADDR)
# 创建子进程
p = Process(target=request, args=(sock,))
p.daemon = True
p.start()
while True:
content = input("管理员消息:")
# 服务端退出
if content == "exit":
break
msg = "C 管理员消息 "+content
# 从父进程发送给子进程
sock.sendto(msg.encode(),ADDR)
if __name__ == '__main__':
main()
客户端
"""
chat room 客户端
发送请求 接收消息
"""
from socket import *
from multiprocessing import Process
import sys
# 服务器地址
ADDR = ("124.70.148.168", 8000)
# 进入聊天室
def login(sock):
while True:
name = input("Name:")
# 给服务端发请求
msg = "L " + name # 根据协议组织消息
sock.sendto(msg.encode(), ADDR)
# 等待结果
result, addr = sock.recvfrom(128)
# 约定OK作为请求成功的标志
if result.decode() == 'OK':
print("进入聊天室")
return name # 以name登录
else:
print("该用户已存在")
# 接收消息
def recv_msg(sock):
while True:
data, addr = sock.recvfrom(1024 * 10)
# 美化打印内容
msg = "\n" + data.decode() + "\n发言:"
print(msg, end="")
# 发送消息
def send_msg(sock, name):
while True:
try:
content = input("发言:")
except KeyboardInterrupt:
content = "exit"
# 输入exit要退出
if content == 'exit':
msg = "E " + name
sock.sendto(msg.encode(), ADDR)
sys.exit("退出聊天室")
msg = "C %s %s" % (name, content)
sock.sendto(msg.encode(), ADDR)
def main():
sock = socket(AF_INET, SOCK_DGRAM)
sock.bind(('0.0.0.0',55667))
# sock.sendto("测试信息".encode(),ADDR)
# 进入聊天室
name = login(sock)
# 创建子进程
p = Process(target=recv_msg, args=(sock,))
p.daemon = True # 子进程随父进程退出
p.start()
send_msg(sock, name) # 父进程发送消息
if __name__ == '__main__':
main()
执行需先执行服务端后再执行客户端