模拟百度云盘版的ftp
思路：
一、分两个大的文件夹，一个是客户端，一个服务端的
二、实现的功能
1. 登陆--对用户名的合法性进行检测（实验账户：alex，123）
注册--设置账户，其中网盘列表设置为空，刚注册的用户给到50下载豆
规则：
用户上传文件是上传到服务端的共享文件库里，同时会添加到自己的网盘里，
上传一个文件就可以奖励50下载豆，并自己上传的文件以后下载不用下载豆
用户去找资源，其实就是把服务端的共享文件库陈列给客户看，客户如果把
服务端的资源文件添加到自己的网盘，必须消耗20下载豆
用户下载不用消耗下载豆，但是只能从自己的网盘里下载
2.进入到FTP客户可以选择：
上传--
遍历自己本地文件库，选择上传，上传过程中显示进度条（支持断点续传）
下载--
遍历自己的网盘，选择下载，下载过程中显示进度条（支持断点下载）
找资源--
遍历服务端的共享文件库，选择添加到自己的网盘
查看我的网盘--
遍历自己的网盘展示
流程图
代码实现
目录：
client端：
client_main.py
import os
import sys
pat = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(pat)
from config import Config
import socket
import time
import json
def main():
#连接服务器
ip_port = ('127.0.0.1',8009)
sk = socket.socket()
sk.connect(ip_port)
#接收欢迎信息
bt = sk.recv(1024)
print(str(bt,encoding='utf-8'))
flag1 = 1
while flag1:
# 让客户选择操作
print('\t1\t登陆\n\t2\t注册')
choose = input('请选择序号选择你的操作：')
if choose == '2':
sk.sendall(bytes(choose,encoding='utf-8'))
print('注册页面！')
while True:
reg_usr = input('请输入注册用户名：')
sk.sendall(bytes(reg_usr,encoding='utf-8'))
ret = str(sk.recv(1024),encoding='utf-8')
if ret == '合法':
reg_pwd = input('请输入注册密码：')
sk.sendall(bytes(reg_pwd,encoding='utf-8'))
bt = sk.recv(1024)
print(str(bt,encoding='utf-8'))
break
else:
print(ret)
elif choose == '1':
sk.sendall(bytes(choose, encoding='utf-8'))
print('登陆页面！')
flag2 = 1
while flag2:
landed_usr = input('请输入用户名：')
sk.sendall(bytes(landed_usr,encoding='utf-8'))
ret = str(sk.recv(1024), encoding='utf-8')
if ret == '存在':
while True:
landed_pwd = input('请输入密码：')
sk.sendall(bytes(landed_pwd,encoding='utf-8'))
bt = sk.recv(1024)
if str(bt,encoding='utf-8') == 'yes':
print('登陆成功！')
flag1 = 0
flag2 = 0
enter_flag = 1
break
else:
print(str(bt, encoding='utf-8'))
else:
print('用户名不存在！请重新输入！')
else:
print('输入有误！请重新输入！')
#+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
while enter_flag:
print('\n\t1\t上传\n'
'\t2\t下载\n'
'\t3\t找资源\n'
'\t4\t查看我的网盘\n')
choose = input('请选择你接下来的操作(输入q退出)：')
#把客户的选择发过去
sk.sendall(bytes(choose, encoding='utf-8'))
if choose == '1':
#遍历本地文件，并让客户选择
for k,file_name in enumerate(Config.local_file_list,1):
print(k,file_name)
m = input('请选择你要上传的文件：')
file_name = Config.local_file_list[int(m)-1]
print(file_name)
#把文件名发过去
sk.sendall(bytes(file_name,encoding='utf-8'))
print('文件名已发过去！')
#接收检测的文件大小
file_size = str(sk.recv(1024),encoding='utf-8')
print('文件大小已收到！',file_size)
file_name_path = os.path.join(Config.PATH,file_name)
all_size = os.stat(file_name_path).st_size
print('总大小：',all_size)
if all_size != int(file_size):
with open(file_name_path,'rb') as f:
num = int(file_size)
f.seek(num)
for line in f:
sk.sendall(line)
file_size = str(sk.recv(1024),encoding='utf-8')
sys.stdout.write('\r') # 每一次清空原行
sys.stdout.write('%s%% |%s' % (int( int(file_size)/ all_size * 100),
int(int(file_size) / all_size / 2 * 100) * '*'))
sys.stdout.flush() # 强制刷新缓冲区
time.sleep(0.3)
sk.sendall(bytes('1',encoding='utf-8'))
print('\n上传完毕！')
else:
print('你要上传的文件已存在！')
elif choose == '2':
s = str(sk.recv(1024), encoding='utf-8')
my_wangpan = json.loads(s)
print('我的网盘：')
for n, file_na in enumerate(my_wangpan, 1):
print(n, file_na)
load_choose = input('请输入序号选择你要下载的文件：')
sk.sendall(bytes(load_choose,encoding='utf-8'))
all_size = str(sk.recv(1024),encoding='utf-8')
print(type(all_size),all_size)
load_file_name = my_wangpan[int(load_choose) - 1]
pth = os.path.join(Config.PATH, load_file_name)
if load_file_name in Config.local_file_list:
file_size = os.stat(pth).st_size
if str(file_size) == all_size:
sk.sendall(bytes('下载过',encoding='utf-8'))
sk.recv(1024)
print('你本地文件里已有要下载的文件！')
else:
print('继续下载...')
sk.sendall(bytes('下载了一点点',encoding='utf-8'))
sk.recv(1024)
sk.sendall(bytes(str(file_size),encoding='utf-8'))
with open(pth,'ab') as f:
while True:
bt = sk.recv(1024)
if int(file_size / int(all_size)) == 1:
break
else:
f.write(bt)
file_size += len(bt)
sys.stdout.write('\r') # 每一次清空原行
sys.stdout.write('%s%% |%s' % (int(file_size / int(all_size) * 100),
int(file_size / int(all_size) / 2 * 100) * '*'))
sys.stdout.flush() # 强制刷新缓冲区
time.sleep(0.3)
print('\n下载完成！')
else:
print('开始下载...')
sk.sendall(bytes('没下载',encoding='utf-8'))
sk.recv(1024)
file_size = 0
sk.sendall(bytes('0',encoding='utf-8'))
with open(pth, 'wb') as f:
while True:
bt = sk.recv(1024)
if int(file_size / int(all_size)) == 1:
break
else:
f.write(bt)
file_size += len(bt)
sys.stdout.write('\r') # 每一次清空原行
sys.stdout.write('%s%% |%s' % (int(file_size / int(all_size) * 100),
int(file_size / int(all_size) / 2 * 100) * '*'))
sys.stdout.flush() # 强制刷新缓冲区
time.sleep(0.3)
print('下载完成！')
elif choose == '3':
print('进入！')
s = str(sk.recv(1024),encoding='utf-8')
print('收到json字符')
li = json.loads(s)
print('资源如下：')
for m,file_name in enumerate(li,1):
print(m,file_name)
choose = input('请选择序号决定你想要添加到网盘的文件（添加一个文件减少20下载豆）：')
sk.sendall(bytes(choose,encoding='utf-8'))
print(str(sk.recv(1024),encoding='utf-8'))
elif choose == '4':
s = str(sk.recv(1024),encoding='utf-8')
my_wangpan = json.loads(s)
print('我的网盘：')
for n,file_na in enumerate(my_wangpan,1):
print(n,file_na)
elif choose.lower() == 'q':
exit()
else:
print('你输入有误！请重新输入！')
if __name__ == '__main__':
main()
config.py
import os import sys pat = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) PATH = os.path.join(pat,'db') local_file_list = os.listdir(PATH)
server端
server_main.py
import os
import sys
pat = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(pat)
from lib import account
from config import Config
import socketserver
import json
import time
import pickle
class MyServer(socketserver.BaseRequestHandler):
def handle(self):
# print self.request,self.client_address,self.server
conn = self.request
conn.sendall(bytes('欢迎来到百度云盘！',encoding='utf-8'))
flag1 = 1
while flag1:
choose = str(conn.recv(1024), encoding='utf-8')
if choose == '2':
while True:
reg_usr = str(conn.recv(1024),encoding='utf-8')
ret = account.usr_ver(reg_usr)
conn.sendall(bytes(ret,encoding='utf-8'))
if ret == '合法':
reg_pwd = str(conn.recv(1024),encoding='utf-8')
hash_pwd = account.encrypt(reg_pwd)
ob = account.Account()
ob.registered(reg_usr,hash_pwd)
account.info_add(ob)
conn.sendall(bytes('注册成功！',encoding='utf-8'))
break
elif choose == '1':
flag2 = 1
while flag2:
landed_usr = str(conn.recv(1024),encoding='utf-8')
ret = account.land_usr_ver(landed_usr)
conn.sendall(bytes(ret,encoding='utf-8'))
if ret == '存在':
while True:
landed_pwd = str(conn.recv(1024),encoding='utf-8')
obj = account.landed_ver(landed_usr,landed_pwd)
if isinstance(obj,account.Account):
conn.sendall(bytes('yes', encoding='utf-8'))
flag1 = 0
flag2 = 0
break
else:
conn.sendall(bytes(obj, encoding='utf-8'))
#########################################################################################
while True:
#接收客户的操作选择
choose = str(conn.recv(1024),encoding='utf-8')
if choose == '1':
#接收客户发来需要上传的文件名
file_name = str(conn.recv(1024),encoding='utf-8')
pat = os.path.join(Config.PATH,file_name)
#判断上传的文件是否存在
if file_name in Config.shared_file_list:
file_size = os.stat(pat).st_size
#对存在的文件，检测大小，把值传过去
conn.sendall(bytes(str(file_size),encoding='utf-8'))
with open(pat,'ab') as f:
flag = 1
while flag:
bt = conn.recv(1024)
if len(bt) == 1 and bt == bytes('1',encoding='utf-8'):
flag = 0
else:
f.write(bt)
file_size += len(bt)
conn.sendall(bytes(str(file_size), encoding='utf-8'))
obj.netdisc.append(file_name)
obj.download_beans += 50
else:
conn.sendall(bytes('0',encoding='utf-8'))
file_size = 0
with open(pat,'wb') as f:
flag = 1
while flag:
bt = conn.recv(1024)
if len(bt) == 1 and bt == bytes('1', encoding='utf-8'):
flag = 0
else:
f.write(bt)
file_size += len(bt)
conn.sendall(bytes(str(file_size), encoding='utf-8'))
obj.netdisc.append(file_name)
obj.download_beans += 50
elif choose == '2':
s = json.dumps(obj.netdisc)
conn.sendall(bytes(s, encoding='utf-8'))
load_choose = str(conn.recv(1024),encoding='utf-8')
index_num = int(load_choose) - 1
download_file_name = obj.netdisc[index_num]
pat = os.path.join(Config.PATH,download_file_name)
all_size = os.stat(pat).st_size
print(type(all_size),all_size)
conn.sendall(bytes(str(all_size),encoding='utf-8'))
if str(conn.recv(1024),encoding='utf-8') != '下载过':
conn.sendall(bytes('知道',encoding='utf-8'))
seek_num = int(str(conn.recv(1024),encoding='utf-8'))
with open(pat,'rb') as f:
f.seek(seek_num)
for line in f:
conn.sendall(line)
time.sleep(3)
conn.sendall(bytes('0',encoding='utf-8'))
else:
conn.sendall(bytes('知道', encoding='utf-8'))
elif choose == '3':
print('进入！')
s = json.dumps(Config.shared_file_list)
conn.sendall(bytes(s,encoding='utf-8'))
print('已发出json格式信息')
#收到客户选择信息
choose = str(conn.recv(1024),encoding='utf-8')
index_num = int(choose) - 1
index_file_name = Config.shared_file_list[index_num]
print(index_file_name)
if obj.download_beans >= 20:
obj.netdisc.append(index_file_name)
obj.download_beans -= 20
conn.sendall(bytes('添加成功',encoding='utf-8'))
else:
conn.sendall(bytes('下载豆不足！添加失败！',encoding='utf-8'))
elif choose.lower() == 'q':
with open(Config.ACCOUNT_INFO_PATH,'rb') as f:
li = pickle.loads(f.read())
lis = []
for i in li:
if i.username == obj.username:
lis.append(obj)
else:
lis.append(i)
with open(Config.ACCOUNT_INFO_PATH,'wb') as f:
f.write(pickle.dumps(lis))
elif choose == '4':
s = json.dumps(obj.netdisc)
conn.sendall(bytes(s,encoding='utf-8'))
if __name__ == '__main__':
server = socketserver.ThreadingTCPServer(('127.0.0.1',8009),MyServer)
server.serve_forever()
config.py
import os import sys pat = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(pat) ACCOUNT_INFO_PATH = os.path.join(pat,'db','用户信息.log') PATH = os.path.join(pat,'db','共享文件') shared_file_list = os.listdir(PATH)
account.py
import os
import sys
pat = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(pat)
from config import Config
import pickle
import hashlib
class Account:
def __init__(self):
self.username = None
self.password = None
self.download_beans = 0
self.netdisc = []
def registered(self,usr,pwd):
self.username = usr
self.password = pwd
self.download_beans = 50
def usr_ver(usr):
'''
注册时验证用户名存不存在
:param usr: 需要验证用户名
:return: 存在--返回用户名已存在，不存在--返回合法
'''
if os.path.exists(Config.ACCOUNT_INFO_PATH):
with open(Config.ACCOUNT_INFO_PATH,'rb') as f:
li = pickle.load(f)
for i in li:
if i.username == usr:
return '用户名已存在！请重新输入！'
return '合法'
else:
return '合法'
def encrypt(pwd):
'''
給密码加密的
:param pwd: 需要加密的密码
:return: 返回加密后的哈西值
'''
m = hashlib.md5(bytes('明天更美好',encoding='utf-8'))
m.update(bytes(pwd,encoding='utf-8'))
return m.digest()
def info_add(obj):
'''
把注册好信息写入文件
:param obj: 注册的账户对象
:return:
'''
if os.path.exists(Config.ACCOUNT_INFO_PATH):
with open(Config.ACCOUNT_INFO_PATH,'rb') as f:
li = pickle.load(f)
li.append(obj)
else:
li = []
li.append(obj)
with open(Config.ACCOUNT_INFO_PATH, 'wb') as f:
pickle.dump(li,f)
def land_usr_ver(usr):
'''
登录时验证用户名存不存在
:param usr: 需要验证用户名
:return: 存在--返回存在，不存在--返回你输入的用户名不存在
'''
if os.path.exists(Config.ACCOUNT_INFO_PATH):
with open(Config.ACCOUNT_INFO_PATH,'rb') as f:
li = pickle.load(f)
for i in li:
if i.username == usr:
return '存在'
return '输入的用户名不存在！请重新输入！'
else:
return '输入的用户名不存在！请重新输入！'
def landed_ver(usr,pwd2):
'''
用于用户名和密码验证
:param usr: 验证用户名
:param pwd: 验证密码
:return:
'''
pwd = encrypt(pwd2)
li = pickle.load(open(Config.ACCOUNT_INFO_PATH,'rb'))
for i in li:
if i.username == usr and i.password == pwd:
return i
return '密码错误！请重新输入！'
绝版FTP：猛戳这里
