FTP上传文件服务器python实现

总结

使用Python的socket和socketserver模块实现ftp文件服务器。不是很难，但在粘包的问题上纠缠了不少时间。解决办法是发送文件信息之后等待一段时间。

#!/usr/bin/python
#coding:utf-8
'''
ftp文件上传 服务器端
'''

import socketserver
import os


class myFtpServer(socketserver.BaseRequestHandler):
    def setup(self):
        pass

    def handle(self):
        dir = r'F:	emp' #存放接收文件的目录

        print('Connected %s:%d' %self.client_address)
        while True: #持续接收文件
            #接收文件的名称和大小
            fileInfo = self.request.recv(1024)
            if fileInfo.decode('utf-8') == 'exit':
                break
            filename, filesize = fileInfo.decode('utf-8').strip().split('|')
            filePath = os.path.join(dir, filename) #创建文件存放路径
            print('Receiving file:%s, filesize:%s'%(filename, filesize))
            with open(filePath, 'wb') as f:
                recvsize = 0
                while True: #持续接收文件内容
                    if recvsize == int(filesize): #当接收的大小与文件大小相等时结束接收
                        break 
                    self.data = self.request.recv(1024)
                    recvsize += len(self.data)
                    f.write(self.data)
                print('upload succeed!!')
                print(recvsize)
        print('Disconnected %s:%d.' %self.client_address)
    '''
    def handle(self):
        base_path = ''
        conn = self.request
        print( 'connected...')
        while True:
            pre_data = conn.recv(1024)
            #获取请求方法、文件名、文件大小
            file_name,file_size = pre_data.decode('utf-8').split('|')
            
            #已经接收文件的大小
            recv_size = 0
            #上传文件路径拼接
            file_dir = os.path.join(base_path,file_name)
            f = open(file_dir,'wb')
            Flag = True
            while Flag:
                #未上传完毕，
                if int(file_size)>recv_size:
                    #最多接收1024，可能接收的小于1024
                    data = conn.recv(1024) 
                    recv_size+=len(data)
                #上传完毕，则退出循环
                else:
                    recv_size = 0
                    Flag = False
                #写入文件
                f.write(data)
            print('upload successed.')
            f.close()            
    def finish(self):
        pass
    
    '''
if __name__ == '__main__':
    addr = ('127.0.0.1', 9999)
    sk = socketserver.ThreadingTCPServer(addr, myFtpServer)
    sk.serve_forever()

#!/usr/bin/python
#coding:utf-8
'''
ftp文件上传 客户端
'''
import socket
import sys
import os
import time

def ftp(address):
    sk = socket.socket(family=socket.AF_INET, type=socket.SOCK_STREAM)
    sk.connect(address)
    while True:
        #filePath = r'C:UsersadminDesktop	est'
        filePath = input('Input file path：')
        if filePath == 'exit': #判断退出
            sk.send(b'exit')
            break
        if not os.path.isfile(filePath): #判断是否文件
            print('Not available file path...')
            continue
        filename = os.path.basename(filePath)
        #dir = os.path.dirname(filePath)
        
        filesize = os.stat(filePath).st_size #获取文件大小
        fileInfo = '%s|%s'%(filename, filesize) #重新组织文件信息，包括文件名和文件大小，用‘|’分割
        
        #发送文件信息
        sk.send(fileInfo.encode(encoding='utf-8', errors='strict'))
        time.sleep(0.01) #为了防止粘包，在文件信息发送之后等待0.01秒
        
        #持续发送文件内容
        with open(filePath, 'rb') as f:
            sendsize = 0
            flag = True
            while flag:
                if sendsize + 1024 > filesize: #判断最后一份文件内容的读取
                    data = f.read(filesize - sendsize)
                    flag = False
                else: #一般文件内容的读取
                    data = f.read(1024)
                    sendsize += 1024
                sk.send(data)
    
    sk.close()

if __name__ == '__main__':
    addr = ('127.0.0.1',9999)
    ftp(addr)

环境：python3.5 win10 x64