关于python:python实现tcp上传下载文件

2次阅读

共计 3674 个字符,预计需要花费 10 分钟才能阅读完成。

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()

正文完
 0