样例1 简单测试
发送端
import socket import os import time filename = input('please enter the filename you want to send:\n') filesize = str(os.path.getsize(filename)) fname1, fname2 = os.path.split(filename) client_addr = ('127.0.0.1',9999) f = open(filename,'rb') count = 0 flag = 1 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) #建立连接: s.connect(('127.0.0.1', 9999)) while True: if count == 0: s.send(filesize.encode()) start = time.time() s.recv(1024) s.send(fname2.encode()) for line in f: s.send(line) print('sending...') s.send(b'end') break s.close end = time.time() print('cost' + str(round(end - start, 2)) + 's')
接收端
import socket import time s = socket.socket(socket.AF_INET,socket.SOCK_STREAM) s.bind(('127.0.0.1',9999)) s.listen(5) print('Waiting for connection...') count=0 while True: sock,addr = s.accept() print('Accept new connection from %s:%s...' % addr) if count == 0: data1 = sock.recv(1024) print(str(data1)) file_total_size = int(data1.decode()) received_size = 0 sock.send('received'.encode()) data = sock.recv(1024) filepath = str(data.decode()) f = open(filepath, 'wb') while received_size < file_total_size: data = sock.recv(1024) f.write(data) received_size += len(data) print('已接收 ',received_size,' Byte') data = sock.recv(1024) if data == b'end': break f.close() s.close()
发送端输入
send/测试文档.docx
接收端保存在
样例2 封装
import socket import time def socketInt(): s = socket.socket(socket.AF_INET,socket.SOCK_STREAM) s.bind(('127.0.0.1',9999)) s.listen(5) print('Waiting for connection...') return s def socketRec(sockSever,savePath): s=sockSever sock,addr = s.accept() print('Accept new connection from %s:%s...' % addr) data1 = sock.recv(1024) print(str(data1)) file_total_size = int(data1.decode()) received_size = 0 sock.send('received'.encode()) data = sock.recv(1024) filepath = str(data.decode()) filepath=savePath+filepath f = open(filepath, 'wb') while received_size < file_total_size: data = sock.recv(1024) f.write(data) received_size += len(data) print('已接收 ',received_size,' Byte') data = sock.recv(1024) if data == b'end': print("接受文件",filepath) #f.close() #break f.close() if __name__ == "__main__": savePath="rec/" # 前面不可以加/ sockSever=socketInt() while True: socketRec(sockSever,savePath) sockSever.close()
发送端
import socket import os import time def SendFile(filename): #filename = input('please enter the filename you want to send:\n') filesize = str(os.path.getsize(filename)) fname1, fname2 = os.path.split(filename) f = open(filename,'rb') count = 0 flag = 1 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) #建立连接: s.connect(('127.0.0.1', 9999)) while True: if count == 0: s.send(filesize.encode()) start = time.time() s.recv(1024) s.send(fname2.encode()) for line in f: s.send(line) print('sending...') s.send(b'end') break s.close end = time.time() print('cost' + str(round(end - start, 2)) + 's') if __name__ == "__main__": while True: filename = input('please enter the filename you want to send:\n') SendFile(filename)