Python实现TCP文件传输
废话少说，直接上代码
1 # 发送端 2 import socket,os 3 from struct import pack 4 5 def send_file(file_name,file_socket:socket.socket): 6 try: 7 f = open(file_name,'rb') 8 size = os.path.getsize(file_name) 9 if size < 1024: 10 read_size = 500 11 elif size < 1024*1024 and size >= 1024: 12 read_size = 500*1024 13 else: 14 read_size = 500*1024*1024 15 file_socket.send(pack('q',size)) 16 del size 17 file_socket.recv(1024) 18 while True: 19 data = f.read(read_size) 20 if not data: 21 break 22 file_socket.send(data) 23 f.close() 24 except FileNotFoundError: 25 print(f'没有找到{file_name}') 26 27 if __name__ == '__main__': 28 try: 29 file_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM) 30 file_socket.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,True) 31 ip = str(input('请输入接收端IP：')) 32 port = int(input('请输入接收端端口号：')) 33 file_socket.connect((ip,port)) 34 print('连接成功，准备开始传输。。。') 35 path = r'%s' % (input('请输入文件所在文件夹：') + '/') 36 print(file_socket.recv(1024).decode('gbk')) 37 file_name = str(input()) 38 file_socket.send(file_name.encode('gbk')) 39 send_file(path + file_name,file_socket) 40 except ConnectionResetError: 41 print('接收端断开连接')
1 # 接收端 2 import socket,os 3 from time import time 4 from struct import unpack 5 from tqdm import tqdm 6 7 DOWNLOAD_PATH = 'D:/Python文件传输/' # 传输目录 8 9 def download(file_name,file_socket:socket.socket): 10 file_size = unpack('q',file_socket.recv(1024))[0] 11 if file_size < 1024: 12 print(f'文件大小：{file_size} B') 13 elif file_size < 1024*1024 and file_size >= 1024: 14 print(f'文件大小：{file_size/1024} KB') 15 else: 16 print(f'文件大小：{file_size/(1024*1024)} MB') 17 f = open(file_name,'wb') 18 print('开始传输...') 19 download_size = 2048 20 file_socket.send('开始传输'.encode('gbk')) 21 start = time() 22 for i in tqdm(range(int(file_size/download_size) + 1)): 23 data = file_socket.recv(download_size) 24 f.write(data) 25 end = time() 26 f.close() 27 print('传输完成！') 28 print(f'大约耗时{end - start}秒') 29 30 if __name__ == '__main__': 31 os.chdir(DOWNLOAD_PATH) 32 try: 33 file_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM) 34 file_socket.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,True) 35 file_socket_port = int(input('请输入端口号：')) 36 file_socket.bind(('',file_socket_port)) 37 print('成功启动，等待连接。。。') 38 file_socket.listen(1) 39 f_socket, f_addr = file_socket.accept() 40 print(f'建立连接{f_addr}') 41 f_socket.send('请输入文件名'.encode('gbk')) 42 file_name = f_socket.recv(1024) 43 download(file_name.decode('gbk'),f_socket) 44 f_socket.close() 45 file_socket.close() 46 except ConnectionResetError: 47 print('发送端已断开连接') 48 except UnicodeDecodeError: 49 print('文件编码错误，请检查文件格式是否正确')
