python 内存中读写文件流

python 内存中读写文件流

作用：避免磁盘io，性能上去了

demo 图片旋转

from typing import Union
from PIL import Image
from io import BytesIO

# BytesIO 可以开辟一段内存空间用来对Bytes类型进行IO操作
def img_rotate(f_b: bytes, angle:Union[int,float], endstuff="PNG") ->bytes:

  """

  图片旋转

  """

  f=BytesIO() # 创建对象

  f.write(f_b) # 写入文件流

  f2 = BytesIO()

  img = Image.open(f)

  if angle >= 0 and angle<360:

​    angle = 360-angle

  elif angle <0 and angle>-360:

​    angle = abs(angle)

  dst1 = img.rotate(angle, resample=Image.BICUBIC, expand=True)

  dst1.save(f2,format=endstuff)

  img_res = f2.getvalue() # 读取文件流

  f.close() #  关闭对象

  f2.close()

  return img_res