基于tcp的云盘上传下载的模拟

老师的博客：

server端

import json

import struct

import json

import struct

import socket

import os

sk = socket.socket()

sk.bind(('192.168.43.155',8080))

sk.listen()

conn,addr = sk.accept()

dic_len = conn.recv(4)  # 4个字节 数字的大小

dic_len = struct.unpack('i',dic_len)[0]

content = conn.recv(dic_len).decode('utf-8')  #

content_dic = json.loads(content)

if content_dic['operate'] == 'upload':

    with open(content_dic['filename'],'wb') as f:

        while content_dic['filesize']:

            file = conn.recv(1024)

            f.write(file)

            content_dic['filesize'] -= len(file)

elif content_dic['operate'] == 'download':

    file_path=content_dic['path']

    file_name=os.path.basename(file_path)

    file_size=os.path.getsize(file_path)

    dic_from={'file_size':file_size,'file_name':file_name}

    dic_from=json.dumps(dic_from).encode('utf-8')

    dic_lenth=struct.pack('i',len(dic_from))

    print(dic_lenth,type(dic_lenth))

    conn.send(dic_lenth + dic_from)

    with open(file_path,mode='rb') as f:

        while file_size>0:

            ret=f.read(1024)

            conn.send(ret)

            file_size -= 1024

    print('传输完成')

else:

    print('错了')

conn.close()

sk.close()

client端

import os

import json

import struct

import socket

sk = socket.socket()

sk.connect(('192.168.43.155',8080))

def get_filename(file_path):

    filename = os.path.basename(file_path)#os.path.basename(path) 返回path最后的文件名。

    # 如果path以／或\结尾，那么就会返回空值。即os.path.split(path)的第二个元素

    return filename

#选择 操作

operate = ['upload','download']

for num,opt in enumerate(operate,1):#枚举函数，第一个参数是对象，第二是是下标的开始计数

    print(num,opt)

num = int(input('请输入您要做的操作序号 : '))

if num == 1:

    '''上传操作'''

    file_path = input('请输入要上传的文件路径 : ')

    file_size = os.path.getsize(file_path)  # 获取文件大小

    file_name = get_filename(file_path)#获取文件路径

    dic = {'operate': 'upload', 'filename': file_name,'filesize':file_size}

    str_dic = json.dumps(dic).encode('utf-8')

    ret = struct.pack('i', len(str_dic))  # 将字典的大小转换成一个定长(4)的bytes

    sk.send(ret + str_dic)

    with open(file_path,'rb') as  f:

        while file_size:

            content = f.read(1024)

            sk.send(content)

            file_size -= len(content)

    print('上传成功')

elif num == 2:

    '''下载操作'''

    download_path=input('请输入你要下载的路径')

    dic=json.dumps({'operate':'download','path':download_path}).encode('utf-8')

    download_path_unm=struct.pack('i',len(dic))

    sk.send(download_path_unm+dic)#发送信息

    get_information=sk.recv(4)#接受即将带来的信息的长度

    get_information_real=struct.unpack('i',get_information)[0]#转化为真是数字

    get_real_information=sk.recv(get_information_real).decode('utf-8')#接受，解压

    get_real_information=json.loads(get_real_information)

    download_file_name=get_real_information['file_name']

    download_file_size=get_real_information['file_size']

    with  open(download_file_name,mode='wb') as f:

            while download_file_size>0:

                print(download_file_size)

                code=sk.recv(1024)

                f.write(code)

                download_file_size -= 1024

    print('下载成功')

sk.close()

说明：

上传老师写的，我自己是吧下载给补充的写了，所以上传和下载的代码应该还可以优化一下。

下面是在写代码中遇到的问题，总结一下：

1.struct中，把数字转化为4的固定长度的bytes类型，所以就不需要再进行encode操作，因为本身就是bytes类型的，而却其编码方式叫特别，只能通过unpack才能转化为数字，记住是int，通过其他方式。例如。gbk。utf-8decode是str类型，但是内容确实空白的·另外关于unpack得到的是个元祖，所以的到的值需要取第0位数，来取到你需要的数字。

2，json模块，由于json是把数据类型给转化为str类型的，所以还需要在进行encode（）的操作，来吧其转化为bytes类型的。所以拿到是，应该首先decode（）。

3，send的是时候可以采用（内容1+内容2）的形式，表示两次发送的

4，下载上传是必需要把文件的大小拿到，然后采用相等的大小传输，所以需要把文件大小先传输过去，然后在进行同时的收发操作。

5.while循环是，条件应该写成长度<0是结束，因为有很大的情况下，大小-=1024是不可能刚好减到0的，如果为负数的时候，条件还是为真，所以不会终止循环，然后就出不来了。

6，数据传输和接受的时候都是以bytes的方式打开或者写入的，而且在不需要写编码方式，因为传输的有可能是视频。