欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

树莓派利用socket传图到pc端(Python)

程序员文章站 2022-04-24 11:13:27
...

1.PC作为socket的服务器端:

import socket
import os
import sys
import struct


def socket_service():
    try:
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        #IP地址留空默认是本机IP地址
        s.bind(('', 8088))
        s.listen(7)
    except socket.error as msg:
        print(msg)
        sys.exit(1)

    print("连接开启,等待传图...")
	
    while True:
        sock, addr = s.accept()
        deal_data(sock, addr)

    s.close()


def deal_data(sock, addr):
    print("成功连接上 {0}".format(addr))

    while True:
        fileinfo_size = struct.calcsize('128sl')
        buf = sock.recv(fileinfo_size)
        if buf:
            filename, filesize = struct.unpack('128sl', buf)
            fn = filename.decode().strip('\x00')
            #PC端图片保存路径
            new_filename = os.path.join('C:/Users/user/Desktop', fn)

            recvd_size = 0
            fp = open(new_filename, 'wb')

            while not recvd_size == filesize:
                if filesize - recvd_size > 1024:
                    data = sock.recv(1024)
                    recvd_size += len(data)
                else:
                    data = sock.recv(1024)
                    recvd_size = filesize
                fp.write(data)
            fp.close()
        sock.close()
        break


if __name__ == '__main__':
    socket_service()

2.树莓派作为socket的客户端:

import socket
import os
import sys
import struct


def sock_client():
    try:
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        #192.168.199.1和8088分别为服务端(pc)的IP地址和网络端口
        s.connect(('192.168.199.1', 8088))
    except socket.error as msg:
        print(msg)
        print(sys.exit(1))

    while True:
        #filepath是要被发送图片的路径
        filepath = '/home/pi/ceshi/image.jpg'
        fhead = struct.pack(b'128sl', bytes(os.path.basename(filepath), encoding='utf-8'), os.stat(filepath).st_size)
        s.send(fhead)
        print('client filepath: {0}'.format(filepath))

        fp = open(filepath, 'rb')
        while 1:
            data = fp.read(1024)
            if not data:
                print('{0} 发送成功...'.format(filepath))
                break
            s.send(data)
        s.close()
        break

if __name__ == '__main__':
    sock_client()

socket连接,如果连接不成功直接退出程序sys.exit(1):

try:
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    #192.168.199.1和8088分别为服务端(pc)的IP地址和网络端口
    s.connect(('192.168.199.1', 8088))
except socket.error as msg:
    print(msg)
    print(sys.exit(1))

 

相关标签: 树莓派