python发送文件

 

 样例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)