Python实现支持并发、断点续传的FTP

参考网上一个FTP程序，重写了一遍，并稍加扩展

一、要求

1. 支持多用户同时登录

2. 可以注册用户，密码使用md5加密

3. 可以登录已注册用户

4.  支持cd切换目录，ls查看目录子文件

5. 支持上传、下载文件

6. 可以执行命令，如：ipconfig

二、代码

配置文件

#!/usr/bin/env python
# -*- coding:utf-8 -*-

import os

BASE_DIR = os.path.dirname(os.path.dirname(__file__))
BASE_HOME = os.path.join(BASE_DIR, 'home')
NAME_PWD = os.path.join(BASE_DIR, 'db', 'name_pwd')
USER_FILE = os.path.join(BASE_DIR, 'db')

服务端：

#!/usr/bin/env python
# -*- coding:utf-8 -*-

import os
import hashlib
import pickle
import subprocess
import socketserver
from config import settings

baseHome = settings.BASE_HOME
class MyServer(socketserver.BaseRequestHandler):
    def recv_file(self):
        '''
        文件传输
        :return:
        '''
        conn = self.request
        a = str(conn.recv(1024),encoding='utf-8')
        file_size, file_name = a.split(',')
        new_file_name = os.path.join(baseHome, file_name)
        if file_name in baseHome:    #检测文件是否已存在，涉及断点续传
            has_recv = os.stat(baseHome).st_size     #计算临时文件大小
            conn.sendall(bytes(has_recv, encoding='utf-8'))
            with open(new_file_name,'ab') as f:     #追加模式
                data = conn.recv(1024)
                f.write(data)
                has_recv += len(data)
        else:
            has_recv = 0
            conn.sendall(bytes('s',encoding='utf-8'))   #客户端收到字符串s，从0开始发送
            with open(new_file_name, 'wb') as f:
                while has_recv<= int(file_size):
                    data = conn.recv(1024)
                    f.write(data)
                    has_recv += len(data)

    def send_file(self, fileName):
        '''
        向客户端发送文件
        :param fileName:
        :return:
        '''
        filePath = os.path.join(self.currDir, fileName)
        conn = self.request
        if os.path.exists(filePath):
            size = os.stat(filePath).st_size
            conn.sendall(bytes(str(size)+','+fileName,encoding='utf-8'))
            ret = conn.recv(1024)
            r = str(ret,encoding='utf-8')
            if r=='s':
                has_send = 0
            else:
                has_send = int(r)
            with open(filePath,'rb') as f:
                f.seek(has_send)
                while has_send<size:
                    data = f.read(1024)
                    conn.sendall(data)
                    has_send+=len(data)
            conn.sendall(bytes('0', encoding='utf-8'))
        else:
            conn.sendall(bytes('0', encoding='utf-8'))

    def createDir(self, currDir, newName):
        '''
        创建文件夹
        :param currDir:当前所在目录
        :param newName: 新文件夹名称
        :return: 是否创建成功
        '''
        mulu = os.path.join(baseHome, currDir)
        newFilePath = os.path.join(mulu, newName)
        if os.path.exists(newFilePath):
            return '2'
        else:
            ret = '0'
            try:
                os.makedirs(newFilePath)
                ret = '1'
            except OSError as e:
                ret = '0'
            return ret

    def command(self):
        '''
        执行命令
        :return:
        '''
        conn = self.request
        a = conn.recv(1024)
        ret = str(a, encoding='utf-8')
        ret2 = subprocess.check_output(ret, shell=True)
        r = divmod(len(ret2), 1024)
        s = r[0]+1
        conn.sendall(bytes(str(s), encoding='utf-8'))
        conn.recv(1024)
        conn.sendall(ret2)

    def md5(self, pwd):
        '''
        判断密码进行加密
        :param pwd:
        :return:
        '''
        hash = hashlib.md5(bytes('xx7',encoding='utf-8'))
        hash.update(bytes(pwd, encoding='utf-8'))
        return hash.hexdigest()

    def login(self, username, pwd):
        '''
        登录
        :param username:用户名
        :param pwd: 密码
        :return: 是否登录成功
        '''
        if os.path.exists(settings.NAME_PWD):
            s = pickle.load(open(settings.NAME_PWD,'rb'))
        if username in s:
            if s[username]==self.md5(pwd):
                return True
            else:
                return False
        else:
            return False

    def regist(self, username, pwd):
        '''
        注册
        :param username:用户名
        :param pwd: 密码
        :return: 是否注册成功
        '''
        conn = self.request
        s = {}
        if os.path.exists(settings.NAME_PWD):
            s = pickle.load(open(settings.NAME_PWD, 'rb'))
        if username in s:
            return False
        else:
            s[username] = self.md5(pwd)
            mulu = os.path.join(settings.USER_FILE, username)
            os.makedirs(mulu)
            pickle.dump(s, open(settings.NAME_PWD, 'wb'))
            return True

    def before(self, username, pwd, ret):
        '''
        判断注册和登录，并展示用户的详细目录信息，支持cd和ls命令
        :param username: 用户名
        :param pwd: 密码
        :param ret:
        :return:
        '''
        conn = self.request
        if ret == '1':
            r = self.login(username,pwd)
            if r:
                conn.sendall(bytes('y',encoding='utf-8'))
            else:
                conn.sendall(bytes('n',encoding='utf-8'))
        elif ret == '2':
            r = self.regist(username, pwd)
            if r:
                conn.sendall(bytes('y',encoding='utf-8'))
            else:
                conn.sendall(bytes('n',encoding='utf-8'))

    def user_file(self, username):
        '''
        展示用户的详细目录信息，支持cd和ls命令
        :param username: 用户名
        :return:
        '''
        conn = self.request
        mulu = baseHome
        self.currDir = mulu
        conn.sendall(bytes(mulu, encoding='utf-8'))
        while True:
            if conn:
                b = conn.recv(1024)
                ret = str(b, encoding='utf-8')
                try:
                    a, b = ret.split(' ',1)
                except Exception as e:
                    a = ret
                if a == 'cd':
                    if b=='..':
                        mulu = os.path.dirname(mulu)
                    else:
                        mulu = os.path.join(mulu, b)
                    self.currDir = mulu
                    conn.sendall(bytes(mulu, encoding='utf-8'))
                elif a=='ls':
                    ls = os.listdir(mulu)
                    print(ls)
                    a = ','.join(ls)
                    if a=='':
                        a = '.'
                    conn.sendall(bytes(a, encoding='utf-8'))
                elif a=='mkdir':
                    m = self.createDir(self.currDir,b)
                    conn.sendall(bytes(m, encoding='utf-8'))
                elif a=='q':
                    break

    def handle(self):
        conn = self.request
        conn.sendall(bytes('welcome',encoding='utf-8'))
        b = conn.recv(1024)
        ret = str(b, encoding='utf-8')
        c = conn.recv(1024)
        r = str(c, encoding='utf-8')
        username, pwd = r.split(',')
        self.before(username, pwd, ret)
        self.user_file(username)
        while True:
            a=conn.recv(1024)
            ret = str(a, encoding='utf-8')
            if ret == '1':
                self.recv_file()
            elif ret=='2':
                self.command()
            elif ret[0:4]=='get:':
                self.send_file(ret[4:])
            elif ret=='q':
                break
            else:
                pass

if __name__ == '__main__':
    server = socketserver.ThreadingTCPServer(('',9999), MyServer)
    server.serve_forever()

客户端：

#!/usr/bin/env python
# -*-coding:utf-8 -*-

import os, sys
import socket

def send_file(file_path):
    '''
    发送文件
    :param file_path:文件名
    :return:
    '''
    size = os.stat(file_path).st_size
    file_name = os.path.basename(file_path)
    obj.sendall(bytes(str(size)+','+file_name,encoding='utf-8'))
    ret = obj.recv(1024)
    r = str(ret, encoding='utf-8')
    if r=='s':  #文件不存在，从头开始传
        has_send = 0
    else:   #文件存在
        has_send = int(r)
    with open(file_path, 'rb') as f:
        f.seek(has_send)    #定位到已经传到的位置
        while has_send<size:
            data = f.read(1024)
            obj.sendall(data)
            has_send+=len(data)
            sys.stdout.write('
')  #情况文件内容
            sys.stdout.write('已发送%s%%|%s' % (int(has_send/size*100), (round(has_send/size*40)*'|')))
            sys.stdout.flush()  #强制刷出内存
    print('上传成功！
')

def recv_file(toPath, getFile):
    '''
    接收要下载的文件
    :param toPath: 本地要保存文件的存放路径
    :param getFile: 要下载的文件名称
    :return:
    '''
    obj.sendall(bytes('get:'+getFile,encoding='utf-8'))
    a = str(obj.recv(1024), encoding='utf-8')
    file_size, file_name = a.split(',')
    file_size = int(file_size)
    if file_size == 0:
        print('没有找到此文件')
    else:
        new_file_name = os.path.join(toPath, file_name)
        if file_name in toPath:
            has_recv = os.stat(toPath).st_size
            obj.sendall(bytes(has_recv, encoding='utf-8'))
            with open(new_file_name,'ab') as f:
                while has_recv<=file_size:
                    data = obj.recv(1024)
                    f.write(data)
                    has_recv+=len(data)
                    sys.stdout.write('
')  # 情况文件内容
                    sys.stdout.write('已接收%s%%|%s' % (int(has_recv / file_size * 100), (round(has_recv / file_size * 40) * '|')))
                    sys.stdout.flush()  # 强制刷出内存
        else:
            has_recv = 0
            obj.sendall(bytes('s', encoding='utf-8'))
            with open(new_file_name, 'wb') as f:
                while has_recv<= file_size:
                    data = obj.recv(1024)
                    f.write(data)
                    has_recv += len(data)
                    sys.stdout.write('
')  # 情况文件内容
                    sys.stdout.write('已接收%s%%|%s' % (int(has_recv / file_size * 100), (round(has_recv / file_size * 40) * '|')))
                    sys.stdout.flush()  # 强制刷出内存
        print('接收成功！
')


def command(command_name):
    '''
    执行命令
    :param command_name:
    :return:
    '''
    obj.sendall(bytes(command_name, encoding='utf-8'))
    ret = obj.recv(1024)    #接受命令需要接受的次数
    obj.sendall(bytes('收到次数',encoding='utf-8'))
    r = str(ret, encoding='utf-8')
    for i in range(int(r)):     #共需接收int(r)次
        ret = obj.recv(1024)    #等待客户端发送
        r = str(ret, encoding='GBK')
        print(r)

def login(username, pwd):
    '''
    登录
    :param username: 用户名
    :param pwd: 密码
    :return: 是否登录成功
    '''
    obj.sendall(bytes(username+','+pwd, encoding='utf-8'))
    ret = obj.recv(1024)
    r = str(ret, encoding='utf-8')
    if r=='y':
        return True
    else:
        return False

def regist(username, pwd):
    '''
    注册
    :param username: 用户名
    :param pwd: 密码
    :return: 是否注册成功
    '''
    obj.sendall(bytes(username+','+pwd, encoding='utf-8'))
    ret = obj.recv(1024)
    r = str(ret, encoding='utf-8')
    if r=='y':
        return True
    else:
        return False

def before(username, pwd):
    '''
    选择注册和登录，并展示用户的详细目录信息，支持cd和ls命令
    :param username: 用户名
    :param pwd: 密码
    :return:
    '''
    a = input('请选择 1.登录  2.注册：')
    obj.sendall(bytes(a, encoding='utf-8'))
    # obj.recv()
    if a=='1':
        ret = login(username, pwd)
        if ret:
            print('登录成功')
            return 1
        else:
            print('用户名或密码错误')
            return 0
    elif a=='2':
        ret = regist(username, pwd)
        if ret:
            print('注册成功')
            return 1
        else:
            print('用户名已存在')
            return 0

def user_file(username):
    # obj.sendall(bytes('打印用户文件路径', encoding='utf-8'))
    ret = obj.recv(1024)
    r = str(ret, encoding='utf-8')
    print(r)
    while True:
        a = input('输入 cd切换目录，ls查看目录详细信息，mkdir创建文件夹，q退出：')
        a = a.strip()
        obj.sendall(bytes(a, encoding='utf-8'))
        if a=='q':
            break
        elif a[0:5]=='mkdir':
            ret = obj.recv(1024)
            r = str(ret, encoding='utf-8')
            if r=='1':
                print('文件夹创建成功')
            elif r=='2':
                print('文件夹已存在！')
            else:
                print('创建失败！')
        else:
            ret = obj.recv(1024)
            r=str(ret, encoding='utf-8')
            if len(r)==1:   #判断是cd结果，还是ls的结果（ls只有一个子目录，直接打印）
                print(r)
            else:
                li = r.split(',')
                for i in li:
                    print(i)

def main(username, pwd):
    ret = obj.recv(1024)
    r = str(ret, encoding='utf-8')
    print(r)
    result = before(username, pwd)  #判断登录/注册
    if result:
        user_file(username)
        while True:
            a = input('请选择 1.传文件 2.执行命令 3.收文件 q 退出：')
            obj.sendall(bytes(str(a),encoding='utf-8'))
            if a=='1':
                b = input('请输入文件路径：')
                if os.path.exists(b):
                    send_file(b)
                    obj.sendall(bytes('hhe', encoding='utf-8'))
            elif a=='2':
                b = input('请输入command：')
                command(b)
            elif a=='3':
                b = input('请输入存放路径：')
                c = input('请输入要获取的文件：')
                recv_file(b, c)
            elif a=='q':
                break
            else:
                print('输入错误！')

    obj.close()

if __name__ == '__main__':
    obj = socket.socket()
    obj.connect(('192.168.1.100',9999))
    username = input('请输入用户名：')
    pwd = input('请输入密码：')
    main(username, pwd)