怎么用python代码实现简易聊天室

#coding =utf-8
"""
chat room
env:python3.5
exc:socket and fork
name:mianmabb
email: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 = "
   欢迎 %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()