socket聊天室

#服务端
from socket import *
import json
def recvMsg(s):
    while True:
        #接收用户的信息
        data,address = s.recvfrom(1024)
        data = json.loads(data)
        print(data,address)

        if data['type'] == 'enter':
            # 将用户进入聊天室的信息发给其它所有在线用户
            sendToAll(('>>系统：%s进入了聊天室'%data['content']).encode('utf-8'),s,address)
            #将用户信息加入users字典
            if address not in users:users[address]=data['content']
        elif data['type'] == 'quit':
            # 将用户离开聊天室的信息发给其它所有在线用户
            sendToAll(('>>系统：%s离开了聊天室' % data['content']).encode('utf-8'), s, address)
            # 将退出的用户从users中删除
            users.pop(address)

        elif data['type'] == 'msg':
            sendToAll(('>>%s：%s'%(users[address],data['content'])).encode('utf-8'),s,address)

        else:
            for i in users:
                if users[i] == data['name']:
                    s.sendto(('>>%s私信你说：%s'%(users[address],data['content'])).encode('utf-8'),i)

def sendToAll(msg,s,address):
    for addr in users:
        if addr != address:
            s.sendto(msg,addr)


def main():
    #创建socket
    s = socket(AF_INET,SOCK_DGRAM)
    #绑定服务器ip和端口
    s.bind(('172.16.11.199',8888))
    #接收用户的信息
    recvMsg(s)
    #发送信息给用户
if __name__ == '__main__':
    #设置一个全局的变量，保存所有进入聊天室的用户信息{'地址':'昵称','地址':'昵称',....}
    users = {}
    # print(users)
    main()



#客户端
from socket import *
import json
from threading import Thread
def sendMsg(c,server):
    while True:
        msg = input("").strip()
        #发送消息
        if not msg:
            continue
        elif msg.lower() == 'quit':
            # 发送进入聊天室的信息
            msg = {'type': 'quit', 'content': username}
            c.sendto(json.dumps(msg).encode('utf-8'), server)
            break
        #私聊
        elif msg.startswith('@'):
            name = str(msg[1:])
            msg2 = input('>>请输入你要对他说的话: ').strip()
            msg3 = {'type':'msg2','content':msg2,'name':name}
            print(">>我对%s说: %s"%(name ,msg2))
            c.sendto(json.dumps(msg3).encode('utf-8'),server)
        else:
            #正常信息
            msg1 = {'type':'msg','content':msg}
            print(">>我说：%s"%msg)
            c.sendto(json.dumps(msg1).encode('utf-8'), server)

def recvMsg(c):
    while True:
        data,address = c.recvfrom(1024)
        #打印出接收到的消息
        print(data.decode('utf-8'))

def main():
    c = socket(AF_INET,SOCK_DGRAM)
    #服务器的地址
    server = ('172.16.11.199',8888)
    #发送进入聊天室的信息
    msg = {'type':'enter','content':username}
    c.sendto(json.dumps(msg).encode('utf-8'),server)
    #用户发送消息
    t1 = Thread(target=sendMsg,args=(c,server))
    #用户接收消息
    t2 = Thread(target=recvMsg,args=(c,))

    t1.start()
    t2.start()
    t1.join()
    t2.join()

    c.close()
if __name__ == '__main__':
    #输入昵称
    username = input('昵称:').strip()
    print("%s,你已成功进入xx聊天室，想要退出请输入'quit'"%username)
    main()