老男孩python全栈s21day29作业(文件的上传、下载(未用反射版本))

最新推荐文章于 2020-12-09 12:17:00 发布

frantichow

最新推荐文章于 2020-12-09 12:17:00 发布

阅读量503

点赞数

分类专栏： python

本文链接：https://blog.csdn.net/Bensonofljb/article/details/101483127

版权

python 专栏收录该内容

36 篇文章 1 订阅

订阅专栏

1.文件的上传、下载

2.结合用户认证

要求

# 1.tcp协议粘包的问题，使用struct模块解决粘包问题
# 2.传递的所有的信息都尽量是json格式
# 3.再server端用上反射

#server.py

import socket
import json
import hashlib
import struct

def get_md5(user_name,user_pwd):
    md5 = hashlib.md5(user_name.encode('utf-8'))
    md5.update(user_pwd.encode('utf-8'))
    return  md5.hexdigest()


sk = socket.socket()
sk.bind(('127.0.0.1',9000))
sk.listen()
conn,addr = sk.accept()
str_user_dic = conn.recv(1024).decode('utf-8')
user_dic = json.loads(str_user_dic)
#{'user_name': 'ljb', 'user_pwd': 'ljb'}
# print(username,)
with open('user_info',mode='r',encoding='utf-8') as f:
    for line in f:
        username,pwd = line.strip().split('|')
        # print(username,pwd)
        if username == user_dic['user_name'] and get_md5(user_dic['user_name'],user_dic['user_pwd']) == pwd:
            res_dic = {'opt':'login','result':True}
            break
    else:
        res_dic = {'opt': 'login', 'result': False}
ret_jsondic = json.dumps(res_dic)
conn.send(ret_jsondic.encode('utf-8'))
if res_dic['result']:
    print(">>>开始准备接收文件！！")
    # file_dic = conn.recv(1024).decode('utf-8')#接收到文件的json字典
    byte_len = conn.recv(4)#解决粘包问题
    file_dic = conn.recv(struct.unpack('i',byte_len)[0]).decode('utf-8')#解决粘包问题
    str_file_dic = json.loads(file_dic)
    print(str_file_dic)
    content = conn.recv(str_file_dic['filesize'])
    with open(str_file_dic['filename'],mode='wb') as f:
        f.write(content)
    conn.close()
else:
    print("客户端登录失败，自动退出！！")
    conn.close()

sk.close()

#client.py

import socket
import json
import os
import struct

sk = socket.socket()
sk.connect(('127.0.0.1',9000))



user_name = input("请输入登录的用户名：")
user_pwd = input("请输入登录的密码：")
user_dic = {'user_name':user_name,'user_pwd':user_pwd}
str_user_dic = json.dumps(user_dic)
sk.send(str_user_dic.encode('utf-8'))
ret = sk.recv(1024).decode('utf-8')
ret_dic = json.loads(ret)
if ret_dic['result']:
    print('登录成功！！')
    file_path = input("请输入需要上传的文件的绝对路径：")
    #需要考虑粘包
    if os.path.isabs(file_path):
        filename = os.path.basename(file_path)
        filesize = os.path.getsize(file_path)
        dic = {'filename': filename, 'filesize': filesize}
        str_dic = json.dumps(dic).encode('utf-8')
        byte_len = struct.pack('i',len(str_dic))#解决粘包问题
        sk.send(byte_len)
        sk.send(str_dic)
        # str_dic = json.dumps(dic)
        # sk.send(str_dic.encode('utf-8'))
        with open(file_path, mode='rb') as f:
            content = f.read()
            sk.send(content)
        print('upload ok!!')
    else:
        print("请输入需要上传文件的绝对路径！！")
else:
    print('登录异常，用户名或密码错误！！')