FTP上传下载文件(面向对象版)

Posted 2022-04-28 zyyhxbs

# 服务端
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'可下载文件:\nfile_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()