def get(self, user_input):
'''
客户端下载方法
:param user_input: 用户输入的指令
:return: 无
'''
import os
if len(user_input) == 2: # 如果是get xxx，则表示下载到当前目录
dst_path = '.'
elif len(user_input) == 3: # 如果是 get xxx xxx，则表示要下载到其他目录
dst_path = user_input[2]
else:
print(self.__code_list['401'])
return None
if os.path.isdir(dst_path): # 判断目标文件是否存在
filename = user_input[1] # 获取下载的文件名
tmp_filename = os.path.join(self.__tmp_path, '%s.ftp' %
os.path.split(filename)[1]) # 定义临时文件
if os.path.isfile(tmp_filename): # 判断临时文件是否存在（也就是是否需要续传）
tmp_file_size = os.path.getsize(tmp_filename) # 获取临时文件的大小
else:
tmp_file_size = 0
self.__sk.send(
mylib.s2b(r'get|{"filename":"%s", "sendsize": %s}' %
(filename, tmp_file_size))) # 向服务端发送下载指令接文件名和已近下载的大小
Confirm = mylib.b2s(self.__sk.recv(100)).split('|') # 获取服务器反馈
if Confirm[0] == 'ready': # 如果是ready则准备下载文件
file_size = json.loads(Confirm[1])['filesize'] # 获取文件大小
md5 = json.loads(Confirm[1])['md5'] # 获取md5
recv_size = tmp_file_size
f = open(tmp_filename, 'ab')
while file_size != recv_size:
# 只要已下载大小不等于文件大小则循环接受数据
try:
data = self.__sk.recv(conf.FILE_PER_SIZE)
recv_size += len(data)
f.write(data)
mylib.process_bar(recv_size, file_size)
except socket.error as e:
print(self.__code_list['306'])
f.close()
break
except IOError as e:
print(self.__code_list['305'])
f.close()
print('')
print('Validating...')
new_md5 = mylib.get_file_md5(tmp_filename)
if new_md5 == md5: # 验证md5，如果验证通过修改成正式文件名，如果没验证过去删除临时文件
import shutil
shutil.move(tmp_filename, os.path.join(dst_path, filename))
else:
os.remove(tmp_filename)
else:
print(self.__code_list[Confirm[1]])
else:
print(self.__code_list['304'])
def put(self, user_input):
'''
客户端上传方法
:param user_input: 指令
:return: 无
'''
import json
if len(user_input) == 2:
file_name = user_input[1]
if not os.path.isfile(file_name):
print(self.__code_list['302'])
return None
self.__sk.sendall(mylib.s2b('checkspacesize'))
res = json.loads(mylib.b2s(self.__sk.recv(100)))
free_size = int(res['freesize']) * 1024 * 1024 # 获取剩余空间大小
file_size = os.path.getsize(file_name)
if free_size > file_size: # 判断剩余空间是否够，够就下载，不够就直接返回
md5 = mylib.get_file_md5(file_name) # 获取上传文件的md5
self.__sk.send(
mylib.s2b(
r'put|{"filename":"%s", "file_size":%s, "md5":"%s"}' %
(os.path.split(file_name)[1], file_size,
md5))) # 发送上传指令，并通知服务端文件名，文件大小和md5
res = mylib.b2s(self.__sk.recv(200)).split('|') # 获取服务端确认信息
if res[0] == 'ready':
send_size = json.loads(res[1])['recv_size'] # 获取已经上传的大小
f = open(file_name, 'rb')
f.seek(send_size) # 设定文件的其实位置为已上传的大小
while file_size != send_size:
# 只要文件大小不等于上传的大小则循环读文件并上传数据
if file_size - send_size > conf.FILE_PER_SIZE:
data = f.read(conf.FILE_PER_SIZE)
send_size += conf.FILE_PER_SIZE
else:
data = f.read(file_size - send_size)
send_size += (file_size - send_size)
self.__sk.send(data)
mylib.process_bar(send_size, file_size)
print(mylib.b2s(self.__sk.recv(200)))
print("")
else:
print(self.__code_list['307'])
else:
print(self.__code_list['401'])
def get(self, user_input):
'''
客户端下载方法
:param user_input: 用户输入的指令
:return: 无
'''
import os
if len(user_input) == 2: # 如果是get xxx，则表示下载到当前目录
dst_path = '.'
elif len(user_input) == 3: # 如果是 get xxx xxx，则表示要下载到其他目录
dst_path = user_input[2]
else:
print(self.__code_list['401'])
return None
if os.path.isdir(dst_path): # 判断目标文件是否存在
filename = user_input[1] # 获取下载的文件名
tmp_filename = os.path.join(self.__tmp_path,'%s.ftp' %os.path.split(filename)[1]) # 定义临时文件
if os.path.isfile(tmp_filename): # 判断临时文件是否存在（也就是是否需要续传）
tmp_file_size = os.path.getsize(tmp_filename) # 获取临时文件的大小
else:
tmp_file_size = 0
self.__sk.send(mylib.s2b(r'get|{"filename":"%s", "sendsize": %s}' %(filename, tmp_file_size))) # 向服务端发送下载指令接文件名和已近下载的大小
Confirm = mylib.b2s(self.__sk.recv(100)).split('|') # 获取服务器反馈
if Confirm[0] == 'ready': # 如果是ready则准备下载文件
file_size = json.loads(Confirm[1])['filesize'] # 获取文件大小
md5 = json.loads(Confirm[1])['md5'] # 获取md5
recv_size = tmp_file_size
f = open(tmp_filename, 'ab')
while file_size != recv_size:
# 只要已下载大小不等于文件大小则循环接受数据
try:
data = self.__sk.recv(conf.FILE_PER_SIZE)
recv_size += len(data)
f.write(data)
mylib.process_bar(recv_size, file_size)
except socket.error as e:
print(self.__code_list['306'])
f.close()
break
except IOError as e:
print(self.__code_list['305'])
f.close()
print('')
print('Validating...')
new_md5 = mylib.get_file_md5(tmp_filename)
if new_md5 == md5: # 验证md5，如果验证通过修改成正式文件名，如果没验证过去删除临时文件
import shutil
shutil.move(tmp_filename, os.path.join(dst_path, filename))
else:
os.remove(tmp_filename)
else:
print(self.__code_list[Confirm[1]])
else:
print(self.__code_list['304'])
def rm(self, user_input):
'''
删除文件方法
:param user_input: 用户命令
:return: 无
'''
if len(user_input) == 2:
self.__sk.send(mylib.s2b('rm|{"path" : "%s"}' %user_input[1])) # 向服务端发送删除指令
res = json.loads(mylib.b2s(self.__sk.recv(200))) # 获取执行结果（错误码）
print(self.__code_list[res['code']])
else:
print(self.__code_list['401'])
def put(self, user_input):
'''
客户端上传方法
:param user_input: 指令
:return: 无
'''
import json
if len(user_input) == 2:
file_name = user_input[1]
if not os.path.isfile(file_name):
print(self.__code_list['302'])
return None
self.__sk.sendall(mylib.s2b('checkspacesize'))
res = json.loads(mylib.b2s(self.__sk.recv(100)))
free_size = int(res['freesize']) * 1024 * 1024 # 获取剩余空间大小
file_size = os.path.getsize(file_name)
if free_size > file_size: # 判断剩余空间是否够，够就下载，不够就直接返回
md5 = mylib.get_file_md5(file_name) # 获取上传文件的md5
self.__sk.send(mylib.s2b(r'put|{"filename":"%s", "file_size":%s, "md5":"%s"}' %(os.path.split(file_name)[1], file_size, md5))) # 发送上传指令，并通知服务端文件名，文件大小和md5
res = mylib.b2s(self.__sk.recv(200)).split('|') # 获取服务端确认信息
if res[0] == 'ready':
send_size = json.loads(res[1])['recv_size'] # 获取已经上传的大小
f = open(file_name, 'rb')
f.seek(send_size) # 设定文件的其实位置为已上传的大小
while file_size != send_size:
# 只要文件大小不等于上传的大小则循环读文件并上传数据
if file_size - send_size > conf.FILE_PER_SIZE:
data = f.read(conf.FILE_PER_SIZE)
send_size += conf.FILE_PER_SIZE
else:
data = f.read(file_size - send_size)
send_size += (file_size - send_size)
self.__sk.send(data)
mylib.process_bar(send_size, file_size)
print(mylib.b2s(self.__sk.recv(200)))
print("")
else:
print(self.__code_list['307'])
else:
print(self.__code_list['401'])
def rm(self, user_input):
'''
删除文件方法
:param user_input: 用户命令
:return: 无
'''
if len(user_input) == 2:
self.__sk.send(mylib.s2b('rm|{"path" : "%s"}' %
user_input[1])) # 向服务端发送删除指令
res = json.loads(mylib.b2s(self.__sk.recv(200))) # 获取执行结果（错误码）
print(self.__code_list[res['code']])
else:
print(self.__code_list['401'])
def checkspacesize(self, instructions):
'''
检查剩余空间是否可以上传文件方法
:param client_cmd: 用户命令
:return: 无
'''
# 获取剩余空间大小
import subprocess
cmd_call = subprocess.Popen('du -s %s' %self.__home_path, shell = True, stdout = subprocess.PIPE)
res = cmd_call.stdout.read()
# 计算剩余空间，并返回
free_size = self.__current_user.get_quota() * 1024 * 1024 - int(mylib.b2s(res).split()[0])
self.request.send(mylib.s2b(r'{"freesize":%s}' %free_size))
def auth(self, user_input):
'''
认证方法
:param user_input: 用户输入命令
:return:
'''
username = input('username: '******'password: '******'auth|%s|%s' % (username, mylib.jiami(password)))) # 调用服务端的认证方法，验证用户名密码
res = mylib.b2s(self.__sk.recv(100)) # 获取验证结果
print(res)
if res == 'ok': # 如果验证成功，修改当前用户名和登录状态
self.__current_user = username
self.__is_login = True
def cd(self, user_input):
'''
切换服务端目录方法
:param user_input: 用户指令
:return: 无
'''
if len(user_input) == 2:
self.__sk.send(mylib.s2b('cd|{"path" : "%s"}' %user_input[1])) # 发送cd指令
res = json.loads(mylib.b2s(self.__sk.recv(200))) # 获取服务端反馈
if res['code'] == '500':
self.__current_path = res['path'] # 修改当前目录等于返回来的目录
else:
print(self.__code_list[res['code']])
else:
print(self.__code_list['401'])
def put(self, client_cmd):
'''
上传文件方法
:param client_cmd: 客户端命令
:return: 无
'''
import os
import shutil
# 拼接要上传的文件名
print(client_cmd)
filename = self.__current_path + os.path.split(client_cmd[1])[1]
file_size = int(client_cmd[2])
#filename = os.path.split(filename)[1]
md5 = client_cmd[3] # 从客户端命令中获取文件的md5值，稍后用于比对
# 定义临时文件
tmp_filename = '%s.ftp' % filename
# 判断临时文件是否存在，如果存在获取临时文件的大小，传给客户端，端点续传
if os.path.isfile(tmp_filename):
tmp_file_size = os.path.getsize(tmp_filename)
else:
tmp_file_size = 0
self.request.sendall(mylib.s2b('put|ready|%s' % tmp_file_size))
Confirm = mylib.b2s(self.request.recv(100))
recv_size = tmp_file_size
if Confirm.startswith('put|ready'):
# 判断是否是续传，如果是续传文件以追加的形式打开，如果不是则以写的方式打开
if recv_size == 0:
f = open('%s.ftp' % filename, 'wb')
else:
f = open('%s.ftp' % filename, 'ab')
# 接收文件知道传输结束
while file_size != recv_size:
data = self.request.recv(conf.FILE_PER_SIZE)
# 如果传输中断，退出整个方法，这里用到了一个技巧，return一个None，强制结束
if len(data) == 0:
f.close()
return None
recv_size += len(data)
f.write(data)
f.close()
# 对比文件md5值如果相同降临时文件重命名，否则删除
new_md5 = mylib.get_file_md5('%s.ftp' % filename)
if md5 == new_md5:
shutil.move('%s.ftp' % filename, filename)
self.request.sendall(mylib.s2b('上传成功'))
else:
os.remove(tmp_filename)
self.request.sendall(mylib.s2b('上传失败'))
def put(self, client_cmd):
'''
上传文件方法
:param client_cmd: 客户端命令
:return: 无
'''
import os
import shutil
# 拼接要上传的文件名
print(client_cmd)
filename = self.__current_path + os.path.split(client_cmd[1])[1]
file_size = int(client_cmd[2])
#filename = os.path.split(filename)[1]
md5 = client_cmd[3] # 从客户端命令中获取文件的md5值，稍后用于比对
# 定义临时文件
tmp_filename = '%s.ftp' %filename
# 判断临时文件是否存在，如果存在获取临时文件的大小，传给客户端，端点续传
if os.path.isfile(tmp_filename):
tmp_file_size = os.path.getsize(tmp_filename)
else:
tmp_file_size = 0
self.request.sendall(mylib.s2b('put|ready|%s' %tmp_file_size))
Confirm = mylib.b2s(self.request.recv(100))
recv_size = tmp_file_size
if Confirm.startswith('put|ready'):
# 判断是否是续传，如果是续传文件以追加的形式打开，如果不是则以写的方式打开
if recv_size == 0:
f = open('%s.ftp' %filename, 'wb')
else:
f = open('%s.ftp' %filename, 'ab')
# 接收文件知道传输结束
while file_size != recv_size:
data = self.request.recv(conf.FILE_PER_SIZE)
# 如果传输中断，退出整个方法，这里用到了一个技巧，return一个None，强制结束
if len(data) == 0:
f.close()
return None
recv_size += len(data)
f.write(data)
f.close()
# 对比文件md5值如果相同降临时文件重命名，否则删除
new_md5 = mylib.get_file_md5('%s.ftp' %filename)
if md5 == new_md5:
shutil.move('%s.ftp' %filename, filename)
self.request.sendall(mylib.s2b('上传成功'))
else:
os.remove(tmp_filename)
self.request.sendall(mylib.s2b('上传失败'))
def cd(self, user_input):
'''
切换服务端目录方法
:param user_input: 用户指令
:return: 无
'''
if len(user_input) == 2:
self.__sk.send(mylib.s2b('cd|{"path" : "%s"}' %
user_input[1])) # 发送cd指令
res = json.loads(mylib.b2s(self.__sk.recv(200))) # 获取服务端反馈
if res['code'] == '500':
self.__current_path = res['path'] # 修改当前目录等于返回来的目录
else:
print(self.__code_list[res['code']])
else:
print(self.__code_list['401'])
def auth(self, user_input):
'''
认证方法
:param user_input: 用户输入命令
:return:
'''
if len(user_input) ==1:
username = input('username: '******'password: '******'auth|{"username":"******", "password":"******"}' % (username, mylib.jiami(password)))) # 调用服务端的认证方法，验证用户名密码
res = mylib.b2s(self.__sk.recv(200)) # 获取验证结果
print(self.__code_list[res])
if res == '200': # 如果验证成功，修改当前用户名和登录状态
self.__current_user = username
self.__is_login = True
else:
print(self.__code_list['401'])
def start(self):
'''
启动客户端方法
:return: 无
'''
print(mylib.b2s(self.__sk.recv(100))) # 显示欢迎信息
while True: # 循环获取用户输入的而命令，如果输入quit退出循环，并退出客户端
user_input = input('%s >> ' %(self.__current_user)).strip()
if len(user_input) == 0: continue
user_input = user_input.split() # 分割用户输入的命令
if user_input[0] == 'quit': # 判断用户输入的命令，quit表示退出
break
if hasattr(self, user_input[0]): # 判断用户命令是否有对应的方法
func = getattr(self, user_input[0]) # 获取方法
func(user_input) # 执行方法
else:
print(self.__code_list['302'])
def auth(self, user_input):
'''
认证方法
:param user_input: 用户输入命令
:return:
'''
if len(user_input) == 1:
username = input('username: '******'password: '******'auth|{"username":"******", "password":"******"}' %
(username, mylib.jiami(password)))) # 调用服务端的认证方法，验证用户名密码
res = mylib.b2s(self.__sk.recv(200)) # 获取验证结果
print(self.__code_list[res])
if res == '200': # 如果验证成功，修改当前用户名和登录状态
self.__current_user = username
self.__is_login = True
else:
print(self.__code_list['401'])
def handle(self):
'''
handle方法
:return: 无
'''
import os
self.__loger = mylib.mylog(conf.LOGS, isstream = True) # 定义日志对象
self.__current_user = user('guest', '', conf.DEFAULT_QUOTA) # 定义默认用户
self.__users = users() # 获取users对象
self.__home_path = os.path.abspath(os.path.join(conf.HOME_PATH,self.__current_user.get_username())).replace('\\', '/') # 获取家目录
self.__current_path = self.__home_path # 定义当前目录
self.__code_list = conf.CODE_LIST # 定义错误里诶表
self.__loger.info('Client %s:%s is connect the server' %self.client_address)
while True:
try:
data = mylib.b2s(self.request.recv(1024)) # 获取客户端命令
except socket.error as e:
self.__loger.info('Has lost client %s:%s' %self.client_address)
break
if not data:
self.__loger.info('Has lost client %s:%s' %self.client_address)
break #如果收不到客户端数据了（代表客户端断开了），就断开
self.instruction_allowcation(data) #客户端发过来的数据统一交给功能分发器处理
def handle(self):
#print(self)
# 定义当前用户，用来保存当前用户信息
self.__current_user = {
"username" : "guest",
"home" : "home/tmp/",
"max_size" : conf.TMP_SPACE_SIZE
}
# 定义用户对象
self.__userobj = users.users()
# self.__users = self.__userobj.get_users()
# print(self.__users)
conn = self.request
# 用户连接后后显示的信息
conn.sendall(bytes('欢迎使用ftp','utf8'))
# 获取当前目录
self.__current_path = self.__current_user['home']
flag = True
while flag:
try:
# 获取用户命令
client_cmd = mylib.b2s(conn.recv(100))
if client_cmd == '': break
# 将用户命令进行分割，客户端传过来的命令需要使用|分割命令本身和命令相关参数
client_cmd = client_cmd.split("|")
# 获取命令
if client_cmd[0] == 'quit': # 如果是quit，退出当前循环，释放连接
flag = False
continue
if hasattr(self, client_cmd[0]): # 根据命令执行相应的方法
func = getattr(self, client_cmd[0])
func(client_cmd)
except Exception:
break
conn.close()
def handle(self):
#print(self)
# 定义当前用户，用来保存当前用户信息
self.__current_user = {
"username": "******",
"home": "home/tmp/",
"max_size": conf.TMP_SPACE_SIZE
}
# 定义用户对象
self.__userobj = users.users()
# self.__users = self.__userobj.get_users()
# print(self.__users)
conn = self.request
# 用户连接后后显示的信息
conn.sendall(bytes('欢迎使用65ftp', 'utf8'))
# 获取当前目录
self.__current_path = self.__current_user['home']
flag = True
while flag:
try:
# 获取用户命令
client_cmd = mylib.b2s(conn.recv(100))
if client_cmd == '': break
# 将用户命令进行分割，客户端传过来的命令需要使用|分割命令本身和命令相关参数
client_cmd = client_cmd.split("|")
# 获取命令
if client_cmd[0] == 'quit': # 如果是quit，退出当前循环，释放连接
flag = False
continue
if hasattr(self, client_cmd[0]): # 根据命令执行相应的方法
func = getattr(self, client_cmd[0])
func(client_cmd)
except Exception:
break
conn.close()