标签：FTP 上传下载 dic 面向对象 file path data self md5

# 服务端
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()

标签：FTP,上传下载,dic,面向对象,file,path,data,self,md5
来源： https://www.cnblogs.com/zyyhxbs/p/11218289.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。