一、SOCKET套接字传送图片
1.1.1 服务端
import socket
import os
import sys
import struct
def socket_service_image():
try:
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
s.bind(('192.168.9.26', 8899))
s.listen(10)
except socket.error as msg:
print(msg)
sys.exit(1)
print('Wat for Connection……………………')
while True:
sock, addr = s.accept() # addr是一个元组(ip, port)
deal_image(sock, addr)
def deal_image(sock, addr):
print('Accept connection from {0}'.format(addr)) # 查看发送端的ip和端口
while True:
fileinfo_size = struct.calcsize('128sq')
buf = sock.recv(fileinfo_size) # 接收图片名
if buf:
filename, filesize = struct.unpack('128sq', buf)
fn = filename.decode().strip('\x00')
new_filename = os.path.join(
'./',
'new_' + 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) # 写入图片数据
sock.close()
break
if __name__ == '__main__':
socket_service_image()
1.1.2 客户端
import socket
import os
import sys
import struct
def sock_client_image():
while True:
try:
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(('192.168.9.195', 6666))
except socket.error as msg:
print(msg)
print(sys.exit(1))
filepath = input('input the file:')
fhead = struct.pack(
b'128sq',
bytes(
os.path.basename(filepath),
encoding='utf-8'
),
os.stat(filepath).st_size
)
s.send(fhead)
fp = open(filepath, 'rb') # 打开要传输的图片
while True:
data = fp.read(1024) # 读入图片数据
if not data:
print('{0} send over ……'.format(filepath))
break
s.send(data) # 以二进制格式发送图片数据
s.close()
if __name__ == '__main__':
sock_client_image()
二、struct内置库以及它的作用
struct是Python的一个内置库,用于处理字节数据和结构化数据的转换。用于将字节数据按照特定的格式解析为结构化的数据或者将结构化的数据打包为字节数据。
在图片上传的过程中,struch库可以帮助我们处理图片的二进制数据。但我们从上传的图片文件中读取二进制数据时,通常是以字节数组的形式存储的。使用struct库,我们可以将这些字节数据解析为图像文件的头部信息,以及像素数据。