Python socket服務(wù)常用操作代碼實(shí)例
套接字(socket)是一個(gè)抽象層,應(yīng)用程序可以通過它發(fā)送或接收數(shù)據(jù),可對(duì)其進(jìn)行像對(duì)文件一樣的打開、讀寫和關(guān)閉等操作。
1. 實(shí)現(xiàn)客戶端發(fā)送字符,服務(wù)器返回大寫的字符:
服務(wù)器:
import socketserverclass MyTCPHandler(socketserver.BaseRequestHandler): # 通過類的繼承,實(shí)現(xiàn) def handle(self): # 重寫父類的handle方法,所有的操作都在此方法中 while True: # 循環(huán),不停的接收從客戶端來的數(shù)據(jù) try:self.data = self.request.recv(1024).strip() # 從客戶端接收數(shù)據(jù),每次收1024字節(jié)print('{} send:'.format(self.client_address), self.data)self.request.send(self.data.upper()) # 從服務(wù)器發(fā)送給客戶端數(shù)據(jù) except ConnectionResetError as e:print(’Error: ’,e)breakif __name__ == ’__main__’: host,port = ’localhost’,9999 server = socketserver.ThreadingTCPServer((host,port),MyTCPHandler) # 通過多線程實(shí)現(xiàn)多個(gè)客戶端連接,每個(gè)客戶端連接都是一個(gè)線程 server.serve_forever()# 一直運(yùn)行服務(wù)
客戶端:
import socketclient = socket.socket() # socket對(duì)象client.connect((’localhost’,9999)) # 連接服務(wù)器地址和端口while True: # 循環(huán),不停的輸入發(fā)送數(shù)據(jù) con = input(’>>>:’).strip() if len(con) ==0: continue # 不能發(fā)送空數(shù)據(jù),否則會(huì)阻塞 client.send(con.encode(’utf-8’)) # 發(fā)送數(shù)據(jù),必須是二進(jìn)制的 data = client.recv(1024) # 接收服務(wù)器返回的數(shù)據(jù) print(data.decode())# 打印 解碼后的數(shù)據(jù)client.close() # 關(guān)閉
2. 通過socket執(zhí)行服務(wù)器命令:
用法:直接在客戶端輸入處輸入命令如:ipconfig
服務(wù)器:
import socketimport osimport threadingdef tcplink(sock, addr): print(’Accept new connection from %s:%s...’ % addr) while True: # 和每個(gè)接入的客戶端,進(jìn)行多次數(shù)據(jù)通信 data = sock.recv(1024) # 接收客戶端數(shù)據(jù) if not data or data.decode(’utf-8’) == ’exit’: # 如果客戶端不發(fā)送數(shù)據(jù)或者發(fā)送了exit print(’client disconnected.’) break content = os.popen(data.decode(’utf-8’)).read() # 對(duì)發(fā)送來的數(shù)據(jù)執(zhí)行cmd命令,獲取結(jié)果 if len(content) == 0: #如果執(zhí)行的命令結(jié)果為空的,就手動(dòng)造一個(gè)結(jié)果。因?yàn)槿绻麨榭諗?shù)據(jù),會(huì)掛起,無法正常發(fā)送。 content = ’cmd not exists.’ sock.send(str(len(content.encode(’utf-8’))).encode(’utf-8’)) # 發(fā)送數(shù)據(jù)的長(zhǎng)度 print(’send length:’, (len(content.encode(’utf-8’)))) # print(’content,’, content.encode(’utf-8’)) recv = sock.recv(1024) # 因?yàn)樯舷露加幸粋€(gè)send連在一起,可能發(fā)生粘包現(xiàn)象,為了防止這種情況,可以讓客戶端重新應(yīng)答一下 print(’Answer:’,recv.decode(’utf-8’)) sock.send(content.encode(’utf-8’)) # 發(fā)送數(shù)據(jù) print(’send finished.’) sock.close() print(’Connection from %s:%s closed.’ % addr)s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)# 監(jiān)聽端口:s.bind((’127.0.0.1’, 9999))s.listen(3)print(’Waiting for connection...’)while True: # 接受一個(gè)新連接: sock, addr = s.accept() # 創(chuàng)建新線程來處理TCP連接: t = threading.Thread(target=tcplink, args=(sock, addr)) t.start()
客戶端:
import socket# AF_INET 代表ipv4,SOCK_STREAM 代表TCPs = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 確定網(wǎng)絡(luò)協(xié)議,生成對(duì)象s.connect((’127.0.0.1’,9999)) # 連接服務(wù)器的地址和端口,元組的形式。while True: msg = input(’>>:’).strip() if len(msg) != 0: # 如果消息為空,會(huì)一直掛起,所以不能為空 if msg ==’exit’: s.close() # 關(guān)閉連接 print(’Connection closed.’) break s.send(msg.encode(’utf-8’)) # 給服務(wù)器發(fā)送數(shù)據(jù),必須是二進(jìn)制的 length = s.recv(1024) # 首先接收服務(wù)器返回的將要接收的數(shù)據(jù)的長(zhǎng)度信息。 s.send(b’Ready to receive...’) # 發(fā)送接收命令 length = int(length.decode(’utf-8’)) print(’receive len:’, length) data_len = 0 data_recv = b’’ while data_len < length: # 已經(jīng)接收的信息的長(zhǎng)度,如果小于總長(zhǎng)度 data = s.recv(1024) # 從服務(wù)器接收數(shù)據(jù) data_recv += data data_len += len(data) print(data_recv.decode(’utf-8’)) # 打印返回的數(shù)據(jù)。
3. 通過socket傳輸文件:
用法:get 文件名
服務(wù)器:
import socketimport osimport hashlibimport threadingdef tcplink(sock, addr): print(’Accept new connection from %s:%s...’ % addr) while True: # 和每個(gè)接入的客戶端,進(jìn)行多次數(shù)據(jù)通信 data = sock.recv(1024) # 接收客戶端數(shù)據(jù) if not data or data.decode(’utf-8’) == ’exit’: # 如果客戶端不發(fā)送數(shù)據(jù)或者發(fā)送了exit print(’client disconnected.’) break oper,filename = data.decode(’utf-8’).split() # 對(duì)接收的數(shù)據(jù)按照空格分割 if oper == ’get’: m = hashlib.md5() if os.path.isfile(filename):size = os.stat(filename).st_size # 獲取文件大小print(’Send size:’,size)sock.send(str(size).encode(’utf-8’)) # 發(fā)送文件大小recv = sock.recv(1024) # 接收客戶端確認(rèn)信息(因?yàn)樯舷挛膬蓚€(gè)send是連著的,所以為了防止粘包,接收一次信息)f = open(filename,’rb’)for line in f: sock.send(line) #讀取文件,發(fā)送給客戶端 m.update(line)# print(’Send finished.’,m.hexdigest()) # 打印md5的值sock.send(m.hexdigest().encode(’utf-8’)) # 把md5的值發(fā)送給客戶端 sock.close() print(’Connection from %s:%s closed.’ % addr)s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)# 監(jiān)聽端口:s.bind((’127.0.0.1’, 9999))s.listen(3)print(’Waiting for connection...’)while True: # 接受一個(gè)新連接: sock, addr = s.accept() # 創(chuàng)建新線程來處理TCP連接: t = threading.Thread(target=tcplink, args=(sock, addr)) t.start()
客戶端:
import socketimport hashlib# AF_INET 代表ipv4,SOCK_STREAM 代表TCPs = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 確定網(wǎng)絡(luò)協(xié)議,生成對(duì)象s.connect((’127.0.0.1’,9999)) # 連接服務(wù)器的地址和端口,元組的形式。while True: msg = input(’>>:’).strip() if len(msg) != 0: # 如果消息為空,會(huì)一直掛起,所以不能為空 if msg ==’exit’: s.close() # 關(guān)閉連接 print(’Connection closed.’) break s.send(msg.encode(’utf-8’)) # 給服務(wù)器發(fā)送數(shù)據(jù),必須是二進(jìn)制的 length = s.recv(1024) # 首先接收服務(wù)器返回的將要接收的數(shù)據(jù)的長(zhǎng)度信息。 s.send(b’Ready to receive...’) # 發(fā)送接收確認(rèn)命令 length = int(length.decode(’utf-8’)) print(’Recv size:’, length) data_len = 0 data_recv = b’’ # 新文件名 fileName = msg.split()[-1].split(’.’)[0] fileExt = msg.split()[-1].split(’.’)[-1] newFile = fileName+’-1.’+fileExt f = open(newFile,’wb’) # 打開文件,準(zhǔn)備寫入服務(wù)器發(fā)過來的文件 m = hashlib.md5() while data_len < length: # 已經(jīng)接收的信息的長(zhǎng)度,如果小于總長(zhǎng)度 size = length - data_len if size > 1024: # 如果剩下的信息長(zhǎng)度大于1024,即不能一次性發(fā)完。size = 1024 else: # 如果能一次性發(fā)完,就只收剩下的信息。目的是準(zhǔn)確的接收文件的大小,把可能粘連的send的數(shù)據(jù)留給下一次recvsize = length-data_len data = s.recv(size) # 從服務(wù)器接收數(shù)據(jù) f.write(data) m.update(data) data_len += len(data) f.close() print(’recv_md5:’,m.hexdigest()) # 打印返回的數(shù)據(jù)。 recv = s.recv(1024) # 接收下一次send的數(shù)據(jù),即md5的值。 print(’orig_md5:’,recv.decode())
以上就是本文的全部?jī)?nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持好吧啦網(wǎng)。
相關(guān)文章:
1. 用xslt+css讓RSS顯示的跟網(wǎng)頁一樣漂亮2. ASP.NET MVC把數(shù)據(jù)庫中枚舉項(xiàng)的數(shù)字轉(zhuǎn)換成文字3. 《CSS3實(shí)戰(zhàn)》筆記--漸變?cè)O(shè)計(jì)(一)4. 測(cè)試模式 - XSL教程 - 55. Ajax實(shí)現(xiàn)異步加載數(shù)據(jù)6. 教你JS更簡(jiǎn)單的獲取表單中數(shù)據(jù)(formdata)7. ASP.NET Core自定義中間件的方式詳解8. html5手機(jī)觸屏touch事件介紹9. CSS3實(shí)現(xiàn)動(dòng)態(tài)翻牌效果 仿百度貼吧3D翻牌一次動(dòng)畫特效10. 讓chatgpt將html中的圖片轉(zhuǎn)為base64方法示例
