Python socket服務(wù)常用操作代碼實例
套接字(socket)是一個抽象層,應(yīng)用程序可以通過它發(fā)送或接收數(shù)據(jù),可對其進行像對文件一樣的打開、讀寫和關(guān)閉等操作。
1. 實現(xiàn)客戶端發(fā)送字符,服務(wù)器返回大寫的字符:
服務(wù)器:
import socketserverclass MyTCPHandler(socketserver.BaseRequestHandler): # 通過類的繼承,實現(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) # 通過多線程實現(xiàn)多個客戶端連接,每個客戶端連接都是一個線程 server.serve_forever()# 一直運行服務(wù)
客戶端:
import socketclient = socket.socket() # socket對象client.connect((’localhost’,9999)) # 連接服務(wù)器地址和端口while True: # 循環(huán),不停的輸入發(fā)送數(shù)據(jù) con = input(’>>>:’).strip() if len(con) ==0: continue # 不能發(fā)送空數(shù)據(jù),否則會阻塞 client.send(con.encode(’utf-8’)) # 發(fā)送數(shù)據(jù),必須是二進制的 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: # 和每個接入的客戶端,進行多次數(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() # 對發(fā)送來的數(shù)據(jù)執(zhí)行cmd命令,獲取結(jié)果 if len(content) == 0: #如果執(zhí)行的命令結(jié)果為空的,就手動造一個結(jié)果。因為如果為空數(shù)據(jù),會掛起,無法正常發(fā)送。 content = ’cmd not exists.’ sock.send(str(len(content.encode(’utf-8’))).encode(’utf-8’)) # 發(fā)送數(shù)據(jù)的長度 print(’send length:’, (len(content.encode(’utf-8’)))) # print(’content,’, content.encode(’utf-8’)) recv = sock.recv(1024) # 因為上下都有一個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: # 接受一個新連接: 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é)議,生成對象s.connect((’127.0.0.1’,9999)) # 連接服務(wù)器的地址和端口,元組的形式。while True: msg = input(’>>:’).strip() if len(msg) != 0: # 如果消息為空,會一直掛起,所以不能為空 if msg ==’exit’: s.close() # 關(guān)閉連接 print(’Connection closed.’) break s.send(msg.encode(’utf-8’)) # 給服務(wù)器發(fā)送數(shù)據(jù),必須是二進制的 length = s.recv(1024) # 首先接收服務(wù)器返回的將要接收的數(shù)據(jù)的長度信息。 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)接收的信息的長度,如果小于總長度 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: # 和每個接入的客戶端,進行多次數(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() # 對接收的數(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) # 接收客戶端確認信息(因為上下文兩個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: # 接受一個新連接: 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é)議,生成對象s.connect((’127.0.0.1’,9999)) # 連接服務(wù)器的地址和端口,元組的形式。while True: msg = input(’>>:’).strip() if len(msg) != 0: # 如果消息為空,會一直掛起,所以不能為空 if msg ==’exit’: s.close() # 關(guān)閉連接 print(’Connection closed.’) break s.send(msg.encode(’utf-8’)) # 給服務(wù)器發(fā)送數(shù)據(jù),必須是二進制的 length = s.recv(1024) # 首先接收服務(wù)器返回的將要接收的數(shù)據(jù)的長度信息。 s.send(b’Ready to receive...’) # 發(fā)送接收確認命令 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)接收的信息的長度,如果小于總長度 size = length - data_len if size > 1024: # 如果剩下的信息長度大于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())
以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持好吧啦網(wǎng)。
相關(guān)文章:
1. 在Android中使用WebSocket實現(xiàn)消息通信的方法詳解2. python matplotlib:plt.scatter() 大小和顏色參數(shù)詳解3. Yii2.0引入CSS,JS文件方法4. JSP數(shù)據(jù)交互實現(xiàn)過程解析5. Python importlib動態(tài)導(dǎo)入模塊實現(xiàn)代碼6. vue使用webSocket更新實時天氣的方法7. 淺談python出錯時traceback的解讀8. android studio 打包自動生成版本號與日期,apk輸入路徑詳解9. Nginx+php配置文件及原理解析10. JavaMail 1.4 發(fā)布
