Python 使用socket实现一对多通信

Posted 2020-11-09 希望中追寻

这个折磨了我快一天的时间，查看官网的socket入门例子，都是一对一的，服务器是处理一对一的形式。如果让服务器处理多个客户端，使用Python提供的socketserver函数和select也是可以解决的，但是这两个只能处理客户端发过来的信息，不能自动发起向已经连接到服务器的某个客户端进行通信，经过了多方的查找资料和总结，终于使用socket和threading解决了这个问题。

服务器代码：

　　

 1 from tkinter import *
 2 from socket import *
 3 import threading
 4 
 5 
 6 address=‘0.0.0.0‘
 7 port=9000
 8 buffsize=1024
 9 s = socket(AF_INET, SOCK_STREAM)
10 s.bind((address,port))
11 s.listen(5)     #最大连接数
12 conn_list = []
13 conn_dt = {}
14 
15 def tcplink(sock,addr):
16     while True:
17         try:
18             recvdata=sock.recv(buffsize).decode(‘utf-8‘)
19             print(recvdata, addr)
20             gui.infoList.config(state=NORMAL)
21             gui.infoList.insert(END, addr, ‘name‘)
22             gui.infoList.insert(END, ‘：\t‘)
23             gui.infoList.insert(END, recvdata, ‘conment‘)
24             gui.infoList.insert(END, ‘\n\n‘)
25             gui.infoList.config(state=DISABLED)
26             if not recvdata:
27                 break
28         except:
29             sock.close()
30             print(addr,‘offline‘)
31             _index = conn_list.index(addr)
32             gui.listBox.delete(_index)
33             conn_dt.pop(addr)
34             conn_list.pop(_index)
35             break
36 
37 def recs():
38     while True:
39         clientsock,clientaddress=s.accept()
40         if clientaddress not in conn_list:
41             conn_list.append(clientaddress)
42             conn_dt[clientaddress] = clientsock
43             gui.listBox.insert(END, clientaddress)
44         print(‘connect from:‘,clientaddress)
45         #在这里创建线程，就可以每次都将socket进行保持
46         t=threading.Thread(target=tcplink,args=(clientsock,clientaddress))
47         t.start()
48 
49 
50 class GUI:
51     def __init__(self, root):
52         self.root = root
53         self.leftFrame = Frame(self.root, width=20, height=30)
54         self.leftFrame.grid(row=0, column=0)
55         self.rightFrame = Frame(self.root, width=20, height=30)
56         self.rightFrame.grid(row=0, column=1)
57         Label(self.leftFrame, text=‘在线IP地址列表‘).grid(row=0, column=0)
58         self.listBox = Listbox(self.leftFrame, width=15, height=10)
59         self.listBox.grid(row=1, column=0)
60         self.entry = Entry(self.rightFrame, font=(‘Serief‘, 18), width=30)
61         self.entry.grid(row=0, column=0)
62         self.sendBtn = Button(self.rightFrame, text=‘发送‘, command=self.send, width=10)
63         self.sendBtn.grid(row=0, column=1)
64         Label(self.rightFrame, text=‘聊天信息‘).grid(row=1, columnspan=2)
65         self.infoList = Text(self.rightFrame, width=40, height=12)
66         self.infoList.grid(row=2, columnspan=2)
67         self.infoList.tag_config(‘name‘, background=‘yellow‘, foreground=‘red‘)
68         self.infoList.tag_config(‘conment‘, background=‘black‘, foreground=‘white‘)
69 
70 
71     def send(self):
72         _index = self.listBox.curselection()
73         conn_dt[self.listBox.get(_index)].sendall(self.entry.get().encode(‘utf-8‘))
74         self.entry.delete(0, END)
75 
76 def createGUI():
77     global gui
78     root = Tk()
79     gui = GUI(root)
80     root.title(‘服务器‘)
81     root.mainloop()
82 
83 if __name__ == ‘__main__‘:
84     t1 = threading.Thread(target=recs, args=(), name=‘rec‘)
85     t2 = threading.Thread(target=createGUI, args=(), name=‘GUI‘)
86 
87     t1.start()
88     t2.start()

客户端代码：

　　

 1 from socket import *
 2 import threading
 3 from tkinter import *
 4 
 5 address=‘127.0.0.1‘   #服务器的ip地址
 6 port=9000
 7 buffsize=1024
 8 s=socket(AF_INET, SOCK_STREAM)
 9 s.connect((address,port))
10 
11 
12 
13 def recv():
14     while True:
15         recvdata = s.recv(buffsize).decode(‘utf-8‘)
16         gui.listBox.insert(END, recvdata)
17         print(‘\n‘ + recvdata + ‘\n‘)
18 
19 class GUI:
20     def __init__(self, root):
21         self.root = root
22         self.listBox = Listbox(self.root)
23         self.listBox.pack()
24         self.entry = Entry(self.root)
25         self.entry.pack()
26         self.sendBtn = Button(self.root, text=‘发送‘, command=self.send)
27         self.sendBtn.pack()
28 
29     def send(self):
30         senddata = self.entry.get()
31         s.send(senddata.encode())
32 
33 def createGUI():
34     global gui
35     root = Tk()
36     gui = GUI(root)
37     root.title(‘客户端‘)
38     root.mainloop()
39 
40 if __name__ == ‘__main__‘:
41     t1 = threading.Thread(target=recv, args=(), name=‘recv‘)
42     t2 = threading.Thread(target=createGUI, args=(), name=‘gui‘)
43 
44     t1.start()
45     t2.start()

加上了界面，可以实现服务端与已连接的某个客户端进行通信，可以处理多个客户端的通信，