socket与socketserver实现简的ftp文件上传下载、断点续传功能

1、配置文件

IP_PORT = ('localhost', 50009)

2、client文件

import socket
from config.settings import IP_PORT
import os

class FtpSocketClient(object):
    def __init__(self):
        self.ftpclient = socket.socket()
        self.ftpclient.connect(IP_PORT)
        server_reply = self.ftpclient.recv(1024)
        print(str(server_reply, encoding='utf-8'))

    def updata(self, path):
        """上传文件"""
        print('准备发送文件')
        file_size = os.path.getsize(path)
        data = '{0}|{1}|{2}'.format('accpet', path, file_size)
        self.ftpclient.send(bytes(data, encoding='utf-8'))
        server_reply = self.ftpclient.recv(1024)
        if str(server_reply, encoding='utf-8') == 'ok':
            size = 0
            with open('test.txt', 'rb') as f:
                while size < file_size:
                    file_data = f.read(500)
                    self.ftpclient.send(file_data)
                    size += len(file_data)
                else:
                    print('文件发送完成')

    def download(self, path):
        """下载文件"""
        if os.path.exists(path):
            with open(path, 'ab') as f:
                p = f.tell()
                data = '{0}|{1}|{2}'.format('recv', path, p)
                self.ftpclient.sendall(bytes(data, encoding='utf-8'))
                file_size = str(self.ftpclient.recv(1024), encoding='utf-8')
                if p >= int(file_size):
                    print('文件已经完整存在')
                    return
                else:
                    self.ftpclient.send(bytes('ok', encoding='utf-8'))
                    while f.tell() != int(file_size):
                        file_data = self.ftpclient.recv(500)
                        f.write(file_data)
                    else:
                        print('文件接收完成！')
        else:
            data = '{0}|{1}|{2}'.format('recv', path, 0)
            self.ftpclient.sendall(bytes(data, encoding='utf-8'))
            file_size = str(self.ftpclient.recv(1024), encoding='utf-8')
            with open(path, 'wb') as f:
                self.ftpclient.send(bytes('ok', encoding='utf-8'))
                size = 0
                while size != int(file_size):
                    file_data = self.ftpclient.recv(500)
                    f.write(file_data)
                    size += len(file_data)
                else:
                    print('文件接收完成！')
                    
    def close(self):
        self.ftpclient.close()



if __name__ == '__main__':
    myClient = FtpSocketClient()
    # myClient.updata('test.txt')
    myClient.download('test.txt')
    myClient.close()

3、server文件

import socketserver
from config.settings import IP_PORT
import os




class MyTCPHandler(socketserver.BaseRequestHandler):
    def handle(self):
        print('New Conn:', self.client_address)
        self.request.send(bytes('链接成功', 'utf-8'))
        while True:
            try:
                data = self.request.recv(1024)
            except:
                data = ''
            if not data:
                break
            cmd_list = data.decode().split('|')
            if hasattr(MyTCPHandler, cmd_list[0]):
                fun = getattr(MyTCPHandler, cmd_list[0])
                fun(self, cmd_list[1:])
            else:
                 print('cmd is error')

    def accpet(self, file_info):
        """收文件"""
        with open(file_info[0], 'wb') as f:
            self.request.send(bytes('ok', encoding='utf-8'))
            size = 0
            while size != int(file_info[1]):
                file_data = self.request.recv(500)
                f.write(file_data)
                size += len(file_data)
            else:
                print('文件接收完成！')


    def recv(self, path):
        """发文件"""
        p = os.path.join(os.getcwd(), path[0])
        if os.path.exists(p):
            file_size = os.path.getsize(p)
            self.request.sendall(bytes(str(file_size), encoding='utf-8'))
            server_reply = self.request.recv(1024)
            if str(server_reply, encoding='utf-8') == 'ok':
                size = int(path[1])
                with open('test.txt', 'rb') as f:
                    f.seek(size)
                    while size < file_size:
                        file_data = f.read(500)
                        self.request.send(file_data)
                        size += len(file_data)
                    else:
                        print('文件发送完成')
        else:
            print('文件不存在')

if __name__ == '__main__':
    server = socketserver.ThreadingTCPServer(IP_PORT, MyTCPHandler)
    server.serve_forever()