python 使用 io.BytesIO 内存文件加速图片生成服务

服务使用了 python3.7，sanic 框架，使用临时文件的大概代码如下：

import os
import sys
import logging
import traceback
import sanic
import asyncio
import matplotlib.pyplot as pltapp = sanic.Sanic('My color patch service', load_env='MY_COLORPATCH_')def _worker(request):data = request.pop('data')longitudes = [r[0] for r in data]latitudes = [r[1] for r in data]values = [r[2] for r in data]fig = plt.figure(figsize=(10, 10), dpi=90)'''draw image by matplotlib'''image_file = os.path.realpath('out1.png')fig.savefig(image_file, transparent=True)plt.close(fig)return image_filedef _worker_wrap(request):try:return _worker(request)except KeyboardInterrupt as e:raise e  # rethrow the Ctrl+C Exception, for terminate the worker processexcept:return traceback.format_exc(limit=5)@app.route('/gen_color_patches', {'POST'})
async def make_color_patch(request):loop = asyncio.get_event_loop()resp = await loop.run_in_executor(None, _worker_wrap, request.json)if os.path.exists(resp):return await sanic.response.file_stream(resp, mime_type='image/png')return sanic.response.json({'code': 1, 'msg': resp}, status=500)if __name__ == '__main__':logging.basicConfig(level=logging.INFO, format='%(asctime)s-%(levelname)s: %(message)s')address_bind = sys.argv[1] if len(sys.argv) > 1 else 'localhost'port = sys.argv[2] if len(sys.argv) > 2 else '8000'app.run(host=address_bind, port=int(port), workers=1)

因为生成临时文件需要IO时间，所以想省去写和读文件开销，所以改写如下：

def _worker(request):'''以上都未变'''image_file = io.BytesIO()fig.savefig(image_file, transparent=True)plt.close(fig)return image_fileasync def make_color_patch(request):loop = asyncio.get_event_loop()resp = await loop.run_in_executor(None, _worker_wrap, request.json)if isinstance(resp, str):return sanic.response.json({'code': 1, 'msg': resp}, status=500)return sanic.response.raw(resp.getbuffer(), content_type='image/png')

临时的测试脚本也一并改写，移去了下载文件后写临时文件再读文件的步骤：

import os
import io
import json
import requests
import matplotlib.pyplot as pltdata_file = r'D:\projects\预报数据.json'
with open(data_file, 'r', encoding='utf-8') as f:data = json.load(f)params = {'adcode': '450000','width': 1080,'height': 1000,'data': [(float(d['longitude']), float(d['latitude']), float(d['tmp'])) for d in data],
}
resp = requests.post('http://127.0.0.1:8000/gen_color_patches', json=params)
from PIL import Image
img = Image.open(io.BytesIO(resp.content))
plt.imshow(img)
plt.show()

BytesIO 作为一个 file-like object 实现，用起来确实很方便

python 使用 io.BytesIO 内存文件加速图片生成服务相关推荐

python图片保存为txt文件_python实现对文件中图片生成带标签的txt文件方法
在深度学习中经常需要生成带标签的图片名称列表,xxxlist.txt文件,下面写一个简单的python脚本生成该文件列表. import os def generate(dir,label): fil ...
Python爬虫之scrapy下载文件和图片
Python爬虫之scrapy下载文件和图片一:pipeline 二:使用scrapy下载图片 2.1 创建scrapy项目 1.2 创建爬虫 1.3 配置文件设置 1.4 编写启动文件 1.5 爬 ...
Python基础-IO编程-读文件read-写文件write
IO 编程初识 IO编程中,Stream(流)是一个很重要的概念,可以把流想象成一个水管,数据就是水管里的水,但是只能单向流动.Input Stream就是数据从外面(磁盘.网络)流进内存,Outpu ...
Python实例 -- 如何将.npy文件转换为图片
为了节省空间,有时会将.jpg文件转换为.npy文件,这样便于存储,然后需要图片的时候,再将其转为.jpg文件,这个过程是如何实现的呢?为了演示这个案例,我们分为2步. 第1步:将.jpg文件保存为. ...
Java以IO流和文件流，生成、下载二维码
场景在系统中生成一个二维码,用户保存下来并分享出去,其他人扫描之后跳到我们的一个活动详情页,查看此活动的内容. 从以上的需求中,可以提炼出以下几点: 当用户点击生成二维码的时候,我们要拿到用户生成的 ...
10 个优秀的占位图片生成服务
placehold.it A quick and simple image placeholder service. 特点:老牌服务,支持 HTTPS 可定制:尺寸.颜色.文字.格式 http://p ...
python读取txt文件写入-python 读取、写入txt文件的示例
写入文件使用open()函数和write()函数但是有两种写法,分别是'a'和'w' 'a' 表示写入文件若无该文件会直接创建一个如果存在这个文件,会接着已有的内容的后面写入 with ope ...
python中io.textio_Python文件读写概述（IO操作、文件读写、stringiobytesio、序列化）,python,的,小,总结,StringIOBytesIO...
IO操作在进行文件的读写之前,需要说明几点.首先,运行的程序和读取的数据都会在内存中缓存. 进入到程序或数据内存其次,用python程序进行文件的读写,需要创建一个小工具–文件流,用来处理数据 ...
python截图直接在内存里调用_Python-按块上传FTP中的内存文件(由API调用生...
我需要能够在Python中通过FTP和SFTP上传文件,但有一些不常见的约束. >绝对不能将文件写入磁盘. >通过调用API并将JSON响应写入文件来生成文件. >有多个对API的调 ...

python 使用 io.BytesIO 内存文件加速图片生成服务

python 使用 io.BytesIO 内存文件加速图片生成服务相关推荐

最新文章

热门文章