本文實(shí)例為大家分享了python3實(shí)現(xiàn)CS架構(gòu)文件傳輸?shù)木唧w代碼，供大家參考，具體內(nèi)容如下
1、目標(biāo)：
基于tcp實(shí)現(xiàn)CS架構(gòu)的文件傳輸
指令列表：（1）get：從服務(wù)器端下載文件
（2）put：向服務(wù)器端上傳文件
（3）list：獲得服務(wù)器端的目錄
2、socket模塊函數(shù)：
（1）send和sendall：send的作用是發(fā)送TCP數(shù)據(jù)，返回發(fā)送的數(shù)據(jù)大小。send函數(shù)不保證將所有數(shù)據(jù)全部發(fā)送，因此可能需要重復(fù)多次才能完成所有數(shù)據(jù)的發(fā)送。sendall的作用是發(fā)送完整的TCP數(shù)據(jù)，成功時(shí)返回None，失敗時(shí)拋出異常
（2）bind：在服務(wù)器端使用，用于將socket綁定在一個(gè)特定的ip地址和端口上。在《UNIX網(wǎng)絡(luò)編程》一書中提到，如果調(diào)用connect或者listen之前沒有bind一個(gè)特定的端口，內(nèi)核會(huì)為相應(yīng)的套接字分配一個(gè)隨機(jī)的端口。因此，在客戶端調(diào)用connect函數(shù)之前不需要bind
（3）listen：通過參數(shù)設(shè)定服務(wù)器端最多可以接受幾個(gè)客戶端的連接，但是只有在完成與第一個(gè)客戶端的傳送后才會(huì)進(jìn)行與第二個(gè)客戶端的傳送
3、代碼：
（1）服務(wù)器端：
import socket import os import sys import time Host = '127.0.0.1' Port = 12000 Addr = (Host, Port) sockListener = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sockListener.bind(Addr) sockListener.listen(5) def recvfile(cliSocket): print('start reveiving file...') msg = 'no problem' msg_utf8 = msg.encode(encoding="utf-8") cliSocket.send(msg_utf8) filename_utf8 = clientSocket.recv(4096) filename = filename_utf8.decode() f = open(filename, 'wb') while True: data_utf8 = cliSocket.recv(4096) data = data_utf8.decode() if data=='EOF': print('received file successfully') break f.write(data_utf8) f.close() def sendfile(cliSocket): print('start sending file...') msg = 'no problem' msg_utf8 = msg.encode(encoding="utf-8") cliSocket.send(msg_utf8) filename_utf8 = cliSocket.recv(4096) filename = filename_utf8.decode() f = open(filename, 'rb') while True: data = f.read(4096) if not data: break cliSocket.sendall(data) f.close() time.sleep(1) msg = 'EOF' msg_utf8 = msg.encode(encoding="utf-8") cliSocket.sendall(msg_utf8) print('sent file successfully') def getList(cliSocket): path = sys.path[0] every_file = os.listdir(path) for filename in every_file: pathmsg = filename + '/n' pathmsg_utf8 = pathmsg.encode(encoding="utf-8") cliSocket.sendall(pathmsg_utf8) time.sleep(1) msg = 'EOF' msg_utf8 = msg.encode(encoding="utf-8") cliSocket.sendall(msg_utf8) print('all filenames have been send') while True: print('waiting for connection...') clientSocket, addr = sockListener.accept() print('... connection from:', addr) while True: msg_utf8 = clientSocket.recv(4096) msg = msg_utf8.decode() if msg=='exit': print(addr, 'close the connection') break elif msg=='get': sendfile(clientSocket) elif msg=='put': recvfile(clientSocket) elif msg=='list': getList(clientSocket) else: print('client error!') break （2）客戶端：
import socket import time Host = '127.0.0.1' Port = 12000 Addr = (Host, Port) clientSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) clientSocket.connect(Addr) def recvfile(filename): print('start receiving file...') f = open(filename, 'wb') filename_utf8 = filename.encode(encoding="utf-8") clientSocket.sendall(filename_utf8) while True: data_utf8 = clientSocket.recv(4096) data=data_utf8.decode() if data=='EOF': print('receive file successfully') break f.write(data_utf8) f.close() def sendfile(filename): print('start sending file...') f = open(filename, 'rb') filename_utf8 = filename.encode(encoding="utf-8") clientSocket.sendall(filename_utf8) while True: data = f.read(4096) if not data: break clientSocket.sendall(data) f.close() time.sleep(1) endmsg = 'EOF' endmsg_utf8 = endmsg.encode(encoding="utf-8") clientSocket.sendall(endmsg_utf8) print('send file successfully') def confirm(confirm_command): confirm_command_utf8 = confirm_command.encode(encoding="utf-8") clientSocket.sendall(confirm_command_utf8) msg_utf8 = clientSocket.recv(4096) msg = msg_utf8.decode() print('reveive message:', msg) if msg=='no problem': return True else: return False def operation1(filename): if confirm('get'): recvfile(filename) else: print('serve error!') def operation2(filename): if confirm('put'): sendfile(filename) else: print('serve error!') def operation3(act): if act=='list': act_utf8 = act.encode(encoding="utf-8") clientSocket.sendall(act_utf8) while True: msg_utf8 = clientSocket.recv(1024) msg = msg_utf8.decode() if msg=='EOF': break print(msg) else: print('wrong command!') try: while True: command = input('>>>') if not command: continue elif command=='exit': command_utf8 = command.encode(encoding="utf-8") clientSocket.sendall(command_utf8) print('the connection is closed') break msg = command.split() if len(msg)==2 and msg[0]=='get': operation1(msg[1]) elif len(msg)==2 and msg[0]=='put': operation2(msg[1]) elif len(msg)==1: operation3(msg[0]) else: print('wrong command!') except socket.error as e: print('error:', e) print('an error causes the connection to close!') clientSocket.close() 以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持VEVB武林網(wǎng)。
新聞熱點(diǎn)
疑難解答
圖片精選
