MoeCounter/src/utils/response.py

#!/usr/bin/env python3
# -- coding:utf-8 --
# @Author: markushammered@gmail.com
# @Development Tool: PyCharm
# @Create Time: 2022/8/30
# @File Name: response.py


from src.db import Database


async def resp(_id: str, length: int = 7, theme: str = 'lewd') -> dict:
    """
    generate dict response
    include all information
    :param _id: name
    :param length:  length for counter
    :param theme:  theme for counter
    :return:
    """
    result = await Database().query(_id)
    times = result[1]
    str_number = str(times)  # 将整形转换为字符串
    len_number = len(str_number)  # 再获取字符串长度
    g_length = length * '0'  # 根据输入的位数来生成0的数量
    show_number = str(g_length[:-len_number] + str_number)
    context = []
    headers = {'cache-control': 'max-age=0, no-cache, no-store, must-revalidate',
               'Content-Type': 'image/svg+xml; charset=utf-8'}
    data = await Database().query_image(theme)
    height = data[0][-1]
    width = data[0][-2]
    counter = 0
    for i, n in zip(data, show_number):
        context.append({
            'position': i[-2] * counter,
            'width': i[-2],
            'height': i[-1],
            'base64': data[int(n)][1]
        })
        counter += 1

    return {
        'context': context,
        'g_width': length * width,
        'g_height': height,
        'headers': headers
    }