def download(sock):
max_index = common.show_file_list(
common.msg_unpack_recv(sock)['file_list'])
file_number = input('请输入文件序号：').strip()
if file_number == 'q':
print('退出下载')
return 102
file_number = int(file_number)
# if file_number > max_index or file_number < 0:
# print('请输入正确的序号！')
# return 202
user_dict_ = {'user_key': settings.user_key, 'file_number': file_number}
common.msg_pack_send(user_dict_, sock)
file_dict = common.msg_unpack_recv(sock)
file_path = input('请输入存储路径：（不输入则使用默认路径）')
if not file_path:
file_path = os.path.join(settings.FILE_PATH, file_dict['filename'])
else:
file_path = os.path.join(file_path, file_dict['filename'])
with open(file_path, mode='wb') as f:
n = 0
while n < file_dict['filesize']:
msg = sock.recv(1024)
f.write(msg)
n += len(msg)
print('\r', f'已下载{round(n / file_dict["filesize"], 2) * 100}%')
print('下载完成！')
log1.info(f'下载文件{file_dict["filename"]}到{file_path}')
return 102
def switch(sock):
common.show_file_list(common.msg_unpack_recv(sock)['file_list'])
file_number = int(input('请输入文件夹名(序号)').strip())
user_operation = {
'file_number': file_number,
'user_key': settings.user_key
}
common.msg_pack_send(user_operation, sock)
file_dict = common.msg_unpack_recv(sock)
if file_dict['status']:
common.show_file_list(file_dict['file_list'])
return 102
else:
print('请输入正确的序号！')
return 202
def register(self):
user_dict = common.msg_unpack_recv(self.request)
username = user_dict['username']
if databases.select(username):
common.msg_pack_send({'status':0},self.request)
else:
password = common.ency(user_dict['password'])
databases.create(username,password)
common.msg_pack_send({'status':1},self.request)
def create_dir(self):
dir_dict = common.msg_unpack_recv(self.request)
current_dir, file_list = common.show_dir_list(self.current)
if self.key == dir_dict['user_key']:
if dir_dict['dirname'] not in file_list:
os.mkdir(os.path.join(self.current,dir_dict['dirname']))
common.msg_pack_send({'status':1,'file_list':file_list},self.request)
else:common.msg_pack_send({'status':0},self.request)
else:common.msg_pack_send({'status':0},self.request)
def upload(self):
file_dict = common.msg_unpack_recv(self.request)
path = os.path.join(self.current, file_dict['filename'])
filesize = file_dict['filesize']
n = 0
with open(path,mode='wb') as f:
while n < filesize:
msg = self.request.recv(1024)
f.write(msg)
n += len(msg)
def login(self):
print(self.key)
print(self.current)
user_dict = common.msg_unpack_recv(self.request)
password = common.ency(user_dict['password'])
flag = databases.select(user_dict['username'])
if flag:
if flag['password'] == password:
self.key = common.ency(f'{datetime.datetime.now()}')
return common.msg_pack_send({'key':self.key},self.request)
else:return common.msg_pack_send({'key':0},self.request)
else:return common.msg_pack_send({'key':0},self.request)
def login(sock):
username = input('请输入用户名：')
password = input('请输入密码：')
user_dict_ = {'username': username, 'password': password}
common.msg_pack_send(user_dict_, sock)
key_dict = common.msg_unpack_recv(sock)
settings.user_key = key_dict['key']
if not settings.user_key:
print('用户名或密码错误')
return 202
else:
print('登录成功')
log1.info(f'{username}登录')
return 102
def switch(self):
current_dir, file_list = common.show_dir_list(self.current)
file_dict = {'file_list':file_list}
common.msg_pack_send(file_dict, self.request)
file_dict = common.msg_unpack_recv(self.request)
if file_dict['user_key'] == self.key:
dirname = file_list[file_dict['file_number']]
if os.path.isdir(os.path.join(self.current,dirname)):
self.current = os.path.join(self.current,dirname)
_,file_list = common.show_dir_list(self.current)
dir_dict = {'status':1,'file_list':file_list}
common.msg_pack_send(dir_dict,self.request)
else:common.msg_pack_send({'status':0},self.request)
else:return 505
def create_dir(sock):
dir_name = input('请输入文件夹名：')
if dir_name == 'q':
return 102
dir_dict = {'user_key': settings.user_key, 'dir_name': dir_name}
common.msg_pack_send(dir_dict, sock)
status = common.msg_unpack_recv(sock)
if status['status']:
print('创建成功！')
for i in status['file_list']:
print(i)
return 102
else:
print('创建失败！')
return 202
def handle(self):
conn = self.request
operation_dict = {
1: 'login',
2: 'register',
3: 'download',
4: 'upload',
5: 'switch',
6: 'create_dir',
}
while 1:
sync_dict = common.msg_unpack_recv(conn)
sync = sync_dict['sync']
if sync == 0:
break
getattr(self,f'{operation_dict[sync]}')()
def download(self):
current_dir, file_list = common.show_dir_list(settings.DIR_PATH)
self.current = current_dir
file_dict = {'file_list':file_list}
common.msg_pack_send(file_dict,self.request)
file_msg = common.msg_unpack_recv(self.request)
if self.key == file_msg['user_key']:
filename = file_list[file_msg['file_number']]
path = os.path.join(current_dir,filename)
file_dict = {'filename':filename,'filesize':os.path.getsize(path)}
common.msg_pack_send(file_dict,self.request)
with open(path,mode='rb') as f:
while 1:
msg = f.read(1024)
if msg:
self.request.send(msg)
else:
break
else:
return 505
def register(sock):
username = input('请输入用户名：')
if username == 'q':
return 102
password = input('请输入密码：')
if len(password) < 8 or len(password) > 16:
print('密码长度不符合规范！')
return 301
elif re.search('[~!@#$%^&*()_+=“‘《》<>?/\\-]', password):
print('密码含有非法字符！')
return 301
re_password = input('请确认密码：')
if re_password == password:
user_dict = {'username': username, 'password': password}
common.msg_pack_send(user_dict, sock)
if common.msg_unpack_recv(sock)['status']:
print('注册成功！')
return 102
else:
print('用户名重复')
return 301
else:
print('两次密码不一致！')
return 301
