Python socket服務(wù)常用操作代碼實(shí)例
套接字(socket)是一個(gè)抽象層,應(yīng)用程序可以通過(guò)它發(fā)送或接收數(shù)據(jù),可對(duì)其進(jìn)行像對(duì)文件一樣的打開(kāi)、讀寫(xiě)和關(guān)閉等操作。
1. 實(shí)現(xiàn)客戶端發(fā)送字符,服務(wù)器返回大寫(xiě)的字符:
服務(wù)器:
import socketserverclass MyTCPHandler(socketserver.BaseRequestHandler): # 通過(guò)類的繼承,實(shí)現(xiàn) def handle(self): # 重寫(xiě)父類的handle方法,所有的操作都在此方法中 while True: # 循環(huán),不停的接收從客戶端來(lái)的數(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) # 通過(guò)多線程實(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. 通過(guò)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ā)送來(lái)的數(shù)據(jù)執(zhí)行cmd命令,獲取結(jié)果 if len(content) == 0: #如果執(zhí)行的命令結(jié)果為空的,就手動(dòng)造一個(gè)結(jié)果。因?yàn)槿绻麨榭諗?shù)據(jù),會(huì)掛起,無(wú)法正常發(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)聽(tīng)端口:s.bind((’127.0.0.1’, 9999))s.listen(3)print(’Waiting for connection...’)while True: # 接受一個(gè)新連接: sock, addr = s.accept() # 創(chuàng)建新線程來(lái)處理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. 通過(guò)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)聽(tīng)端口:s.bind((’127.0.0.1’, 9999))s.listen(3)print(’Waiting for connection...’)while True: # 接受一個(gè)新連接: sock, addr = s.accept() # 創(chuàng)建新線程來(lái)處理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’) # 打開(kāi)文件,準(zhǔn)備寫(xiě)入服務(wù)器發(fā)過(guò)來(lái)的文件 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. CSS3實(shí)例分享之多重背景的實(shí)現(xiàn)(Multiple backgrounds)2. XML入門的常見(jiàn)問(wèn)題(四)3. vue實(shí)現(xiàn)復(fù)制文字復(fù)制圖片實(shí)例詳解4. XML基本概念XPath、XSLT與XQuery函數(shù)介紹5. 前端html+css實(shí)現(xiàn)動(dòng)態(tài)生日快樂(lè)代碼6. 關(guān)于html嵌入xml數(shù)據(jù)島如何穿過(guò)樹(shù)形結(jié)構(gòu)關(guān)系的問(wèn)題7. WML的簡(jiǎn)單例子及編輯、測(cè)試方法第1/2頁(yè)8. 不要在HTML中濫用div9. el-input無(wú)法輸入的問(wèn)題和表單驗(yàn)證失敗問(wèn)題解決10. XML入門的常見(jiàn)問(wèn)題(三)
