Python实现FTP命令执行和文件的断点续传

川长思鸟来 2022-02-05 12:27 194阅读 0赞

### 一.实现功能 ###

1.客户端发送cmd命令给服务端，服务端返回执行结果给客户端

2.客户端上传文件到服务端，上传过程中如出现故障没有上传完整，下次上传可以接着上次的上传结果继续上传

### 二.流程图 ###

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xpbmNvY280OQ_size_16_color_FFFFFF_t_70][]

### 三.文件目录 ###

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xpbmNvY280OQ_size_16_color_FFFFFF_t_70 1][]

### 四.server\_.py ###

#!/usr/bin/env python
    # -*- coding:utf-8 -*-
    
    
    import socketserver
    import subprocess
    import os
    
    # IP 和 端口号PORT需要自己填
    IP_PORT = ('10.10.10.10', 9999)
    BASE_PATH = os.path.dirname(__file__)
    
    
    class Myserver(socketserver.BaseRequestHandler):
    
        def handle(self):
            self.request.sendall(bytes('欢迎光临！', 'utf-8'))
            while True:
                try:
                    recv_command = str(self.request.recv(1024), 'utf-8')
                    if recv_command.startswith('cmd'):
                        # 数据格式：cmd|dir
                        self.handle_cmd(recv_command)
                    elif recv_command.startswith('put'):
                        # 数据格式：put|file_size|file_name|despath
                        self.handle_put(recv_command)
                except ConnectionResetError:
                    break
    
        def handle_cmd(self, inp):
            """
            处理cmd命令
            :param inp: 输入字符串
            :return:
            """
            command = inp.split('|', 1)[1]
            result_str = subprocess.getoutput(command)
            result_bytes = bytes(result_str, 'utf-8')
            len_bytes = len(result_bytes)
            self.request.sendall(bytes(str(len_bytes), 'utf-8'))
            self.request.recv(1024)
            self.request.sendall(result_bytes)
    
        def handle_put(self, inp):
            """
            文件断点续传
            :param inp: 输入字符串
            :return:
            """
            func, file_size, file_name, despath = inp.split('|', 3)
            path = os.path.join(BASE_PATH, despath, file_name)
            print(BASE_PATH, despath, file_name)
            file_size = int(file_size)
            has_recv = 0
    
            if os.path.exists(path):
                # 文件已存在
                self.request.sendall(bytes('300', 'utf-8'))
                is_continue = str(self.request.recv(1024), 'utf-8')
                if is_continue == '302':
                    # 续传
                    f = open(path, 'ab')
                    exist_size = os.stat(path).st_size
                    self.request.sendall(bytes(str(exist_size), 'utf-8'))
                    has_recv += exist_size
                elif is_continue == '303':
                    # 不续传
                    f = open(path, 'wb')
            else:
                # 文件不存在
                self.request.sendall(bytes('301', 'utf-8'))
                ack = self.request.recv(1024)
                f = open(path, 'wb')
    
            self.request.sendall(bytes('send_ack', 'utf-8'))
    
            f.seek(has_recv)  # 文件指针重定向
            while has_recv < file_size:
                data = self.request.recv(1024)
                f.write(data)
                has_recv += len(data)
            f.close()
    
    
    if __name__ == '__main__':
        server = socketserver.ThreadingTCPServer(IP_PORT, Myserver)
        server.serve_forever()

### 五.client\_.py ###

#!/usr/bin/env python
    # -*- coding:utf-8 -*-
    
    import socket
    import os
    
    # IP 和 端口号PORT需要自己填
    IP_PORT = ('10.10.10.10', 9999)
    
    # 创建socket对象
    c_sk = socket.socket()
    c_sk.connect(IP_PORT)
    print(str(c_sk.recv(1024), 'utf-8'))
    
    
    def handle_cmd(inp, sk):
        """
        处理cmd命令
        :param inp: 用户输入的字符串
        :param sk: 客户端的socket对象
        :return:
        """
        # 发送命令
        sk.sendall(bytes(inp, 'utf-8'))
        len_bytes = int(str(sk.recv(1024), 'utf-8'))
        sk.sendall(bytes('ack', 'utf-8'))
        has_recv = 0
        data_bytes = bytes()
        # 接收返回结果
        while has_recv < len_bytes:
            data = sk.recv(1024)
            has_recv += len(data)
            data_bytes = data_bytes + data
        print(str(data_bytes, 'utf-8'))
    
    
    def handle_put(inp, sk):
        """
        文件的断点续传
        :param inp: 用户输入的字符串
        :param sk: 客户端的socket对象
        :return:
        """
        # 处理用户输入数据
        func, file_path = inp.split('|', 1)
        localpath, despath = file_path.split(' ', 1)
        file_size = os.stat(localpath).st_size
        file_name = os.path.basename(localpath)
        # 发送文件上传信息
        sent_str = "put|{file_size}|{file_name}|{despath}".format(file_size=file_size, file_name=file_name, despath=despath)
        sent_bytes = bytes(sent_str, 'utf-8')
        sk.sendall(sent_bytes)
        # 接收返回码：300：文件已存在；301：文件不存在
        code = str(sk.recv(1024), 'utf-8')
        has_sent = 0
        if code == '300':
            answer = input("文件已存在，续传Y/N？\n>>>")
            if answer.upper() == 'Y':
                # 续传
                sk.sendall(bytes('302', 'utf-8'))
                exist_size = int(str(sk.recv(1024), 'utf-8'))
                print(file_size, exist_size)
                has_sent += exist_size
            elif answer.upper() == 'N':
                # 重传
                sk.sendall(bytes('303', 'utf-8'))
        elif code == '301':
            sk.sendall(bytes('304', 'utf-8'))
    
        sk.recv(1024)  # send_ack
        # 发送文件
        with open(localpath, 'rb') as file_obj:
            file_obj.seek(has_sent)
            while has_sent < file_size:
                data = file_obj.read(1024)
                sk.sendall(data)
                has_sent += len(data)
        print('上传成功')
    
    
    if __name__ == '__main__':
        while True:
            # cmd|dir
            # put|localpath despath
            inp = input('请输入：')
            if inp.startswith('cmd'):
                handle_cmd(inp, c_sk)
            elif inp.startswith('put'):
                handle_put(inp, c_sk)

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xpbmNvY280OQ_size_16_color_FFFFFF_t_70]: /images/20220205/cccb3156215a4f1694ca9f84b247b717.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xpbmNvY280OQ_size_16_color_FFFFFF_t_70 1]: /images/20220205/9da00ffdab62487b96aaed3a7543deb0.png