# 服务端
import socketserver
import os
import json
import hashlib
import struct
class MySocketServer(socketserver.BaseRequestHandler):
users = []
@staticmethod
def get_dic(): # 获取密码库
user_dic = {}
file_name = 'register'
if file_name not in os.listdir(os.path.dirname(__file__)):
return user_dic
else:
with open(os.path.join(os.path.dirname(__file__), file_name), mode='r', encoding='utf-8') as f:
for line in f:
if line:
username, psw = line.strip().split('|')
user_dic[username] = psw
return user_dic
def logon(self): # 注册
user_dic = self.get_dic()
time = 3
while time > 0:
username, psw = self.request.recv(1024).decode('utf-8').split('|')
if 0 <= len(username) <= 20 and 6 <= len(psw) <= 20 and (username not in user_dic):
with open(os.path.join(os.path.dirname(__file__), 'register'), mode='a', encoding='utf-8') as f:
md5 = hashlib.md5()
md5.update(psw.encode('utf-8'))
f.write(f'{username}|{md5.hexdigest()}\n')
self.request.send('True'.encode())
MySocketServer.users.append(username)
self.user = username
return True
else:
time -= 1
self.request.send(str(time).encode('utf-8'))
return False
def login(self): # 登录
time = 3
user_dic = self.get_dic()
while time > 0:
username, psw = self.request.recv(1024).decode('utf-8').split('|')
md5 = hashlib.md5()
md5.update(psw.encode('utf-8'))
if user_dic.get(username) == md5.hexdigest() and username not in MySocketServer.users:
self.request.send('True'.encode())
MySocketServer.users.append(username)
self.user = username
return True
else:
time -= 1
self.request.send(str(time).encode('utf-8'))
return False
def up(self): # 上传
while 1:
len_head_dic = struct.unpack('i', self.request.recv(4))[0]
head_dic = json.loads(self.request.recv(len_head_dic).decode('utf-8'))
file_name = head_dic['file_name']
file_md5 = head_dic['file_md5']
file_size = head_dic['file_size']
if self.user not in os.listdir(os.path.join(os.path.dirname(__file__), '用户文件')):
os.makedirs(f'用户文件/{self.user}')
file_path = os.path.join(os.path.dirname(__file__), f'用户文件/{self.user}', file_name)
with open(file_path, mode='wb') as f:
md5 = hashlib.md5()
len_data = 0
while len_data < file_size:
data = self.request.recv(1024)
len_data += len(data)
md5.update(data)
f.write(data)
data_md5 = md5.hexdigest()
if data_md5 == file_md5:
self.request.send('True'.encode('utf-8'))
return True
else:
self.request.send('False'.encode('utf-8'))
def down(self): # 下载
file_path = os.path.join(os.path.dirname(__file__), '用户文件')
file_user = os.listdir(file_path)
if self.user not in file_user or not os.listdir(os.path.join(file_path, f'{self.user}')):
self.request.send('False'.encode('utf-8'))
else:
file_lst = os.listdir(os.path.join(file_path, f'{self.user}'))
file_show = ''
for n, file in enumerate(file_lst, 1):
file_show += f'序号:{n}\t文件名:{file}\n'
to_client = f'可下载文件:\n{file_show}'.encode('utf-8')
self.request.send(to_client)
while 1:
from_client = self.request.recv(1024).decode('utf-8')
try:
with open(os.path.join(file_path, f'{self.user}', file_lst[int(from_client) - 1]), mode='rb') as f:
file_size = 0
md5 = hashlib.md5()
while 1:
data = f.read(1024)
if data:
md5.update(data)
file_size += len(data)
else:
break
file_md5 = md5.hexdigest()
self.request.send('True'.encode('utf-8'))
except Exception:
self.request.send('False.'.encode('utf-8'))
continue
else:
head_dic = {
'file_md5': file_md5,
'file_name': file_lst[int(from_client) - 1],
'file_size': file_size
}
head_dic_json_bytes = json.dumps(head_dic).encode('utf-8')
len_head_dic_bytes = struct.pack('i', len(head_dic_json_bytes))
self.request.send(len_head_dic_bytes)
self.request.send(head_dic_json_bytes)
with open(os.path.join(file_path, f'{self.user}', file_lst[int(from_client) - 1]), mode='rb') as f:
while 1:
data = f.read(1024)
if data:
self.request.send(data)
else:
return True
def handle(self):
while 1:
try:
from_client = self.request.recv(1024).decode('utf-8')
func = getattr(self, from_client)
ret = func()
if ret == False:
break
except Exception:
MySocketServer.users.remove(self.user)
break
server = socketserver.ThreadingTCPServer(('192.168.13.19', 2020), MySocketServer)
server.serve_forever() # 对象执行serve_forever方法，开启服务端
# 客户端
import socket
import os
import json
import hashlib
import struct
import sys
import time
def progress(percent, width=50): # 打印进度条
if percent >= 1:
percent = 1
show_str = ('%%-%ds' % width) % (int(width*percent)*'|')
print('\r%s %d%%' %(show_str, int(100*percent)), end='')
class MyClient:
login_logon = {'1': 'login',
'2': 'logon'}
func_dic = {
'1': 'up',
'2': 'down',
'3': 'exit'
}
def __init__(self, name, ip, port):
self.name = name
self.socket = socket.socket()
self.socket.connect((ip, port))
def logon(self): # 注册
print('-----欢迎访问注册页面-----')
while 1:
username = input('请输入用户名:').strip()
psw = input('请输入密码:').strip()
self.socket.send(f'{username}|{psw}'.encode('utf-8'))
from_server = self.socket.recv(1024).decode('utf-8')
if from_server == 'True':
print('恭喜你,注册成功~~~~~~~~')
return True
elif from_server == '0':
print('啊哈,注册失败了,正在退出....')
return False
else:
print(f'再来一次吧,还有{from_server}次机会~~~~')
def login(self): # 登录
print('-----欢迎访问登录页面-----')
while 1:
username = input('请输入用户名:').strip()
psw = input('请输入密码:').strip()
self.socket.send(f'{username}|{psw}'.encode('utf-8'))
from_server = self.socket.recv(1024).decode('utf-8')
if from_server == 'True':
print('恭喜你,登录成功~~~~~~~~')
return True
elif from_server == '0':
print('啊哈,登录失败了,正在退出....')
return False
else:
print(f'再来一次吧,还有{from_server}次机会~~~~')
def up(self): # 上传
print('-----欢迎访问上传页面-----')
while 1:
file_path = input('请输入上传的文件路径:').strip()
try:
with open(file_path, mode='rb') as f:
file_size = 0
md5 = hashlib.md5()
while 1:
data = f.read(1024)
if data:
md5.update(data)
file_size += len(data)
else:
break
file_md5 = md5.hexdigest()
except Exception:
print('路径错误,请重新输入.')
continue
else:
head_dic = {
'file_md5': file_md5,
'file_name': os.path.basename(file_path),
'file_size': file_size
}
head_dic_json_bytes = json.dumps(head_dic).encode('utf-8')
len_head_dic_bytes = struct.pack('i', len(head_dic_json_bytes))
self.socket.send(len_head_dic_bytes)
self.socket.send(head_dic_json_bytes)
with open(file_path, mode='rb') as f:
data_size = 0
while 1:
data = f.read(1024)
if data:
self.socket.send(data)
data_size += len(data)
percent = data_size / file_size
progress(percent, width=70)
else:
break
from_server = self.socket.recv(1024).decode('utf-8')
if from_server == 'True':
print('恭喜你,上传成功.')
return True
else:
print('啊哈,上传失败.')
def down(self): # 下载
print('-----欢迎访问下载页面-----')
from_server = self.socket.recv(1024).decode('utf-8')
if from_server == 'False':
print('你的云盘为空,请先上传文件~~~~~~')
else:
print(from_server)
while 1:
choice = input('请输入选择的序号:').strip().encode('utf-8')
self.socket.send(choice)
return_from_server = self.socket.recv(1024).decode('utf-8')
if return_from_server == 'True':
len_head_dic = struct.unpack('i', self.socket.recv(4))[0]
head_dic = json.loads(self.socket.recv(len_head_dic).decode('utf-8'))
file_name = head_dic['file_name']
file_md5 = head_dic['file_md5']
file_size = head_dic['file_size']
file_path = os.path.join(os.path.dirname(__file__), '下载', file_name)
with open(file_path, mode='wb') as f:
md5 = hashlib.md5()
len_data = 0
while len_data < file_size:
data = self.socket.recv(1024)
len_data += len(data)
md5.update(data)
f.write(data)
percent = len_data / file_size
progress(percent, width=70)
percent = len_data / file_size
progress(percent, width=70)
data_md5 = md5.hexdigest()
if data_md5 == file_md5:
print('下载成功.')
return True
else:
print('下载失败.')
else:
print('输入错误,请重新输入.')
def exit(self): # 退出
print('正在退出~~~~~~')
exit()
def run(self):
print('-----[1]登录[2]注册-----')
while 1:
choice = input('请输入选项:').strip()
if choice in self.login_logon:
self.socket.send(self.login_logon[choice].encode('utf-8'))
func = getattr(self, self.login_logon[choice])
ret = func()
if ret:
while 1:
print('-----[1]上传[2]下载[3]退出-----')
choice_func = input('请输入选项:').strip()
if choice in self.func_dic:
self.socket.send(self.func_dic[choice_func].encode('utf-8'))
method = getattr(self, self.func_dic[choice_func])
method()
else:
print('输入错误,请重新输入.')
else:
break
else:
print('输入错误,请重新输入.')
client = MyClient('客户端', '192.168.13.19', 2020)
client.run()