kodo/utils/redis/rimage.py

# -*- coding: utf-8 -*-

import base64

from utils.redis.connect import r
from utils.redis.rkeys import IMAGES_B64_DATA


def set_image_data(file, file_path=None, file_url=None):
    if not file:
        return
    file_name = file_path.split('/')[-1]
    file.seek(0)
    data = file.read()
    data = base64.b64encode(data).decode('utf-8')
    r.hset(IMAGES_B64_DATA, file_name, data)


def get_images_data(file_names):
    file_b64strs = r.hmget(IMAGES_B64_DATA, file_names)
    return [{
        'FileBody': v,
        'FileName': k
    } for k, v in zip(file_names, file_b64strs) if v]