自学教程：python实现简易聊天室(Linux终端)

#coding =utf-8'''chat roomenv:python3.5exc:socket and forkname:mianmabbemail:mianmabb@163.com服务端功能: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)        return    else:    #昵称不存在,则发送约定好的'OK'        s.sendto(b'OK',addr)    #功能:有人进入聊天室,其他人会收到消息    msg = '/n   欢迎 %s 进入聊天室   '%name    for i in user:    #发送该条消息给其他用户        s.sendto(msg.encode(),user[i])        user[name] = addr   #将该用户插入数据结构(字典)#处理聊天def do_chat(s,name,text):    msg = '%s : %s'%(name,text)   #设置消息显示格式    for i in user:        s.sendto(msg.encode(),user[i])#处理退出def do_quit(s,name):    msg = '%s 退出了聊天室'%name    for i in user:        if i != name:   #给其他人发送该用户退出的消息            s.sendto(msg.encode(),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'%text            s.sendto(msg.encode(),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.sendto(msg.encode(),ADDR)            exit('您已退出聊天室')        else:            msg = 'C %s %s'%(name,text)  #消息类型,姓名,消息            s.sendto(msg.encode(),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:            return        msg = 'L ' +name        #发送请求        s.sendto(msg.encode(),ADDR)        #等待回复        data,addr = s.recvfrom(1024)        if data.decode() == 'OK':            print('您已进入聊天室')            break        else:   #登录失败            print(data.decode())  #直接打印服务端的错误信息            #创建进程    pid = fork()    if pid < 0:        print('Error')    elif pid == 0:   #子进程发消息        send_msg(s,name)    else:      #父进程收消息        recv_msg(s)main()