import socket
import threading

num=0

def chat(service_client_socket,addr): 
    # 等待接收客戶(hù)端消息存放在2個(gè)變量service_client_socket和addr里
    if not addr in user:  
        print('Accept new connection from %s:%s...' % addr)
        # 如果addr不在user字典里則執(zhí)行以下代碼
        for scs in serv_clie_socket: 
            serv_clie_socket[scs].send(data +' 進(jìn)入聊天室...'.encode('utf-8'))  
            # 發(fā)送user字典的data和address到客戶(hù)端
        user[addr] = data.decode('utf-8')  #data 是最新進(jìn)入聊天室的客戶(hù)，解壓后放入user
        serv_clie_socket[addr] = service_client_socket #將服務(wù)器與服務(wù)器端口號(hào)為addr的套接字放入字典
        # 接收的消息解碼成utf-8并存在字典user里,鍵名定義為addr
    #print("可以開(kāi)始聊天了>>>>>>")       
    # 如果addr在user字典里，跳過(guò)本次循環(huán)
    while True:
        d = service_client_socket.recv(1024)
        if (('EXIT'.lower() in d.decode('utf-8'))|(d.decode('utf-8') == 'error1')):
            #如果EXIT在發(fā)送的data里
            name = user[addr]   
            #user字典addr鍵對(duì)應(yīng)的值賦值給變量name
            user.pop(addr) 
            serv_clie_socket.pop(addr)     
            #刪除user里的addr
            for scs in serv_clie_socket:     
                #從user取出address
                serv_clie_socket[scs].send((name + ' 離開(kāi)了聊天室...').encode('utf-8'))     
                #發(fā)送name和address到客戶(hù)端
            print('Connection from %s:%s closed.' % addr)
            global num
            num = num-1
            break
        else: 
            print('"%s" from %s:%s' %(d.decode('utf-8'), addr[0], addr[1]))  
            for scs in serv_clie_socket:    
                #從user遍歷出address
                if serv_clie_socket[scs] != service_client_socket:  
                    #address不等于addr時(shí)，執(zhí)行下面的代碼
                    serv_clie_socket[scs].send(d)     
                    #發(fā)送data到客戶(hù)端

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # 創(chuàng)建socket對(duì)象

addr = ('127.0.0.1', 9999)
s.bind(addr)  # 綁定地址和端口

s.listen(128)

print('TCP Server on', addr[0], ":",addr[1],"......")

user = {}  # 存放字典{addr:name}
serv_clie_socket = {} #存放{socket:不同線(xiàn)程的套接字}
while True:
    try:
        print("等待接收客戶(hù)端的連接請(qǐng)求....")
        service_client_socket, addr = s.accept() # 等待接收客戶(hù)端的連接請(qǐng)求
        print("接收到客戶(hù)端的連接請(qǐng)求....")
    except ConnectionResetError:
        print('Someone left unexcept.')
    data = service_client_socket.recv(1024)
    if data.decode()=='error1':
        print(addr,"關(guān)閉了登錄窗口。。。")
        continue
    print("data = ",data.decode())
    
         
    #為服務(wù)器分配線(xiàn)程
    num=num+1
    r = threading.Thread(target=chat, args=(service_client_socket,addr), daemon=True) 
    r.start()
    print("聊天室人數(shù)：",num)

#客戶(hù)端
import tkinter
from tkinter import font
import tkinter.messagebox
import socket
import threading
import time

string=''
def my_string(s_input):
    string = s_input.get()

def Send(sock):
    '''
        發(fā)送數(shù)據(jù)的方法
        參數(shù)：
            sock：定義一個(gè)實(shí)例化socket對(duì)象
            server：傳遞的服務(wù)器IP和端口
    '''
    if string!='':
        message = name + ' : ' + string
        data = message.encode('utf-8')
        sock.send(data)
        if string.lower() == 'EXIT'.lower():
            exit()


def recv(sock):
    sock.send(name.encode('utf-8'))
    while True:
        data = sock.recv(1024)
        #加一個(gè)時(shí)間戳
        time_tuple = time.localtime(time.time())
        str = ("{}點(diǎn){}分".format(time_tuple[3],time_tuple[4]))
        rrecv = tkinter.Label(t,text=data.decode('utf-8'),width=40,anchor='w',bg='pink')#接收的消息靠左邊
        rrecv.pack()

def left():
    global string
    string = rv1.get()
    Send(s)
    if string!='':
        rleft = tkinter.Label(t,text=string,width=40,anchor='e')#發(fā)送的消息靠右邊
        rleft.pack()
        rv1.set('')

def Creat():
    global name
    name = n.get()
    
    #接收進(jìn)程
    tr = threading.Thread(target=recv, args=(s,), daemon=True)
    # daemon=True 表示創(chuàng)建的子線(xiàn)程守護(hù)主線(xiàn)程，主線(xiàn)程退出子線(xiàn)程直接銷(xiāo)毀
    tr.start()
     
    l.destroy()
    e.destroy() 
    b.destroy()
    t.title("聊天室")
    t.geometry("500x600")
    rL0 = tkinter.Label(t,text='%s的聊天室'%name,width=40)
    rL0.pack()
    rL1 = tkinter.Label(t,text='請(qǐng)輸入消息：',width=20, height=1)
    rL1.place(x=0,y=450)
    rE1 = tkinter.Entry(t, textvariable = rv1)
    rE1.place(x=200,y=450) 
    rB1 = tkinter.Button(t, text="發(fā)送",command=left)  
    rB1.place(x=380,y=450)
    #發(fā)送進(jìn)程

def JieShu():
    tkinter.messagebox.showwarning(title='你確定退出嗎？', message='剛才你點(diǎn)擊了關(guān)閉按鈕')
    s.send("error1".encode('utf-8'))
    exit(0)   
    
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)    
server = ('10.100.207.40', 9999)
s.connect(server)#建立連接
t=tkinter.Tk()
t.title("多人聊天室")
t.geometry("300x200+500+200")
l = tkinter.Label(t,text='多人聊天室歡迎您，請(qǐng)輸入你的名稱(chēng)',width=40, height=8)
l.pack()
n = tkinter.StringVar()
e = tkinter.Entry(t, width=15,textvariable = n)
e.pack() 
rv1 = tkinter.StringVar()
name = n.get()

b = tkinter.Button(t, text="登錄",width=40, height=10,command=Creat) 

b.pack()
t.protocol("WM_DELETE_WINDOW", JieShu)
t.mainloop()

s.close()