首页 > 代码库 > Python开发程序:支持多用户在线的FTP程序

Python开发程序:支持多用户在线的FTP程序

作业:开发一个支持多用户在线的FTP程序

要求:

  1. 用户加密认证
  2. 允许同时多用户登录
  3. 每个用户有自己的家目录 ,且只能访问自己的家目录
  4. 对用户进行磁盘配额,每个用户的可用空间不同
  5. 允许用户在ftp server上随意切换目录
  6. 允许用户查看当前目录下文件
  7. 允许上传和下载文件,保证文件一致性
  8. 文件传输过程中显示进度条
  9. 附加功能:支持文件的断点续传(仅下载)

 

程序

1、最最重要的readme:

技术分享
### 作者介绍:* author:lzl### 博客地址:* http://www.cnblogs.com/lianzhilei/p/5813986.html### 功能实现    作业:开发一个支持多用户在线的FTP程序    要求:        用户加密认证        允许同时多用户登录        每个用户有自己的家目录 ,且只能访问自己的家目录        对用户进行磁盘配额,每个用户的可用空间不同        允许用户在ftp server上随意切换目录        允许用户查看当前目录下文件        允许上传和下载文件,保证文件一致性        文件传输过程中显示进度条        附加功能:支持文件的断点续传### 目录结构:    FTP    │    ├── ftpclient #客户端程序    │      ├── __init__.py    │      └── ftpclient.py  #客户端主程序    └── ftpserver #服务端程序            ├── README.txt            ├── ftpserver.py #服务端入口程序            ├── conf #配置文件目录            │   ├── __init__.py            │   └── setting.py            ├── modules #程序核心目录            │   ├── __init__.py            │   ├── auth_user.py  #用户认证模块            │   └── sokect_server.py  #sokectserver模块            ├── database #用户数据库            │   ├── alex.db            │   ├── lzl.db            │   └── eric.db            ├── home #用户宿主目录            │   ├── alex            │   ├── lzl            │   └── eric            └── log                ├── __init__.py                └── log  #待扩展....### 功能实现    1、conf目录下settings.py模块记录可操作用户信息,根据用户信息生成用户字典和宿主目录,已经生成的不再新建    2、每个用户的宿主目录磁盘空间配额默认为10M,可在settings.py模块里进行修改    3、程序运行在windows系统上,程序要求全部实现,下面是具体命令操作    4、切换目录:cd  ..  返回上一级目录    cd dirname   进入dirname                 用户登录后默认进入宿主目录,只可在宿主目录下随意切换    5、创建目录:mkdir dirname                 在当前目录下创建目录,如果目录存在则报错,不存在创建    6、查看当前路径: pwd    7、查看当前路径下的文件名和目录名: dir    8、下载文件(可续传):get filename                 ①、服务端当前目录存在此文件,客户端不存在此文件,直接下载                 ②、服务端当前目录存在此文件,客户端存在此文件名,之前下载中断,文件可续传,进行续传                 ③、服务端当前目录存在此文件,客户端存在此文件名,大小与服务端一致,不下载    9、上传文件:put  filename                 判断宿主目录磁盘空间是否够用,可以,上传文件;否则,报错### 状态码    400 用户认证失败    401 命令不正确    402 文件不存在    403 创建文件已经存在    404 磁盘空间不够    405 不续传    200 用户认证成功    201 命令可以执行    202 磁盘空间够用    203 文件具有一致性    205 续传    000 系统交互码
Readme

 

2、程序目录结构:

技术分享

 

3、ftp客户端

技术分享
#!/usr/bin/env python# -*- coding:utf-8 -*-#-Author-Lianimport socketimport os,sysimport hashlibclass Myclient():    ‘‘‘ftp客户端‘‘‘    def __init__(self,ip_port):        self.ip_port = ip_port    def connect(self):        ‘‘‘连接服务器‘‘‘        self.client = socket.socket()        self.client.connect(self.ip_port)    def start(self):        ‘‘‘程序开始‘‘‘        self.connect()        while True:            username = input("输入用户名:").strip()            password = input("输入密码:").strip()            login_info = ("%s:%s" %(username, password))            self.client.sendall(login_info.encode())        #发送用户密码信息            status_code = self.client.recv(1024).decode()   #返回状态码            if status_code == "400":                print("[%s]用户密码认证错误"%status_code)                continue            else:print("[%s]用户密码认证成功"%status_code)            self.interactive()    def interactive(self):        ‘‘‘开始交互‘‘‘        while True:            command = input("->>").strip()            if not command:continue            #if command == "exit":break            command_str = command.split()[0]            if hasattr(self,command_str):           # 执行命令                func = getattr(self,command_str)                func(command)            else:print("[%s]命令不存在"%401)    def get(self,command):        ‘‘‘下载文件‘‘‘        self.client.sendall(command.encode())  #发送要执行的命令        status_code = self.client.recv(1024).decode()        if status_code == "201":            #命令可执行            filename = command.split()[1]            # 文件名存在,判断是否续传            if os.path.isfile(filename):                revice_size = os.stat(filename).st_size     #文件已接收大小                self.client.sendall("403".encode())                response = self.client.recv(1024)                self.client.sendall(str(revice_size).encode())   #发送已接收文件大小                status_code = self.client.recv(1024).decode()                # 文件大小不一致,续传                if status_code == "205":                    print("继续上次上传位置进行续传")                    self.client.sendall("000".encode())                # 文件大小一致,不续传,不下载                elif status_code == "405":                    print("文件已经存在,大小一致")                    return            # 文件不存在            else:                self.client.sendall("402".encode())                revice_size = 0            file_size = self.client.recv(1024).decode() #文件大小            file_size = int(file_size)            self.client.sendall("000".encode())            with open(filename,"ab") as file:      #开始接收                #file_size 为文件总大小                file_size +=revice_size                m = hashlib.md5()                while revice_size < file_size:                    minus_size = file_size - revice_size                    if minus_size > 1024:                        size = 1024                    else:                        size = minus_size                    data = self.client.recv(size)                    revice_size += len(data)                    file.write(data)                    m.update(data)                    self.__progress(revice_size,file_size,"下载中")      #进度条                new_file_md5 = m.hexdigest()        #生成新文件的md5值                server_file_md5 = self.client.recv(1024).decode()                if new_file_md5 == server_file_md5:     #md5值一致                    print("\n文件具有一致性")        else:print("[%s] Error!"%(status_code))    def put(self,command):        ‘‘‘上传文件‘‘‘        if len(command.split()) > 1:            filename = command.split()[1]            #file_path = self.current_path + r"\%s"%filename            if os.path.isfile(filename):               #文件是否存在                self.client.sendall(command.encode())  #发送要执行的命令                response = self.client.recv(1024)      #收到ack确认                file_size = os.stat(filename).st_size  # 文件大小                self.client.sendall(str(file_size).encode())  # 发送文件大小                status_code = self.client.recv(1024).decode()  # 等待响应,返回状态码                if status_code == "202":                    with open(filename,"rb") as file:                        m = hashlib.md5()                        for line in file:                            m.update(line)                            send_size = file.tell()                            self.client.sendall(line)                            self.__progress(send_size, file_size, "上传中")  # 进度条                    self.client.sendall(m.hexdigest().encode())     #发送文件md5值                    status_code = self.client.recv(1024).decode()  #返回状态码                    if status_code == "203":                        print("\n文件具有一致性")                else:print("[%s] Error!"%(status_code))            else:                print("[402] Error")        else: print("[401] Error")    def dir(self,command):        ‘‘‘查看当前目录下的文件‘‘‘        self.__universal_method_data(command)        pass    def pwd(self,command):        ‘‘‘查看当前用户路径‘‘‘        self.__universal_method_data(command)        pass    def mkdir(self,command):        ‘‘‘创建目录‘‘‘        self.__universal_method_none(command)        pass    def cd(self,command):        ‘‘‘切换目录‘‘‘        self.__universal_method_none(command)        pass    def __progress(self, trans_size, file_size,mode):        ‘‘‘        显示进度条        trans_size: 已经传输的数据大小        file_size: 文件的总大小        mode: 模式        ‘‘‘        bar_length = 100    #进度条长度        percent = float(trans_size) / float(file_size)        hashes = = * int(percent * bar_length)    #进度条显示的数量长度百分比        spaces =   * (bar_length - len(hashes))    #定义空格的数量=总长度-显示长度        sys.stdout.write(            "\r%s:%.2fM/%.2fM %d%% [%s]"%(mode,trans_size/1048576,file_size/1048576,percent*100,hashes+spaces))        sys.stdout.flush()    def __universal_method_none(self,command):        ‘‘‘通用方法,无data输出‘‘‘        self.client.sendall(command.encode())  # 发送要执行的命令        status_code = self.client.recv(1024).decode()        if status_code == "201":  # 命令可执行            self.client.sendall("000".encode())  # 系统交互        else:            print("[%s] Error!" % (status_code))    def __universal_method_data(self,command):        ‘‘‘通用方法,有data输出‘‘‘        self.client.sendall(command.encode())   #发送要执行的命令        status_code = self.client.recv(1024).decode()        if status_code == "201":    #命令可执行            self.client.sendall("000".encode())      #系统交互            data = http://www.mamicode.com/self.client.recv(1024).decode()            print(data)        else:print("[%s] Error!" % (status_code))if __name__ == "__main__":    ip_port =("127.0.0.1",9999)         #服务端ip、端口    client = Myclient(ip_port)            #创建客户端实例    client.start()                      #开始连接
ftpclient.py

 

4、ftp服务端

技术分享
#!/usr/bin/env python# -*- coding:utf-8 -*-#-Author-Lianimport os,hashlibimport jsonfrom conf import settingsfrom modules import auth_userfrom modules import sokect_serverdef create_db():    ‘‘‘创建用户数据库文件‘‘‘    user_database={}    encryption = auth_user.User_operation()    limitsize = settings.LIMIT_SIZE    for k,v in settings.USERS_PWD.items():        username = k        password = encryption.hash(v)        user_db_path  = settings.DATABASE + r"\%s.db"%username        user_home_path = settings.HOME_PATH + r"\%s"%username        user_database["username"] = username        user_database["password"] = password        user_database["limitsize"] = limitsize        user_database["homepath"] = user_home_path        if not os.path.isfile(user_db_path):            with open(user_db_path,"w") as file:                file.write(json.dumps(user_database))def create_dir():    ‘‘‘创建用户属主目录‘‘‘    for username in settings.USERS_PWD:        user_home_path = settings.HOME_PATH + r"\%s" %username        if not os.path.isdir(user_home_path):            os.popen("mkdir %s" %user_home_path)if __name__ == "__main__":    ‘‘‘初始化系统数据并启动程序‘‘‘    create_db()         #创建数据库    create_dir()        #创建属主目录                        #启动ftp服务    server = sokect_server.socketserver.ThreadingTCPServer(settings.IP_PORT, sokect_server.Myserver)    server.serve_forever()
ftpserver.py

 

5、conf配置文件

技术分享
#!/usr/bin/env python# -*- coding:utf-8 -*-#-Author-Lianimport os,sys#程序主目录文件BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))#添加环境变量sys.path.insert(0,BASE_DIR)#数据库目录DATABASE = os.path.join(BASE_DIR,"database")#用户属主目录HOME_PATH = os.path.join(BASE_DIR,"home")#用户字典USERS_PWD = {"alex":"123456","lzl":"8888","eric":"6666"}#磁盘配额   10MLIMIT_SIZE = 10240000#ftp服务端口IP_PORT = ("0.0.0.0",9999)
settings.py

 

6、database用户数据库(系统初始化自动生成)

技术分享
{"username": "alex", "password": "e10adc3949ba59abbe56e057f20f883e", "limitsize": 10240000, "homepath": "C:\\Users\\L\\PycharmProjects\\s14\\homework\\Day8\\FTP\\ftpserver\\home\\alex"}
lzl.db
技术分享
{"username": "eric", "password": "e9510081ac30ffa83f10b68cde1cac07", "limitsize": 10240000, "homepath": "C:\\Users\\L\\PycharmProjects\\s14\\homework\\Day8\\FTP\\ftpserver\\home\\eric"}
eric.db
技术分享
{"username": "alex", "password": "e10adc3949ba59abbe56e057f20f883e", "limitsize": 10240000, "homepath": "C:\\Users\\L\\PycharmProjects\\s14\\homework\\Day8\\FTP\\ftpserver\\home\\alex"}
alex.db

 

7、modules模块dd

技术分享
#!/usr/bin/env python# -*- coding:utf-8 -*-#-Author-Lianimport jsonimport sys,osimport hashlibfrom conf import settingsclass User_operation():    ‘‘‘对登录信息进行认证,登录成功返回用户名,失败返回None‘‘‘    def authentication(self,login_info):        list = login_info.split(":")            #对信息进行分割        login_name = list[0]        login_passwd = self.hash(list[1])        DB_FILE = settings.DATABASE + r"\%s.db"%login_name        if os.path.isfile(DB_FILE):            user_database = self.cat_database(DB_FILE)  #用户数据库信息            if login_name == user_database["username"]:                if login_passwd == user_database["password"]:                    return user_database    def cat_database(self,DB_FILE):        #获取数据库信息        with open(DB_FILE,"r") as file:            data = json.loads(file.read())            return  data    def hash(self,passwd):        ‘‘‘对密码进行md5加密‘‘‘        m = hashlib.md5()        m.update(passwd.encode("utf-8"))        return m.hexdigest()
auth_user.py
技术分享
#!/usr/bin/env python# -*- coding:utf-8 -*-#-Author-Lianimport socketserverimport sys,osimport hashlibfrom os.path import join, getsizefrom conf import settingsfrom modules import auth_userclass Myserver(socketserver.BaseRequestHandler):    ‘‘‘ftp服务端‘‘‘    def handle(self):        try:            self.conn = self.request            while True:                login_info = self.conn.recv(1024).decode()    # 接收客户端发的的账号密码信息                result = self.authenticat(login_info)                status_code = result[0]                self.conn.sendall(status_code.encode())                if status_code == "400":                    continue                self.user_db = result[1]             #当前登录用户信息                self.current_path = self.user_db["homepath"]    #用户当前目录                self.home_path = self.user_db["homepath"]  #用户宿主目录                while True:                    command = self.conn.recv(1024).decode()                    command_str = command.split()[0]                    if hasattr(self,command_str):                        func = getattr(self,command_str)                        func(command)                    else:self.conn.sendall("401".encode())        except ConnectionResetError as e:            self.conn.close()            print(e)    def authenticat(self,login_info):        ‘‘‘认证用户‘‘‘        auth = auth_user.User_operation()       # 创建认证实例        result = auth.authentication(login_info)         # 认证用户        if result:return "200",result        else:return "400",result    def get(self,command):        ‘‘‘下载文件‘‘‘        if len(command.split()) > 1:            filename = command.split()[1]            file_path = self.current_path + r"\%s"%filename            if os.path.isfile(file_path):               #文件是否存在                self.conn.sendall("201".encode())      #命令可执行                file_size = os.stat(file_path).st_size  # 文件总大小                status_code = self.conn.recv(1024).decode()                # 客户端存在此文件                if status_code == "403":                    self.conn.sendall("000".encode())   #系统交互                    has_send_size = self.conn.recv(1024).decode()                    has_send_size = int(has_send_size)                    # 客户端文件不完整可续传                    if has_send_size < file_size:                        self.conn.sendall("205".encode())                        file_size -= has_send_size  #续传文件大小                        response = self.conn.recv(1024)  # 等待响应                    # 客户端文件完整不可续传、不提供下载                    else:                        self.conn.sendall("405".encode())                        return                # 客户端不存在此文件                elif status_code == "402":                    has_send_size = 0                with open(file_path,"rb") as file:                    self.conn.sendall(str(file_size).encode())       #发送文件大小                    response = self.conn.recv(1024)     #等待响应                    file.seek(has_send_size)                    m = hashlib.md5()                    for line in file:                        m.update(line)                        self.conn.sendall(line)                self.conn.sendall(m.hexdigest().encode())     #发送文件md5值            else:self.conn.sendall("402".encode())        else:self.conn.sendall("401".encode())    def put(self,command):        ‘‘‘上传文件‘‘‘        filename = command.split()[1]        file_path = self.current_path + r"\%s" % filename        self.conn.sendall("000".encode())   #发送确认        file_size = self.conn.recv(1024).decode()  # 文件大小        file_size = int(file_size)        limit_size = self.user_db["limitsize"]      #磁盘额度        used_size = self.__getdirsize(self.home_path)   #已用空间大小        if limit_size >= file_size+used_size:            self.conn.sendall("202".encode())            with open(file_path, "wb") as file:  # 开始接收                revice_size = 0                m = hashlib.md5()                while revice_size < file_size:                    minus_size = file_size - revice_size                    if minus_size > 1024:                        size = 1024                    else:                        size = minus_size                    data = self.conn.recv(size)                    revice_size += len(data)                    file.write(data)                    m.update(data)                new_file_md5 = m.hexdigest()  # 生成新文件的md5值                server_file_md5 = self.conn.recv(1024).decode()                if new_file_md5 == server_file_md5:  # md5值一致                    self.conn.sendall("203".encode())        else:self.conn.sendall("404".encode())    def dir(self,command):        ‘‘‘查看当前目录下的文件‘‘‘        if len(command.split()) == 1:            self.conn.sendall("201".encode())            response = self.conn.recv(1024)            send_data = os.popen("dir %s"%self.current_path)            self.conn.sendall(send_data.read().encode())        else:self.conn.sendall("401".encode())    def pwd(self,command):        ‘‘‘查看当前用户路径‘‘‘        if len(command.split()) == 1:            self.conn.sendall("201".encode())            response = self.conn.recv(1024)            send_data = self.current_path            self.conn.sendall(send_data.encode())        else:self.conn.sendall("401".encode())    def mkdir(self,command):        ‘‘‘创建目录‘‘‘        if len(command.split()) > 1:            dir_name = command.split()[1]       #目录名            dir_path = self.current_path + r"\%s"%dir_name #目录路径            if not os.path.isdir(dir_path):     #目录不存在                self.conn.sendall("201".encode())                response = self.conn.recv(1024)                os.popen("mkdir %s"%dir_path)            else:self.conn.sendall("403".encode())        else:self.conn.sendall("401".encode())    def cd(self,command):        ‘‘‘切换目录‘‘‘        if len(command.split()) > 1:            dir_name = command.split()[1]       #目录名            dir_path = self.current_path + r"\%s" %dir_name    #目录路径            user_home_path = settings.HOME_PATH + r"\%s"%self.user_db["username"]   #宿主目录            if dir_name == ".." and len(self.current_path)>len(user_home_path):                self.conn.sendall("201".encode())                response = self.conn.recv(1024)                self.current_path = os.path.dirname(self.current_path)  #返回上一级目录            elif  os.path.isdir(dir_path)   :                self.conn.sendall("201".encode())                response = self.conn.recv(1024)                if dir_name != "." and dir_name != "..":                    self.current_path += r"\%s"%dir_name        #切换目录            else:self.conn.sendall("402".encode())        else:self.conn.sendall("401".encode())    def __getdirsize(self,home_path):        ‘‘‘统计目录空间大小‘‘‘        size = 0        for root, dirs, files in os.walk(home_path):            size += sum([getsize(join(root, name)) for name in files])        return size
sokect_server.py

 

效果图:

下载:

技术分享

续传:

技术分享

 

Python开发程序:支持多用户在线的FTP程序