python socket实现聊天室

本文实例为大家分享了python socket实现聊天室的具体代码,供大家参考,具体内容如下

server端

import socket
import json,struct
from concurrent.futures import ThreadPoolExecutor
debug = True

s = socket.socket()
s.bind((\"127.0.0.1\",8848))
s.listen()
clients = {}

pool = ThreadPoolExecutor(100)

def send_msg(soc, msg):
    l = len(msg.encode(\"utf-8\"))
    soc.send(struct.pack(\"q\",l))

    # 发数据
    soc.send(msg.encode(\"utf-8\"))

\'\'\'用于监听已经建立连接的客户端发来的消息\'\'\'
def task(ip,c):
    while True:
        l = c.recv(8)
        ls = struct.unpack(\"q\",l)[0]
        data = json.loads(c.recv(ls).decode(\"utf-8\"))   #接收到来自客户端的消息。 eg:    data = {\'to_addr\':\'msg\':\'\'}
        print(\'来自%s的消息:%s\'%(ip,data[\'msg\']))
        # 客户端发过来的数据
        # 数据有两种情况 一种是发给所有人的 另一种单独发给某一个人的
        if data.get(\'to_addr\'): #传过来一个json格式的字典,如果这个to_addr不是空的话,就私发给目标客户
            target_ip = data[\"to_addr\"]        #从data字典里中获取目标ip
            target_conn = clients.get(target_ip)   #获取目标客户的conn链接
            send_msg(target_conn,data[\'msg\'])    #发送讯息
        else:
            for ip,conn in clients.items():
                # if c != target_conn:
                send_msg(conn,data[\'msg\'])#data[\'msg\']


        #     # 从所有客户端列表中找到这一个  发给它
        #     to_addr = data[\"to_addr\"]
        #     # print(data[\"to_addr\"],\"_______________\")
        #     soc = clients.get(to_addr)
        #     send_msg(soc,data[\"msg\"])
        # else:
        #     # 遍历所有客户端 发给每一个人
        #     for k,soc in clients.items():
        #         # if soc != c:
        #             send_msg(soc,data[\"msg\"])

while True:
    \'\'\'以下代码用于检测是否有客户端连接\'\'\'
    c,add = s.accept()
    print(\"%s\" % add[0],\"连接到服务器!\")
    clients[add[0]] = c  # 把ip地址作为key,conn作为value存入clients字典中 ,clients = {\'192.188.3.4\':conn链接}
    if debug:
        print(\'clients=%s\'%clients)

    \'\'\'以下代码用于监听已经建立连接的客户端发来的消息\'\'\'
    pool.submit(task,add,c)

client端

import socket,json,struct
from threading import Thread

c = socket.socket()
c.connect((\"127.0.0.1\",8848))
print(\"连接服务器成功!\")
def recver():
    while True:
        lens_bytes = c.recv(8)
        lens = struct.unpack(\"q\", lens_bytes)[0]
        recv_msg = c.recv(lens).decode(\"utf-8\")
        print(recv_msg)

# 开启线程处理接收数据
Thread(target=recver).start()


while True:
    msg = input(\">>>>:\").strip() # 阻塞
    if \"@\" in msg:
        info = {\"msg\":msg.split(\"@\")[0],\"to_addr\":msg.split(\"@\")[1]}
    else:
        info = {\"msg\": msg}

    data = json.dumps(info).encode(\"utf-8\")
    c.send(struct.pack(\"q\",len(data)))
    c.send(data)

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。

© 版权声明
THE END
喜欢就支持一下吧
点赞0 分享
评论 抢沙发

请登录后发表评论

    暂无评论内容