您好,登錄后才能下訂單哦!
本文實例為大家分享了python3實現CS架構文件傳輸的具體代碼,供大家參考,具體內容如下
1、目標:
基于tcp實現CS架構的文件傳輸
指令列表:(1)get:從服務器端下載文件
(2)put:向服務器端上傳文件
(3)list:獲得服務器端的目錄
2、socket模塊函數:
(1)send和sendall:send的作用是發送TCP數據,返回發送的數據大小。send函數不保證將所有數據全部發送,因此可能需要重復多次才能完成所有數據的發送。sendall的作用是發送完整的TCP數據,成功時返回None,失敗時拋出異常
(2)bind:在服務器端使用,用于將socket綁定在一個特定的ip地址和端口上。在《UNIX網絡編程》一書中提到,如果調用connect或者listen之前沒有bind一個特定的端口,內核會為相應的套接字分配一個隨機的端口。因此,在客戶端調用connect函數之前不需要bind
(3)listen:通過參數設定服務器端最多可以接受幾個客戶端的連接,但是只有在完成與第一個客戶端的傳送后才會進行與第二個客戶端的傳送
3、代碼:
(1)服務器端:
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()
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持億速云。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。