一、作业需求
1. 用户加密认证
2. 多用户同时登陆
3. 每个用户有自己的家目录且只能访问自己的家目录
4. 对用户进行磁盘配额、不同用户配额可不同
5. 用户可以登陆server后,可切换目录
6. 查看当前目录下文件
7. 上传下载文件,保证文件一致性
8. 传输过程中现实进度条
9.支持断点续传
二、实现功能
1、多用户同时登录注册(已有用户:japhi、alex;密码都是123)
2、上传/下载文件(已有示例文件)
3、查看不同用户自己家得目录下文件,且只能访问自己的家目录
4、对用户进行磁盘配额,不同用户配额不同(使用random函数,随机给用户一个内存大小(10m-20m))
5、用户登录后,可对用户目录下文件目录进行操作,包含:ls(查看当前操作目录下文件)、cd(切换当前操作目录)、rm(删除文件)、mkdir(创建目录)
6、上传下载文件,保证文件一致性,且在传输过程中实现进度条
7、支持断点续传(*******暂未实现*******)
三、目录说明
FTP/
|-- FTPClient/ #客户端文件夹
| |-- 示例文件夹/ #客户端上传/下载示例文件夹
| |-- Client_start.py #客户端启动程序
|
|-- FTPServer/ #服务端文件夹
| |-- bin/
| | |-- __init__.py
| | |-- Server_start.py #程序启动的主入口
| |
| |-- conf/
| | |-- setting.py #配置文件
| |
| |-- db/ #用户数据
| | |-- alex #用户名alex的数据文件夹
| | |-- japhi #用户名japhi的数据文件夹
| |
| |-- home/
| | |-- alex/ #用户alex用户家目录
| | |-- japhi/ #用户japhi用户家目录
| |
|-- |-- log/
| |-- log_sys.log #日志文件(未启用)
|
|-- |-- src/
| | |-- __init__.py
| | |-- common.py #公共功能
| | |-- Server_start.py #程序启动的主入口
| | |-- user.py #用户类及方法
|-- db/ #用户数据
| | |-- alex #用户名alex的数据文件夹
| | |-- japhi #用户名japhi的数据文件夹
|-- FTP.png #流程图
|-- README.txt
四、流程图
五、代码说明
1、FTPClient/Client_start.py
from __future__ import division import socket,os,sys,time,hashlib,math updir = os.path.join(os.path.dirname(os.path.abspath(__file__)),"示例文件夹") HOST = "localhost" PORT = 9998 def upload(client,user_info,name): ''' 客户端上传文件的函数 :param client:scoket客户端标志 :param user_info:客户端登陆用户的信息 :param name:客户端登陆用户的名字 :return:none ''' print(" 33[1;37m当前可选上传 33[0m".center(40,"*")) dic = {} for root, dirs, files in os.walk(updir): for i,j in enumerate(files): k = i+1 dic[k] = j print(" 33[1;37m%s:%s 33[0m"%(k,j)) choice = input("请输入要上传的文件序号:>>>").strip() if choice.isdigit() and 0 < int(choice) <= len(dic): command = "upload+"+user_info+"+"+dic[int(choice)] client.sendall(bytes(command,encoding="utf-8")) res = client.recv(1024) if str(res,encoding="utf-8") == "True": dir = os.path.join(updir,dic[int(choice)]) f = open(dir,"rb") md5 = hashlib.md5() length = os.stat(dir).st_size client.send(str(length).encode()) sign = client.recv(1024).decode() if sign == "ok": data = f.read() md5.update(data) client.sendall(data) f.close() client.send(md5.hexdigest().encode()) res_sign = client.recv(1024) if res_sign == b'True': print(" 33[1;37m文件上传成功 33[0m") elif res_sign == b'False': print(" 33[1;37m文件上传失败 33[0m") exit() elif sign == "no": print(" 33[1;37m磁盘空间不足 33[0m") exit() else: print(" 33[1;37m输入有误 33[0m") def download(client,user_info,name): ''' 客户端下载文件的函数 :param client: scoket客户端标志 :param user_info: 客户端登陆的用户信息 :param name:客户端登陆的用户名字 :return: none ''' dic = {} command = "download+"+user_info client.sendall(bytes(command, encoding="utf-8")) data = client.recv(4069) res = eval(str(data, encoding="utf-8")) if len(res) == 0: print(" 33[1;31m当前目录下暂无文件 33[0m".center(40, "-")) else: for i,j in enumerate(res): k = i + 1 dic[k] = j print(" 33[1;37m%s:%s 33[0m" % (k, j)) choice = input("请选择要下载的文件序号:>>>") cm = dic[int(choice)] client.sendall(bytes(cm, encoding="utf-8")) print(" 33[1;37m准备开始下载文件 33[0m") dir = os.path.join(updir, dic[int(choice)]) res = str(client.recv(1024).decode()).split("+") res_length = res[0] or_md5 = res[1] # print(or_md5) length = 0 f = open(dir, "wb") m = hashlib.md5() while length < int(res_length): if int(res_length) - length > 1024: # 要收不止一次 size = 1024 else: # 最后一次了,剩多少收多少 size = int(res_length) - length # print("最后一次剩余的:", size) data = client.recv(size) length += len(data) m.update(data) f.write(data) progressbar(length, int(res_length)) else: new_md5 = m.hexdigest() # print(new_md5) f.close() if new_md5 == or_md5: print(" 33[1;37m文件下载成功 33[0m") return True else: print(" 33[1;37m文件下载失败 33[0m") return False def switch(client,user_info,name): ''' 切换目录操作函数,包括“ls”,“cd”,“rm”,“mkdir” :param client: 客户端 :param user_info: 用户信息 :param name: 用户名 :return: none ''' command = ''' ls cd rm mkdir 目录名 ''' # while True: print(" 33[1;33m%s 33[0m" % command) c = input("请输入您要操作的命令:>>>").strip() if c == "ls": view_file(client, user_info, name) elif c == "cd": cm = "cd+" + user_info client.sendall(cm.encode("utf-8")) dirs = eval(client.recv(1024).decode()) if len(dirs) != 0: for j in dirs: print(" 33[1;37m目录:%s 33[0m" % (j)) choice = input("请输入“cd”的目录名称:>>").strip() if choice in dirs: client.sendall(choice.encode("utf-8")) sign = client.recv(1024).decode() # print(len(res[0]),len(res[1])) if sign == "True": print(" 33[1;31m%s目录切换成功 33[0m" % (choice)) else: print(" 33[1;31m%s目录切换失败 33[0m" % (choice)) else: print(" 33[1;31m输入有误 33[0m") client.sendall("error".encode("utf-8")) exit() else: print(" 33[1;31m无其它目录 33[0m") elif c.split(" ")[0] == "mkdir": cm = "mkdir+" + user_info + "+" + c.split(" ")[1] # print(cm) client.sendall(cm.encode("utf-8")) res = client.recv(1024).decode() if res == "True": print(" 33[1;37m目录创建成功 33[0m") else: print(" 33[1;37m目录创建失败 33[0m") elif c == "rm": cm = "rm+" + user_info # print(cm) client.sendall(cm.encode("utf-8")) res_list = eval(client.recv(1024)) if len(res_list) == 0: print(" 33[1;37m无其它文件 33[0m") client.sendall("error".encode("utf-8")) else: for i in res_list: print(" 33[1;37m文件:%s 33[0m" % i) choice = input("请输入“rm”的文件名称:>>").strip() if choice in res_list: client.sendall(choice.encode("utf-8")) if client.recv(1024).decode() == "True": print(" 33[1;37m文件删除成功 33[0m") else: print(" 33[1;37m文件删除失败 33[0m") else: print(" 33[1;37m输入有误 33[0m") client.sendall("error".encode("utf-8")) exit() else: print(" 33[1;37m输入有误 33[0m") exit() def view_file(client,user_info,name): ''' 客户端查看当前目录下文件的函数 :param client: scoket客户端标志 :param user_info: 客户端登陆的用户信息 :param name: 客户端登陆的用户名字 :return: none ''' command = "view+"+user_info client.sendall(bytes(command,encoding="utf-8")) dirs = client.recv(1024) # print(dirs,"1111111111111111") if dirs.decode() == "False": dir = [] else: dir = eval(str(dirs,encoding="utf-8")) files = client.recv(1024) file = eval(str(files, encoding="utf-8")) client.sendall("true".encode("utf-8")) storage = str(client.recv(1024).decode()) # print(storage) if len(file) == 0 and len(dir) == 0: print(" 33[1;31m当前目录下暂无文件 33[0m".center(40, "-")) else: print(" 33[1;33m当前目录包含以下文件内容 33[0m".center(30,"*")) print(" 33[1;35m磁盘大小:%skb 33[0m" % storage) for j in dir: print(" 33[1;35m目录:%s 33[0m"%j) for i in file: print(" 33[1;35m文件:%s 33[0m"%i) print("".center(33,"*")) def operate(client,user_info,name): ''' 客户端操作主函数 :param client: scoket客户端标志 :param user_info: 客户端登陆的用户信息 :param name: 客户端登陆的用户名字 :return: none ''' dic = {"1":upload,"2":download,"4":view_file,"3":switch} info = '''------操作指令------ 1、上传文件 2、下载文件 3、切换目录操作 4、查看目录下文件 5、退出 ''' while True: print(" 33[1;33m%s 33[0m" % info) choice = input("请输入你要操作的命令:>>>").strip() if choice.isdigit() and 0 < int(choice) <= len(dic): dic.get(choice)(client,user_info,name) elif choice.isdigit() and int(choice) == 5: break else: print(" 33[1;31m输出错误 33[0m".center(40, "-")) def com_parse(client,com): ''' 客户端用户登陆注册命中解析函数 :param client: 客户端scoket标志 :param com: 命令 :return: 登陆成功返回True,否则False ''' # print(com) client.sendall(bytes(com,encoding="utf-8")) re = client.recv(4096) if str(re,encoding="utf-8") == "Success": return True elif str(re, encoding="utf-8") == "Success": return False def login(client,data): ''' 客户端用户登陆函数 :param client: 客户端scoket标志 :param data: 数据 :return: none ''' name = input("请输入您的名字:>>>").strip() psd = input("请输入密码:>>>").strip() user_info = name+"+"+psd com = "login+"+user_info # com_parse(client, com) if com_parse(client,com): print(" 33[1;31m登陆成功 33[0m") operate(client,user_info,name) else: print(" 33[1;31m登陆出现异常 33[0m") def register(client,data): ''' 客户端用户注册函数 :param client: 客户端scoket标志 :param data: 数据 :return: none ''' name = input("请输入您的名字:>>>").strip() psd = input("请输入密码:>>>").strip() com = "register+" + name + "+" + psd if com_parse(client,com): print(" 33[1;31m注册成功 33[0m") user_info = name + "+" + psd operate(client, user_info, name) else: print(" 33[1;31m注册出现异常 33[0m") def quit(client,data): ''' 程序退出函数 :param client: 客户端scoket标志 :param data: 用户数据 :return: none ''' exit() def main_func(client,data): ''' 客户端主菜单函数 :param client: 客户端scoket标志 :param data: 数据 :return: none ''' dic = {"1":login,"2":register,"3":quit} info = '''------用户登录界面------*{0}* 1、登陆 2、注册 3、退出 '''.format(data) print(" 33[1;33m%s 33[0m"%info) what = input("你要干嘛?>>>").strip() if what.isdigit() and 0 < int(what) <= len(dic): dic.get(what)(client,data) else: print(" 33[1;31m输出错误 33[0m".center(40,"-")) def progressbar(cur, total): ''' 进度条处理函数 :param cur: 当前文件内容长度 :param total: 总长度 :return: none ''' percent = '{:.2%}'.format(cur / total) sys.stdout.write(' ') sys.stdout.write('[%-50s] %s' % ('=' * int(math.floor(cur * 50 / total)), percent)) # print('[%-50s] %s ' % ('=' * int(math.floor(cur * 50 / total)), percent),) sys.stdout.flush() time.sleep(0.01) if cur == total: sys.stdout.write(' ') if __name__ == '__main__': client = socket.socket(socket.AF_INET,socket.SOCK_STREAM) client.connect(("localhost",PORT)) client.send("True".encode("utf-8")) # print("True 发送成功") # main_func(client,client.recv(1024)) main_func(client,"connect") client.close()
2、FTPServer/conf/settings.py
import os basedir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) user_home = "%s/FTPServer/home"%basedir user_info = "%s/db"%basedir # print(user_home) # print(user_info) HOST = "0.0.0.0" PORT = 9998
3、FTPServer/src/common.py
from __future__ import division import logging,os,pickle,sys,uuid,math,time frame = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(frame) # from conf import setting # # def sys_logging(content,levelname): # ''' # 程序记录日志函数 # :param content: 日志的内容 # :param levelname: 日志的等级 # :return: none # ''' # _filename = os.path.join(setting.log_dir,"log_sys.log") # log = logging.getLogger(_filename) # logging.basicConfig(filename=_filename,level=logging.INFO,format='%(asctime)s-%(levelname)s-%(message)s', datefmt='%m/%d/%Y %I:%M:%S %p') # if levelname == 'debug': # logging.debug(content) # elif levelname == 'info': # logging.info(content) # elif levelname == 'warning': # logging.warning(content) # elif levelname == 'error': # logging.error(content) # elif levelname == 'critical': # logging.critical(content) def show(msg,msg_type): ''' 程序不同信息打印的字体颜色 :param msg: 打印信息 :param msg_type: 打印信息的类型 :return: none ''' if msg_type == "info": show_msg = " 33[1;35m%s 33[0m"%msg elif msg_type == "error": show_msg = " 33[1;31m%s 33[0m"%msg elif msg_type == "msg": show_msg = " 33[1;37m%s 33[0m"%msg else: show_msg = " 33[1;32m%s 33[0m"%msg print(show_msg) def progressbar(cur, total): ''' 进度条输出函数 :param cur: 目前的长度 :param total: 总共的长度 :return: none ''' percent = '{:.2%}'.format(cur / total) sys.stdout.write(' ') sys.stdout.write('[%-50s] %s' % ('=' * int(math.floor(cur * 50 / total)), percent)) # print('[%-50s] %s ' % ('=' * int(math.floor(cur * 50 / total)), percent),) sys.stdout.flush() time.sleep(0.01) if cur == total: sys.stdout.write(' ')
4、FTPServer/src/Server_start.py
import socketserver,os,sys Base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(Base_dir) from conf import settings from common import show from user import User class FTPserver(socketserver.BaseRequestHandler): ''' 服务端类 ''' def handle(self): ''' 重构handle :return: none ''' if self.request.recv(1024) == b'True': show("收到{0}的连接请求,正在通信中。。。".format(self.client_address),"info") # try: while True: self.cmd = self.request.recv(4069) # print(self.cmd) if not self.cmd: break elif self.cmd == b'': break else: data = str(self.cmd.decode(encoding="utf-8")) res = data.split("+") if hasattr(self,res[0]): func = getattr(self,res[0]) func(res) else: show("wrong action","error") # except Exception as e: # print(e) # show("客户端发生错误","erroe") def login(self,res): ''' 登陆函数 :param res: 命令结果 :return: none ''' show("收到客户端登陆的请求,正在登陆。。。", "msg") name = res[1] psd = res[2] user = User(name, psd) sign = user.login() if sign: self.request.sendall(bytes("Success", encoding="utf-8")) else: self.request.sendall(bytes("Failure", encoding="utf-8")) def register(self,res): ''' 注册 :param res: 命令结果 :return: none ''' show("收到客户端注册的请求,正在注册。。。", "msg") name = res[1] psd = res[2] user = User(name, psd) if user.register(): self.request.sendall(bytes("Success", encoding="utf-8")) else: self.request.sendall(bytes("Failure", encoding="utf-8")) def view(self,res): ''' 查看当前目录下文件函数 :param res: 命令结果 :return: none ''' show("收到客户端查看当前目录文件的请求。。。", "msg") name = res[1] psd = res[2] user = User(name, psd) dirs,files = user.view_file() # print(dirs,files) dir = str(dirs) file = str(files) if len(dirs) == 0: self.request.sendall("False".encode("utf-8")) else: self.request.sendall(bytes(dir, encoding="utf-8")) self.request.sendall(bytes(file, encoding="utf-8")) self.request.recv(1024) dic = User.info_read(name) storage = str(dic["storage"]) # print(storage,type(storage)) self.request.sendall(bytes(storage, encoding="utf-8")) show("当前目录文件查看或创建成功", "info") def upload(self,res): ''' 上传文件函数 :param res: 命令结果 :return: none ''' show("收到客户端上传文件的请求。。。", "msg") name = res[1] filename = res[3] self.request.sendall(bytes("True", encoding="utf-8")) res = int(self.request.recv(1024).decode()) # print(res,"nice") if User.receive(filename, name, res,self.request): self.request.sendall(bytes("True", encoding="utf-8")) else: self.request.sendall(bytes("False", encoding="utf-8")) def download(self,res): ''' 下载文件函数 :param res: 命令结果 :return: none ''' show("收到客户端下载文件的请求。。。", "msg") name = res[1] psd = res[2] user = User(name, psd) dirs,files = user.view_file() file = str(files) self.request.sendall(bytes(file, encoding="utf-8")) res = self.request.recv(1024).decode() # print(str(res)) if User.download_file(res,name,self.request): show("文件下载成功", "info") else: show("文件下载失败", "error") def cd(self,res): ''' “cd”函数 :param res: 命令结果 :return: none ''' show("收到客户端“cd”的请求。。。", "msg") name = res[1] psd = res[2] user = User(name,psd) res = user.cd_command(self.request) dirs = str(res) self.request.sendall(str(dirs).encode("utf-8")) dir1 = self.request.recv(1024).decode() if dir1 == "error": show("客户端输入错误","error") self.request.close() else: sign = user.cd_dir(self.request,dir1,name) # print(sign) self.request.sendall(str(sign).encode("utf-8")) def mkdir(self,res): show("收到客户端“mkdir”的请求。。。", "msg") name = res[1] psd = res[2] user = User(name,psd) dir_name = res[3] # print(dir_name) sign = user.mkdir(self.request,dir_name) # print(sign) if sign: self.request.sendall("True".encode("utf-8")) else: self.request.sendall("False".encode("utf-8")) def rm(self,res): show("收到客户端“rm”的请求。。。", "msg") name = res[1] psd = res[2] user = User(name,psd) files = user.rm(self.request) # print(files) self.request.sendall(str(files).encode("utf-8")) file_name = self.request.recv(1024).decode() if file_name == "error": show("客户端不稳定","error") else: sign = user.rm_file(self.request,file_name) if sign: show("文件删除成功", "info") self.request.sendall("True".encode("utf-8")) else: self.request.sendall("False".encode("utf-8")) if __name__ == '__main__': show("等待客户端连接。。。", "info") server = socketserver.ThreadingTCPServer(("localhost",settings.PORT),FTPserver) server.serve_forever()
5、FTPServer/src/user.py
import os,sys,pickle,socket,time,random,hashlib Base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(Base_dir) from conf import settings from common import show,progressbar class User(object): ''' 用户类 ''' def __init__(self,username,psd): self.name = username self.password = psd self.home_path = settings.user_home + "/" +self.name def login(self): ''' 用户登陆方法 :return: ''' user_dic = User.info_read(self.name) print(user_dic,"qqqq") if user_dic.get(self.name) == self.password: show("登陆成功","info") user_dic["dir"] = self.home_path User.info_write(self.name,user_dic) return True else: show("登陆失败,用户名或密码错误","error") return False def register(self): ''' 用户注册方法 :return: ''' dic = {} dic[self.name] = self.password dic["storage"] = random.randint(10240,20480) dic["dir"] = self.home_path if User.info_write(self.name,dic): show("注册成功","info") os.mkdir(self.home_path) os.mkdir("%s/others" % self.home_path) with open("%s空白文件" % self.home_path, "w") as f: f.write("空白文件") return True else: show("注册失败","error") return False def view_file(self): ''' 查看当前目录下文件 :return: 目录下文件名组成的列表 ''' if not os.path.exists(self.home_path): os.mkdir(self.home_path) os.mkdir("%s/others"%self.home_path) with open("%s空白文件"%self.home_path,"w") as f: f.write("空白文件") user_dic = User.info_read(self.name) if user_dic["dir"] == os.path.join(os.path.join(Base_dir, "home"), self.name): dir = os.path.join(os.path.join(Base_dir, "home"), self.name) else: dir = user_dic["dir"] for root, dirs, files in os.walk(dir): return dirs,files def cd_command(self,con): for root,dirs,files in os.walk(self.home_path): return dirs def cd_dir(self,con,dir,name): next_dir = self.home_path+"/" +dir user_dic = User.info_read(name) # print(user_dic) user_dic["dir"] = next_dir User.info_write(name,user_dic) return True def mkdir(self,con,res_dir): user_dic = User.info_read(self.name) if user_dic["dir"] == os.path.join(os.path.join(Base_dir, "home"), self.name): dir = os.path.join(os.path.join(Base_dir, "home"), self.name) else: dir = user_dic["dir"] next_dir = dir+"/" +res_dir if os.path.exists(next_dir): show("该目录已存在", "error") return False else: os.mkdir(next_dir) show("目录创建成功","info") return True def rm(self,con): user_dic = User.info_read(self.name) if user_dic["dir"] == os.path.join(os.path.join(Base_dir, "home"), self.name): dir = os.path.join(os.path.join(Base_dir, "home"), self.name) else: dir = user_dic["dir"] for root,dirs,files in os.walk(dir): return files def rm_file(self,con,file_name): user_dic = User.info_read(self.name) if user_dic["dir"] == os.path.join(os.path.join(Base_dir, "home"), self.name): dir = os.path.join(os.path.join(Base_dir, "home"), self.name) else: dir = user_dic["dir"] os.remove(dir+"/" +file_name) return True @staticmethod def download_file(filename,name,con): ''' 下载文件静态方法 :param filename: 文件名 :param name: 用户名 :param con: 标志 :return: none ''' user_dic = User.info_read(name) if user_dic["dir"] == os.path.join(os.path.join(Base_dir, "home"), name): user_dir = os.path.join(os.path.join(Base_dir, "home"), name) else: user_dir = user_dic["dir"] dir = os.path.join(user_dir, filename) f = open(dir,"rb") m = hashlib.md5() data = f.read() m.update(data) a = str(len(data))+"+"+ m.hexdigest() # print(a) con.sendall(bytes(a, encoding="utf-8")) con.sendall(data) f.close() return True @staticmethod def receive(filename,name,res,con): ''' 接收文件静态方法 :param filename: 文件名 :param name: 用户名 :param con: 标志 :return: none ''' user_dic = User.info_read(name) if user_dic["dir"] == os.path.join(os.path.join(Base_dir, "home"), name): dir = os.path.join(os.path.join(os.path.join(Base_dir, "home"), name), filename) else: dir_name = user_dic["dir"] dir = os.path.join(dir_name, filename) # print(res,user_dic["storage"]) if res/1024 < user_dic["storage"]: con.sendall("ok".encode("utf-8")) length = 0 f = open(dir, "wb") md5 = hashlib.md5() while length < res: if res - length > 1024: # 要收不止一次 size = 1024 else: # 最后一次了,剩多少收多少 size = res - length # print("最后一次剩余的:", size) data = con.recv(size) length += len(data) md5.update(data) f.write(data) progressbar(length, res) else: new_md5 = md5.hexdigest() f.close() # Num = User.download_Progress(res, length, Num) or_md5 = con.recv(1024) # print(new_md5) # print(or_md5) if new_md5 == or_md5.decode(): show("文件下载成功", "info") return True else: show("文件不一致", "error") return False elif res/1024 > user_dic["storage"]: con.sendall("no".encode("utf-8")) show("磁盘空间不足", "error") return False @staticmethod def info_read(name): ''' 读取用户数据的静态方法 :param name: 用户名 :return: 字典 ''' user_dir = os.path.join(settings.user_info,name) if os.path.exists(user_dir): with open(user_dir,"rb") as f: dic = pickle.load(f) return dic else: print("用户数据为空") @staticmethod def info_write(name,dic): ''' 写入用户数据的静态方法 :param name:用户名 :param dic:用户信息字典 :return:True ''' user_dir = os.path.join(settings.user_info, name) with open(user_dir,"wb") as f: pickle.dump(dic,f) return True