python-实现tcp上传下载文件

Posted 2021-09-03 Gendan

import socket
import struct
import json
import os
sc= socket.socket()
sc.connect((\'127.0.0.1\',9997))
\'\'\' 验证用户名密码的函数，验证区 \'\'\'
def send_user(user,pwd):#发送用户的账户和密码

user_len = len(user.encode(\'utf-8\'))
leng = struct.pack(\'i\',user_len)
sc.send(leng)#用户的字节码长度
sc.send(user.encode(\'utf-8\'))#发送用户名的字节码

​

pwd_len = len(pwd.encode(\'utf-8\'))
leng = struct.pack(\'i\', pwd_len)
sc.send(leng)#密码的字节码长度
sc.send(pwd.encode(\'utf-8\'))#密码的字节码

\'\'\' 接收服务端的功能的函数，功能区 \'\'\'

上传文件的路径

上传文件的路径

path = r\'F:\\installsotf\\python file\\python全栈\\day30\\作业\\test_up\'

下载文件的保存位置

load_path=r\'F:\\installsotf\\python file\\python全栈\\day30\\作业\\download_file\'

接收服务端提供的功能列表，功能循环中第一个调用

def recv_function_list(sc):

leng = sc.recv(4)
leng = struct.unpack(\'i\', leng)[0]
cont = sc.recv(leng).decode(\'utf-8\')
return cont

将要发送的文件名和大小写到字典中,下面三个函数，实现文件的上传

def get_file(path):

file=os.path.basename(path)
size=os.path.getsize(path)
dic = {\'file\':file,\'size\':size}
return dic

def up_file(path,size):

with open(path, \'rb\')as fp:
    while size > 0:
        msg = fp.read(1024)
        sc.send(msg)
        size -= len(msg)

def upto_file(path):

# path=input(\'输入文件的绝对路径：\').strip()
if os.path.exists(path):
    dic = get_file(path)  # 获取字典【如果将path改成输入，就可以随意上传文件了】
    size = dic[\'size\']
    dic_byte = json.dumps(dic).encode(\'utf-8\')  # 先转成json，再转成bytes
    dic_leng = len(dic_byte)  # 获取bytes的长度
    leng = struct.[PayPal下载](https://www.gendan5.com/wallet/PayPal.html)pack(\'i\', dic_leng)
    sc.send(leng)  # 发送字典dic的长度
    sc.send(dic_byte)  # 发送字典的内容
    # 打开要操作的文件，用rb模式
    up_file(path, size)  # 调用上传文件
    return True
else:
    return False

下面这个函数实现文件的下载：

def down_file(path):#将文件下载到哪个位

# 收到用户的选择。
# print(58)
leng = sc.recv(4)  # 接收文件名和大小组成的字典
# print(60,len(leng))
leng = struct.unpack(\'i\', leng)[0]
dic = sc.recv(leng).decode(\'utf-8\')
dic = json.loads(dic)  # 获取到{‘file’:file,\'size\':size}
# print(dic)
# print(65)
path = os.path.join(path, dic[\'file\'])
size = dic[\'size\']
with open(path, \'wb\')as fp:
    while size > 0:
        cont = sc.recv(1024)
        fp.write(cont)
        size -= len(cont)

while 1:

print(\'1、登录\\n2、注册\')
chioce = input(\'输入的选择（q/Q退出）：\').strip()
chioce_list=[\'登录\',\'注册\']
if chioce==\'1\' or chioce==\'2\':
    user = input(\'输入用户名：\').strip()
    pwd = input(\'输入密码：\').strip()
    sc.send(chioce.encode(\'utf-8\'))#发送选择给服务端
    send_user(user,pwd)
    ret = sc.recv(3).decode(\'utf-8\')#yes 或者 not
    if ret ==\'yes\':
        print(f\'{chioce_list[int(chioce)-1]}成功。\')
        \'\'\'  接收服务端提供功能的代码 :功能区\'\'\'
        if chioce==\'1\':#代表用户是在登录，且登录成功了，那么就可以享受服务端提供的上传下载文件功能。
            while 1:
                cont = recv_function_list(sc)
                print(cont)
                choice = input(\'输入你的选择（e/E退出）：\').strip()
                if choice.isdecimal():
                    sc.send(choice.encode(\'utf-8\'))  # 告诉服务器要进行的操作
                    choice = int(choice)
                    if choice == 1:  # 上传文件
                        while 1:
                            path = input(\'输入文件绝对路径：\')
                            path = path.replace(\'\\\\\', \'/\')
                            ret = upto_file(path)
                            if ret:
                                print(\'上传成功\')
                            num = input(\'回车继续上传，stop退出上传：\').strip()
                            if num==\'goto\' or \'stop\':
                                sc.send(num.encode(\'utf-8\'))
                                if num == \'stop\':
                                    break
                            else:
                                sc.send(\'none\'.encode(\'utf-8\'))
                    elif choice == 2:
                        # 1、接收文件列表的长度和内容
                        len_list = sc.recv(4)  # 接收列表的长度字节码
                        leng = struct.unpack(\'i\', len_list)[0]  # unpacke成数字
                        leng = int(leng)
                        file_list = sc.recv(leng).decode(\'utf-8\')
                        file_list = json.loads(file_list)
                        while 1:
                            print(\'-*\' * 20)
                            print()
                            for i, name in enumerate(file_list, 1):
                                print(f\'{i},{name}\')
                            num = input(\'输入你选择：\').strip()
                            if num.isdecimal():
                                num = int(num)
                                if num > 0 and num <= len(file_list):
                                    # 2、发送文件序号的长度和内容
                                    leng = len(str(num).encode(\'utf-8\'))
                                    leng = struct.pack(\'i\', leng)
                                    sc.send(leng)  # 将文件序号的字节长度发送过去
                                    sc.send(str(num).encode(\'utf-8\'))  # 将文件序号发送过去。
                                    # print(\'调用down_file前\')#走到这里了
                                    # 3、接收文件的内容
                                    down_file(load_path)
                                    print(\'下载文件成功。\')
                                    # print(\'调用down_file后\')
                                    yes_not = input(\'回车继续下载，输入 stop 退出：\').strip()
                                    if yes_not==\'stop\' or yes_not==\'goto\':
                                        sc.send(yes_not.encode(\'utf-8\'))
                                        if yes_not == \'stop\':
                                            break
                                    else:
                                        sc.send("none".encode(\'utf-8\'))
                                else:
                                    print(\'无此选择。\')
                                    sc.send(\'0\'.encode(\'utf-8\'))
                            else:
                                print(\'输入有误。\')
                    else:
                        print(\'无此选择。\')
                elif choice.upper() == \'E\':
                    sc.send(\'E\'.encode(\'utf-8\'))
                    break
                else:
                    print(\'输入有误。\')
                \'\'\'接收服务端提供功能的代码的尾部：功能区\'\'\'
    else:
        print(f\'{chioce_list[int(chioce)-1]}失败。\')
elif chioce.upper()==\'Q\':
    #发送Q给服务端，说明要退出。
    sc.send(\'Q\'.encode(\'utf-8\'))
    print(\'退出系统。\')
    break
else:
    print(\'输入有误。\')

sc.close()