Python基于TCP实现聊天功能

Server端

import socket
import queue
import threading
import time

#socket.AF_INET指定使用IPv4地址族，socket.SOCK_STREAM指定使用TCP协议
serversocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
host = socket.gethostname()
print("服务器IP: "+socket.gethostbyname(host))
serversocket.bind((host, 9090))
serversocket.listen(5)
#存放已连接的对象
clients = []
#存放公共消息的容器
public_message = dict()

#接收新的对象
def init():
    while True:    
        # 当心客户端连接时
        client,addr = serversocket.accept()  #阻塞线程
        if client in clients:
            print('老用户')
        else:
            print('新的用户加入:',end='')
            print(client.getpeername()[0])
            client.send(bytes('欢迎来到聊天室(匿名)!'.encode('utf-8')))
            clients.append(client)
			#线程启动时会调用receive_msg函数,args为传递给函数的参数元组
            r = threading.Thread(target=receive_msg,args=(client,))
            r.start()

#接收消息
def receive_msg(client):
    while True:
        time.sleep(1)
        try:
            if client in clients:
                data = client.recv(1024).decode('utf-8')
                if data!='':
                    print(data)
                    public_message[client] = queue.Queue()
                    public_message[client].put(data)
                else:
                    if client in clients:
                        print("用户优雅的退出了")
                        clients.remove(client)
           
        except BaseException as error:

            print('用户强制中断了一个连接')
            # print('错误:',error)
            if client in clients:
                clients.remove(client)


#转发消息(非/阻塞)
def broadcast():
    while True:
        if len(clients)>1:
            public_message_clone = [i for i in public_message]   #解决字典迭代中操作报错的问题
            for client in clients:
                for i in public_message_clone:
                    if i!=client and public_message[i].empty()==False:
                        data = public_message[i].get_nowait()  #注意
                        if data !='':
                            # 转发消息给客户端
                            client.send(bytes(data.encode('utf-8'))) 
                            print('服务器转发了消息')
         
                    
t1 = threading.Thread(target=init)
t2 = threading.Thread(target=broadcast)

t1.start()
t2.start()


#主线程监听在线人数
while (True):
    print("当前在线人数为:%d"%(len(clients)))
    time.sleep(5)

Client

import socket,threading
#客户端想要发消息和收消息同时进行,需要使用多线程达到并发效果

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
# 获取计算机name
host = socket.gethostname()
# 获取根据计算机name获取计算机IP
host = socket.gethostbyname(host);
print(host)
s.connect((host, 9090))

# 接收消息
def receive():
    while True:
        data = s.recv(1024).decode('utf-8')
        if data!='':
            print(data)

# 发送消息
def send_msg():
    while (True):
        msg = input(':')
        if msg=='exit':
            s.close()
            break
        s.send(bytes(msg.encode('utf-8')))
        
t1 = threading.Thread(target=receive,daemon=True)
t1.start()

send_msg()