树莓派利用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))