def start():
server_ip = settings.FTP_SERVER_IP
server_port = settings.FTP_SERVER_PORT
common.show_message(template.START_MENU, "INFO")
common.show_message(
"正在连接FTP服务器 {0}:{1} ......".format(server_ip, server_port), "INFO")
# 创建一个client对象
client = Client(server_ip, server_port)
# 连接服务器，返回结果
conn_result = client.connect()
# 连接成功
if conn_result == codes.CONN_SUCC:
common.show_message("连接成功!", "NOTICE")
# 客户端登录
client_login(client)
# 登录成功
if client.login_status:
# 退出系统
exit_flag = False
while not exit_flag:
# 显示登录后的菜单
show_menu = template.LOGINED_MENU.format(
client.username,
str(int(client.totalspace / 1024 /
1024)), #固定配额 不存在为小数 int 去整
str(round(client.usedspace / 1024 / 1024, 2))) #保留两位小数点 显示
common.show_message(show_menu, "NOTICE")
# 输入命令串
input_command = common.input_command("[ 输入命令 ] : ")
if input_command == "quit":
exit_flag = True
else:
# 获取命令的的command
cmd_func = input_command.split("|")[0]
try:
# 通过反射调用Client类的对应command方法,并返回执行结果
if hasattr(Client, cmd_func):
func = getattr(Client, cmd_func)
exec_result = func(client, input_command)
print(exec_result)
else:
common.writelog(
"Client {0} 未找到".format(input_command),
"error")
except Exception as e:
common.writelog(e, "error")
else:
common.show_message("连接失败", "ERROR")
def connect(self):
"""
客户端连接方法,连接失败记录日志，返回状态码
:return: 成功 10000 / 失败 10001
"""
try:
self.socket.connect(self._server)
recv_server_data = self.socket.recv(100)
if str(recv_server_data, 'utf-8') == "OK":
return codes.CONN_SUCC
else:
return codes.CONN_FAIL
except Exception as e:
common.writelog(e, "error")
return codes.CONN_FAIL
def connect(self):
"""
客户端连接方法,连接失败记录日志，返回状态码
:return: 成功 10000 / 失败 10001
"""
try:
self.socket.connect(self._server)
recv_server_data = self.socket.recv(100)
if str(recv_server_data, 'utf-8') == "OK":
return codes.CONN_SUCC
else:
return codes.CONN_FAIL
except Exception as e:
common.writelog(e, "error")
return codes.CONN_FAIL
def start():
server_ip = settings.FTP_SERVER_IP
server_port = settings.FTP_SERVER_PORT
common.show_message(template.START_MENU, "INFO")
common.show_message("正在连接FTP服务器 {0}:{1} ......".format(server_ip, server_port), "INFO")
# 创建一个client对象
client = Client(server_ip, server_port)
# 连接服务器，返回结果
conn_result = client.connect()
# 连接成功
if conn_result == codes.CONN_SUCC:
common.show_message("连接成功!", "NOTICE")
# 客户端登录
client_login(client)
# 登录成功
if client.login_status:
# 退出系统
exit_flag = False
while not exit_flag:
# 显示登录后的菜单
show_menu = template.LOGINED_MENU.format(client.username,
str(int(client.totalspace / 1024 / 1024)),
str(int(client.usedspace / 1024 / 1024)))
common.show_message(show_menu, "NOTICE")
# 输入命令串
input_command = common.input_command("[ 输入命令 ] : ")
if input_command == "quit":
exit_flag = True
else:
# 获取命令的的command
cmd_func = input_command.split("|")[0]
try:
# 通过反射调用Client类的对应command方法,并返回执行结果
if hasattr(Client, cmd_func):
func = getattr(Client, cmd_func)
exec_result = func(client, input_command)
print(exec_result)
else:
common.writelog("Client {0} 未找到".format(input_command), "error")
except Exception as e:
common.writelog(e, "error")
else:
common.show_message("连接失败", "ERROR")
def show(self, *args):
"""
客户端执行show命令
:param args: 对于此方法此参数无效
:return:返回文件列表
"""
try:
# 发送命令到服务端
self.socket.send(bytes("show|", encoding='utf8'))
# 接收服务端发送结果的大小
total_data_len = self.socket.recv(100).decode()
# 收到了 发送一个ready标识
self.socket.send(bytes("ready", 'utf8'))
# 开始接收数据， 客户端每次接收2048bytes
recved_data_len = 0
exec_result = bytes("", 'utf8')
print(total_data_len)
while recved_data_len < int(total_data_len):
if int(total_data_len) - recved_data_len > 2048:
recv_data = self.socket.recv(2048)
recved_data_len += len(recv_data)
# exec_result += str(recv_data, 'utf-8')
exec_result += recv_data
else:
# 小于2048了
recv_data = self.socket.recv(
int(total_data_len) - recved_data_len)
# exec_result += str(recv_data, 'utf-8')
recved_data_len += len(recv_data)
exec_result += recv_data
# 获取结果中文件及文件夹的数量
return_result = str(exec_result, 'utf-8')
print(len(return_result))
file_count = int(return_result.split("|")[0])
if file_count == 0:
return_result = "目前无上传记录"
else:
return_result = return_result.split("|")[1]
return return_result
except Exception as e:
common.writelog("client - show - {0}".format(e), "error")
def show(self, *args):
"""
客户端执行show命令
:param args: 对于此方法此参数无效
:return:返回文件列表
"""
try:
# 发送命令到服务端
self.socket.send(bytes("show|", encoding='utf8'))
# 接收服务端发送结果的大小
total_data_len = self.socket.recv(100).decode()
# 收到了 发送一个ready标识
self.socket.send(bytes("ready", 'utf8'))
# 开始接收数据， 客户端每次接收2048bytes
recved_data_len = 0
exec_result = bytes("", 'utf8')
print(total_data_len)
while recved_data_len < int(total_data_len):
if int(total_data_len) - recved_data_len > 2048:
recv_data = self.socket.recv(2048)
recved_data_len += len(recv_data)
# exec_result += str(recv_data, 'utf-8')
exec_result += recv_data
else:
# 小于2048了
recv_data = self.socket.recv(int(total_data_len) - recved_data_len)
# exec_result += str(recv_data, 'utf-8')
recved_data_len += len(recv_data)
exec_result += recv_data
# 获取结果中文件及文件夹的数量
return_result = str(exec_result, 'utf-8')
print(len(return_result))
file_count = int(return_result.split("|")[0])
if file_count == 0:
return_result = "目前无上传记录"
else:
return_result = return_result.split("|")[1]
return return_result
except Exception as e:
common.writelog("client - show - {0}".format(e), "error")
def get(self, command):
"""
从服务器端下载文件,下载必须进入到文件所在的目录
:param command: 下载命令 "get|filename"
:return: 结果
"""
return_result = ""
# 发送基本信息到服务端 (command,username,file)
self.socket.send(bytes(command, encoding='utf-8'))
# 先接收到命令是否正确标识,1 文件存在, 0 文件不存在
ack_by_server = self.socket.recv(100)
try:
# 文件名错误,当前路径下找不到
if str(ack_by_server, encoding='utf-8') == "0":
return_result = "\n当前目录下未找到指定的文件,请到存在目录下执行get操作!"
else:
# 给服务端回应收到，防止连包
self.socket.send(bytes("ok", 'utf8'))
# 文件存在,开始接收文件基本信息(大小,文件名)
file_info = self.socket.recv(100).decode()
file_size = int(file_info.split("|")[0])
file_name = file_info.split("|")[1]
file_md5 = file_info.split("|")[2]
# 2 发送 ready 标识，准备开始接收文件
self.socket.send(bytes("ready", 'utf8'))
# 3 开始接收数据了,每次接收2048
recv_size = 0
with open(os.path.join(settings.DOWNLOAD_FILE_PATH, file_name), 'a+b') as fw:
while file_size - recv_size > 2048:
recv_data = self.socket.recv(2048)
fw.write(recv_data)
recv_size += len(recv_data)
common.print_process(file_size, recv_size)
else:
recv_data = self.socket.recv(file_size - recv_size)
fw.write(recv_data)
common.print_process(file_size, file_size)
return_result = "\n文件下载成功"
common.writelog("download file<{0}> from server successful".format(file_name), "info")
# md5 验证
check_md5 = common.encry_md5(os.path.join(settings.DOWNLOAD_FILE_PATH, file_name))
if check_md5 == file_md5:
common.writelog("md5 check for file<{0}> succ!".format(file_name), "info")
return_result += ", MD5 验证成功! "
else:
return_result += ", MD5 验证不匹配啊! "
return return_result
except Exception as e:
common.writelog(e, "error")
def get(self, command):
"""
从服务器端下载文件,下载必须进入到文件所在的目录
:param command: 下载命令 "get|filename"
:return: 结果
"""
return_result = ""
# 发送基本信息到服务端 (command,username,file)
self.socket.send(bytes(command, encoding='utf-8'))
# 先接收到命令是否正确标识,1 文件存在, 0 文件不存在
ack_by_server = self.socket.recv(100)
try:
# 文件名错误,当前路径下找不到
if str(ack_by_server, encoding='utf-8') == "0":
return_result = "\n当前目录下未找到指定的文件,请到存在目录下执行get操作!"
else:
# 给服务端回应收到，防止连包
self.socket.send(bytes("ok", 'utf8'))
# 文件存在,开始接收文件基本信息(大小,文件名)
file_info = self.socket.recv(100).decode()
file_size = int(file_info.split("|")[0])
file_name = file_info.split("|")[1]
file_md5 = file_info.split("|")[2]
# 2 发送 ready 标识，准备开始接收文件
self.socket.send(bytes("ready", 'utf8'))
# 3 开始接收数据了,每次接收2048
recv_size = 0
with open(os.path.join(settings.DOWNLOAD_FILE_PATH, file_name), 'a+b') as fw:
while file_size - recv_size > 2048:
recv_data = self.socket.recv(2048)
fw.write(recv_data)
recv_size += len(recv_data)
common.print_process(file_size, recv_size)
else:
recv_data = self.socket.recv(file_size - recv_size)
fw.write(recv_data)
common.print_process(file_size, file_size)
return_result = "\n文件下载成功"
common.writelog("download file<{0}> from server successful".format(file_name), "info")
# md5 验证
check_md5 = common.encry_md5(os.path.join(settings.DOWNLOAD_FILE_PATH, file_name))
if check_md5 == file_md5:
common.writelog("md5 check for file<{0}> succ!".format(file_name), "info")
return_result += ", MD5 验证成功! "
else:
return_result += ", MD5 验证不匹配啊! "
return return_result
except Exception as e:
common.writelog(e, "error")
def put(self, command):
"""
文件上传方法,传入文件全路径
:param command: 命令 "put|file"
:return: 成功 3000 / 失败 3002 / 文件不存在 3001
"""
file_name = command.split("|")[1]
# 判断文件是否存在
if os.path.isfile(file_name):
# 获取文件名
fname = os.path.basename(file_name)
fsize = os.path.getsize(file_name)
fmd5 = common.encry_md5(file_name)
# 将基本信息发过去
cmd = "put"
file_msg = "{cmd}|{file}|{filesize}|{filemd5}".format(cmd=cmd,
file=fname,
filesize=fsize,
filemd5=fmd5)
self.socket.send(bytes(file_msg, encoding='utf8'))
common.writelog("send file basic info:{0}".format(file_msg), "info")
# 获取服务端返回状态(是否是新文件及接收到的大小)"status|recved_size"
# 未收过 "0|0", 断点续传 "1|recved_size"
ack_by_server = self.socket.recv(100)
common.writelog("upload file, received from server:{0}".format(str(ack_by_server, 'utf-8')), "info")
if str(ack_by_server, encoding='utf-8').split("|")[0] == "0":
# 新文件
sended_size = 0
# 判断文件是否超过磁盘配额空间
if self.usedspace + fsize > self.totalspace:
return "超过磁盘配额,无法上传文件,请联系管理员!"
else:
self.usedspace += fsize
else:
sended_size = int(str(ack_by_server, encoding='utf-8').split("|")[1])
print("此文件已发送过,但未发送完，开始断点续传!......")
# 开始发送文件
try:
with open(file_name, 'rb') as f:
f.seek(sended_size)
while fsize - sended_size > 2048:
contant = f.read(2048)
r_size = len(contant)
self.socket.send(contant)
sended_size += r_size
common.print_process(fsize, sended_size)
else:
contant = f.read(fsize - sended_size)
self.socket.send(contant)
common.print_process(fsize, fsize)
common.writelog("upload file<{0}> successful".format(file_name), "info")
return "\n文件发送成功"
except Exception as e:
common.writelog("upload file failed!{0}".format(e), "error")
return "文件发送失败!"
else:
return "文件不存在"
def put(self, command):
"""
文件上传方法,传入文件全路径
:param command: 命令 "put|file"
:return: 成功 3000 / 失败 3002 / 文件不存在 3001
"""
file_name = command.split("|")[1]
# 判断文件是否存在
if os.path.isfile(file_name):
# 获取文件名
fname = os.path.basename(file_name)
fsize = os.path.getsize(file_name)
fmd5 = common.encry_md5(file_name)
# 将基本信息发过去
cmd = "put"
file_msg = "{cmd}|{file}|{filesize}|{filemd5}".format(
cmd=cmd, file=fname, filesize=fsize, filemd5=fmd5)
self.socket.send(bytes(file_msg, encoding='utf8'))
common.writelog("send file basic info:{0}".format(file_msg),
"info")
# 获取服务端返回状态(是否是新文件及接收到的大小)"status|recved_size"
# 未收过 "0|0", 断点续传 "1|recved_size"
ack_by_server = self.socket.recv(100)
common.writelog(
"upload file, received from server:{0}".format(
str(ack_by_server, 'utf-8')), "info")
if str(ack_by_server, encoding='utf-8').split("|")[0] == "0":
# 新文件
sended_size = 0
# 判断文件是否超过磁盘配额空间
if self.usedspace + fsize > self.totalspace:
return "超过磁盘配额,无法上传文件,请联系管理员!"
else:
self.usedspace += fsize
else:
sended_size = int(
str(ack_by_server, encoding='utf-8').split("|")[1])
print("此文件已发送过,但未发送完，开始断点续传!......")
# 开始发送文件
try:
with open(file_name, 'rb') as f:
f.seek(sended_size)
while fsize - sended_size > 2048:
contant = f.read(2048)
r_size = len(contant)
self.socket.send(contant)
sended_size += r_size
common.print_process(fsize, sended_size)
else:
contant = f.read(fsize - sended_size)
self.socket.send(contant)
common.print_process(fsize, fsize)
common.writelog(
"upload file<{0}> successful".format(file_name), "info")
return "\n文件发送成功"
except Exception as e:
common.writelog("upload file failed!{0}".format(e), "error")
return "文件发送失败!"
else:
return "文件不存在"
def put(self, command):
"""
文件上传方法,传入文件全路径
:param command: 命令 "put|file"
:return: 成功 3000 / 失败 3002 / 文件不存在 3001
"""
file_name = command.split("|")[1].strip()
#判断是否为目录
if os.path.isdir(file_name): #文件夹的处理
#压缩目录 ，获得包数据
zip_url= common.createZip(file_name,os.path.split(file_name)[0])
#获取压缩文件的，数据资源
fname =os.path.split(zip_url)[1]
fsize =common.getDirSize(zip_url)
fmd5 = common.encry_md5(zip_url)
cmd = "put"
file_msg = "{cmd}|{file}|{filesize}|{filemd5}|{is_dir}".format(cmd=cmd,
file=fname,
filesize=fsize,
filemd5=fmd5,
is_dir='1')
self.socket.send(bytes(file_msg, encoding='utf8'))
common.writelog("send file basic info:{0}".format(file_msg), "info")
# 获取服务端返回状态(是否是新文件及接收到的大小)"status|recved_size"
# 未收过 "0|0", 断点续传 "1|recved_size"
ack_by_server = self.socket.recv(100)
common.writelog("upload file, received from server:{0}".format(str(ack_by_server, 'utf-8')), "info")
if str(ack_by_server, encoding='utf-8').split("|")[0] == "0":
# 新文件
sended_size = 0
# 判断文件是否超过磁盘配额空间
if self.usedspace + fsize > self.totalspace:
return "超过磁盘配额,无法上传文件,请联系管理员!"
else:
self.usedspace += fsize
else:
sended_size = int(str(ack_by_server, encoding='utf-8').split("|")[1])
print("此文件已发送过,但未发送完，开始断点续传!......")
# 开始发送文件
try:
with open(zip_url, 'rb') as f:
f.seek(sended_size) #
while fsize - sended_size > 2048:
contant = f.read(2048)
r_size = len(contant)
self.socket.send(contant)
sended_size += r_size
# 调用进度条函数
# common.print_process(fsize, sended_size)
else:
contant = f.read(fsize - sended_size)
self.socket.send(contant)
# 调用进度条函数
common.print_process(fsize, fsize)
common.writelog("upload file<{0}> successful".format(zip_url), "info")
#文件发送成功时，删除打包的文件 （给用户造成一个 自己是上传文件的假象）
os.remove(zip_url)
return "\n文件发送成功"
except Exception as e:
common.writelog("upload file failed!{0}".format(e), "error")
return "文件发送失败!"
# 判断文件是否存在 （文件的处理）
elif os.path.isfile(file_name):
# 获取文件名
fname = os.path.basename(file_name)
fsize = os.path.getsize(file_name)
fmd5 = common.encry_md5(file_name)
# 将基本信息发过去
cmd = "put"
file_msg = "{cmd}|{file}|{filesize}|{filemd5}|{is_dir}".format(cmd=cmd,
file=fname,
filesize=fsize,
filemd5=fmd5,
is_dir='0')
self.socket.send(bytes(file_msg, encoding='utf8'))
common.writelog("send file basic info:{0}".format(file_msg), "info")
# 获取服务端返回状态(是否是新文件及接收到的大小)"status|recved_size"
# 未收过 "0|0", 断点续传 "1|recved_size"
ack_by_server = self.socket.recv(100)
common.writelog("upload file, received from server:{0}".format(str(ack_by_server, 'utf-8')), "info")
if str(ack_by_server, encoding='utf-8').split("|")[0] == "0":
# 新文件
sended_size = 0
# 判断文件是否超过磁盘配额空间
if self.usedspace + fsize > self.totalspace:
return "超过磁盘配额,无法上传文件,请联系管理员!"
else:
self.usedspace += fsize
else:
sended_size = int(str(ack_by_server, encoding='utf-8').split("|")[1])
print("此文件已发送过,但未发送完，开始断点续传!......")
# 开始发送文件
try:
with open(file_name, 'rb') as f:
f.seek(sended_size) #
while fsize - sended_size > 2048:
contant = f.read(2048)
r_size = len(contant)
self.socket.send(contant)
sended_size += r_size
#调用进度条函数
common.print_process(fsize, sended_size)
else:
contant = f.read(fsize - sended_size)
self.socket.send(contant)
#调用进度条函数
common.print_process(fsize, fsize)
common.writelog("upload file<{0}> successful".format(file_name), "info")
return "\n文件发送成功"
except Exception as e:
common.writelog("upload file failed!{0}".format(e), "error")
return "文件发送失败!"
else:
return "文件不存在"