文件上传和多线程通信

文件上传

基于TCP的方法

client

import socket
import os
import json
import struct
FILE_PATH = os.path.join(os.path.dirname(__file__),"demo.mp4")

def socket_client():
    client = socket.socket()

    client.connect(('127.0.0.1',8848))

    #制作字典形式的报头

    head_dic = {
        'md5': 13212314114141313,
        'file_name':os.path.basename(FILE_PATH),
        'file_size':os.path.getsize(FILE_PATH),
        'new_file_name':'demo_new.mp4'
    }
    #获取json形式的报头
    head_dic_json = json.dumps(head_dic)

    #获取bytes形式的报头
    head_dic_json_bytes = head_dic_json.encode('utf-8')

    #获取bytes报头的总字节数
    head_len = len(head_dic_json_bytes)

    #将报头的总字节数转化为固定的四个字节
    four_bytes = struct.pack('i',head_len)

    #发送四个固定字节的报头
    client.send(four_bytes)

    #发送字典的bytes报头
    client.send(head_dic_json_bytes)

    #发送总数据
    with open(FILE_PATH,mode = 'rb') as f1:
        while 1:
            every_data = f1.read(1024)
            if every_data:
                client.send(every_data)
            else:
                break
    client.close()
socket_client()

server

import socket
import struct
import json
import os
MY_FILE = os.path.join(os.path.dirname(__file__),'my_file')
def socket_sever():
    sever = socket.socket()

    sever.bind(('127.0.0.1',8848))

    sever.listen(5)

    conn,addr = sever.accept()

    #接受固定长度的四个字节
    four_bytes = conn.recv(4)

    #利用struct反解
    head_len = struct.unpack('i',four_bytes)[0]

    #接受bytes类型的报头
    head_dic_json_bytes = conn.recv(head_len)

    #将bytes类型的报头转化成json
    head_dic_json = head_dic_json_bytes.decode('utf-8')

    #将json类型报头转化成字典形式的报头
    head_dic = json.loads(head_dic_json)

    #MD5校验

    #接受源数据
    with open(os.path.join(MY_FILE,head_dic["new_file_name"]),mode = 'wb') as f1:
        total_size =0
        every_data = ''
        while total_size < head_dic['file_size']:
            every_data = conn.recv(1024)
            f1.write(every_data)
            total_size +=len(every_data)
    conn.close()
    sever.close()
socket_sever()

6基于UDP协议的socket通信

udp协议:不可靠,相对来说不安全的协议,面向数据(无连接)的协议,速度快,原因是tcp需要建立连接,当数据传输完成后输出缓存区的数据暂时不会消失,等待回调函数传回信息后,再清空输出缓存区的数据,如图

socketserver

从源码解读

代码如下:可实现多线程的通信

udp_server

import socketserver


class MySever(socketserver.BaseRequestHandler):  # 继承的类固定

    def handle(self):  # 必须是handle名字

        while 1:
            from_client_data = self.request.recv(1024).decode('utf-8')#self.request相当于conn通道
            print(from_client_data)
            to_client_data = input('>>>').strip()
            self.request.send(to_client_data.encode('utf-8'))

if __name__ == '__main__':
    ip_port = ('127.0.0.1',8848)
    sever = socketserver.ThreadingTCPServer(ip_port,MySever)#基于tcp协议的多线程服务端
    sever.serve_forever()

udp_client

import socket

phone = socket.socket()

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

while 1:
    to_sever = input(">>>").strip()
    phone.send(f"小米:{to_sever}".encode('utf-8'))

    from_server_data = phone.recv(1024)
    print(f"来自服务器的消息:{from_server_data.decode('utf-8')}")