Python实现简易聊天室(Linux 终端)

Python实现简易聊天室(Linux 终端)

群聊聊天室
1.功能：类似qq群聊功能
1.有人进入聊天室需要输入姓名，姓名不能重复
2.有人进入聊天室，其他人会受到通知
xxx进入聊天室
3.一个人发消息，其他人会受到消息
xxx：xxxxxxxx
4.有人退出聊天室，其他人也会收到通知
xxx退出聊天室
5.扩展功能：服务端消息公告，服务端发送消息所有人都能收到
管理员消息：xxxxxxxx
2.确定技术模型
1.服务端和客户端
服务端处理请求，发送管理员消息
客户端执行各种功能
2.套接字选择：udp套接字
3.消息发送模型：转发
客户端 ~> 服务端 ~> 其他客户端
4.存储用户信息：{name：addr}
5.处理收发关系：多进程分别处理收发
3.注意事项
1.设计封装方案
2.写一个功能模块测试一个模块
3.注意注释的添加

#coding =utf-8
'''
chat room
env:python3.5
exc:socket and fork
name:mianmabb
email:mianmabb@163
服务端功能:
1.搭建网络通信
2.处理进入聊天室* 接收姓名* 判断是否允许进入* 将结果反馈给客户端* 如果不允许则结束，允许则将用户插入数据结构* 给其他人发送通知
3.处理聊天* 接收消息，判断消息类型,分为L(输入姓名),C(发消息),Q(退出聊天室)* 将消息转发
4.处理退出聊天室
5.发送管理员消息	
'''from socket import *
from os import *
from sys import *user = {}   #创建空字典用来存储用户的昵称和地址#处理登录
def do_login(s,name,addr):if name in user:    #判断昵称是否已经存在s.sendto('该昵称已被占用'.encode(),addr)returnelse:    #昵称不存在,则发送约定好的'OK's.sendto(b'OK',addr)#功能:有人进入聊天室,其他人会收到消息msg = 'n   欢迎 %s 进入聊天室   '%namefor i in user:    #发送该条消息给其他用户s.de(),user[i])user[name] = addr   #将该用户插入数据结构(字典)#处理聊天
def do_chat(s,name,text):msg = '%s : %s'%(name,text)   #设置消息显示格式for i in user:s.de(),user[i])#处理退出
def do_quit(s,name):msg = '%s 退出了聊天室'%namefor i in user:if i != name:   #给其他人发送该用户退出的消息s.de(),user[i])else:   #给该用户客户端发送约定好的EXIT让父进程退出s.sendto(b'EXIT',user[i])del user[name]   #删除字典中该用户#处理请求
def do_request(s):#循环接受所有客户请求while True:try:data,addr = s.recvfrom(1024)except KeyboardInterrupt:    #捕获父进程直接退出错误exit('服务端退出!')# print(data.decode())msgList = data.decode().split()   #按空格拆分为列表,方便索引if msgList[0] == 'L':    #判断消息类型do_login(s,msgList[1],addr)elif msgList[0] == 'C':text = ' '.join(msgList[2:])  #将消息中可能有的空格加回来do_chat(s,msgList[1],text)elif msgList[0] == 'Q':do_quit(s,msgList[1])def main():s = socket(AF_INET,SOCK_DGRAM)ADDR = ('0.0.0.0',8888)s.bind(ADDR)#创建进程pid = fork()if pid < 0:print('Error')elif pid == 0:   #子进程用来发送管理员消息while True:try:text = input('管理员 : ')except KeyboardInterrupt:    #捕获子进程直接退出错误exit()msg ='C 管理员 %s'%texts.de(),ADDR)else:   #父进程用来处理请求do_request(s)main()

'''
客户端功能：
1.搭建通信
2.进入聊天室* 输入姓名* 发送给服务器* 接收服务器反馈* 不允许则重新输入，允许则进入聊天室* 创建新的进程用于消息收发
3.聊天* 循环发送消息  消息类型分为L(输入姓名),C(发消息),Q(退出聊天室)* 循环接收消息
4.退出聊天室
5.接受管理员消息
'''from socket import *
from os import *
from sys import *ADDR = ('127.0.0.1',8888)    #填写服务端地址#循环发送消息
def send_msg(s,name):while True:try:text = input()   #客户输入要发送的消息except KeyboardInterrupt:   #子进程 防止用户Ctrl+C直接退出text = 'quit'if text.strip() == 'quit':    #规定输入quit退出msg = 'Q ' + name     #消息类型,姓名s.de(),ADDR)exit('您已退出聊天室')else:msg = 'C %s %s'%(name,text)  #消息类型,姓名,消息s.de(),ADDR)#循环接收信息
def recv_msg(s):while True:try:data,addr = s.recvfrom(1024)except KeyboardInterrupt:    #父进程 防止用户Ctrl+C直接退出exit()if data.decode() =='EXIT':   #当用户退出,无需再收消息,约定EXIT让父进程退出exit()   #退出父进程print(data.decode())#创建网络连接
def main():s = socket(AF_INET,SOCK_DGRAM)while True:name = input('请输入昵称:')     #输入姓名if not name:returnmsg = 'L ' +name#发送请求s.de(),ADDR)#等待回复data,addr = s.recvfrom(1024)if data.decode() == 'OK':print('您已进入聊天室')breakelse:   #登录失败print(data.decode())  #直接打印服务端的错误信息#创建进程pid = fork()if pid < 0:print('Error')elif pid == 0:   #子进程发消息send_msg(s,name)else:      #父进程收消息recv_msg(s)main()

先运行服务端,再运行客户端