Python--网络编程-----文件传输简单版本

一、服务端代码：

import socket
import subprocess
import struct
import json
import os

phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# phone.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
phone.bind(('127.0.0.1', 8081))  # 0-65535:0-1024给操作系统使用
phone.listen(5)

print('starting......')
while True:  # 链接循环
    conn, client_addr = phone.accept()
    print(client_addr)

    while True:  # 通信循环
        try:
            # 1、收命令
            res = conn.recv(8096)  # b'get a.txt'
            if not res:  # 适用于linux操作系统
                break

            # 2、解析命令，提取相应命令参数
            cmds = res.decode('utf-8').split()  # ['get', 'a.txt']
            filename = cmds[1]
            # 3、以读的方式打开文件，读取文件内容发送到客户端

            # 第一步：制作固定长度的报头
            header_dic = {
                'filename': filename,
                'md5': 'xxdxx',
                'file_size': os.path.getsize(filename)
            }

            header_json = json.dumps(header_dic)
            header_bytes = header_json.encode('utf-8')

            # 第二步：先发送报头的长度
            conn.send(struct.pack('i', len(header_bytes)))

            # 第三步：把包头发送给客户端
            conn.send(header_bytes)

            # 第三步：再发送真实的数据
            with open(filename, 'rb') as f:
                # conn.send(f.read())
                for line in f:
                    conn.send(line)

        except ConnectionResetError:  # 适用于Windows操作系统
            break

    conn.close()

phone.close()

二、客户端代码：

import socket
import struct
import json


phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

phone.connect(('127.0.0.1', 8081))

while True:
    # 1、发命令
    cmd = input('>>:').strip()  # get a.txt
    if not cmd:
        continue
    phone.send(cmd.encode('utf-8'))

    # 2、以写的方式打开一个新文件，接收服务器端发来的文件内容写入客户的新文件

    # 第一步：先收报头的长度
    obj = phone.recv(4)
    header_size = struct.unpack('i', obj)[0]

    # 第二步：再收报头
    header_bytes = phone.recv(header_size)

    # 第三步：从报头中解析出对真实数据的描述信息
    header_json = header_bytes.decode('utf-8')
    header_dic = json.loads(header_json)
    print(header_dic)
    total_size = header_dic['file_size']
    filename = header_dic['filename']

    # 第四步： 接收真实的数据
    with open(filename, 'wb') as f:
        recv_size = 0
        while recv_size < total_size:
            line = phone.recv(1024)
            f.write(line)
            recv_size += len(line)


phone.close()

先启动服务端，再启动客户端，然后在客户端执行 get 1.txt, get test.py  都显示下载成功：

客户端打印字典信息为：

>>:get 1.txt
{'filename': '1.txt', 'md5': 'xxdxx', 'file_size': 8}
>>:get test.py
{'filename': 'test.py', 'md5': 'xxdxx', 'file_size': 60}
>>:

以上代码已经实现了客户端从服务端下载文件的功能，如果想把数据从客户端上传到服务端，则把服务端和客户端代码

做相应调整即可。

如果server提供下载文件夹share，client指定下载文件夹download，

客户端和服务端代码改为：

客户端代码：
import socket
import struct
import json

download_dir = r'C:Usersxu516PycharmProjectsPython全栈开发第三模块网络编程6_文件传输clientdownload'
phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

phone.connect(('127.0.0.1', 8081))

while True:
    # 1、发命令
    cmd = input('>>:').strip()  # get a.txt
    if not cmd:
        continue
    phone.send(cmd.encode('utf-8'))

    # 2、以写的方式打开一个新文件，接收服务器端发来的文件内容写入客户的新文件

    # 第一步：先收报头的长度
    obj = phone.recv(4)
    header_size = struct.unpack('i', obj)[0]

    # 第二步：再收报头
    header_bytes = phone.recv(header_size)

    # 第三步：从报头中解析出对真实数据的描述信息
    header_json = header_bytes.decode('utf-8')
    header_dic = json.loads(header_json)
    print(header_dic)
    total_size = header_dic['file_size']
    filename = header_dic['filename']

    # 第四步： 接收真实的数据
    with open('%s\%s' % (download_dir, filename), 'wb') as f:
        recv_size = 0
        while recv_size < total_size:
            line = phone.recv(1024)
            f.write(line)
            recv_size += len(line)
            print('总大小：%s  已下载大小：%s' % (total_size, recv_size))


phone.close()

服务端代码：
import socket
import subprocess
import struct
import json
import os

share_dir = r'C:Usersxu516PycharmProjectsPython全栈开发第三模块网络编程6_文件传输servershare'
phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# phone.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
phone.bind(('127.0.0.1', 8081))  # 0-65535:0-1024给操作系统使用
phone.listen(5)

print('starting......')
while True:  # 链接循环
    conn, client_addr = phone.accept()
    print(client_addr)

    while True:  # 通信循环
        try:
            # 1、收命令
            res = conn.recv(8096)  # b'get a.txt'
            if not res:  # 适用于linux操作系统
                break

            # 2、解析命令，提取相应命令参数
            cmds = res.decode('utf-8').split()  # ['get', 'a.txt']
            filename = cmds[1]
            # 3、以读的方式打开文件，读取文件内容发送到客户端

            # 第一步：制作固定长度的报头
            header_dic = {
                'filename': filename,
                'md5': 'xxdxx',
                'file_size': os.path.getsize('%s\%s' % (share_dir, filename))
            }

            header_json = json.dumps(header_dic)
            header_bytes = header_json.encode('utf-8')

            # 第二步：先发送报头的长度
            conn.send(struct.pack('i', len(header_bytes)))

            # 第三步：把包头发送给客户端
            conn.send(header_bytes)

            # 第三步：再发送真实的数据
            with open('%s\%s' % (share_dir, filename), 'rb') as f:
                # conn.send(f.read())
                for line in f:
                    conn.send(line)

        except ConnectionResetError:  # 适用于Windows操作系统
            break

    conn.close()

phone.close()

 客户端下载 get  test.py 运行结果为：

>>:get test.py
{'filename': 'test.py', 'md5': 'xxdxx', 'file_size': 60}
总大小：60  已下载大小：60
>>: