ai_qback/make_docx_demo/docx_other_func.py

# -*- coding: utf-8 -*-
import datetime
import io
import time
from base64 import b64decode
from functools import wraps

import matplotlib.pyplot as plt
import qrcode
from docx.shared import RGBColor

plt.switch_backend('Agg')
from io import BytesIO
from tools.loglog import logger, log_err_e
from threading import Lock
from config.read_config import address

lock = Lock()

width_cm, height_cm = 5.4, 3
width_in = width_cm
height_in = height_cm

plt.figure(figsize=(width_in, height_in))


def hex_to_rgb(hex_color: str):
    hex_color = hex_color.lstrip('#')
    return RGBColor(int(hex_color[0:2], 16), int(hex_color[2:4], 16), int(hex_color[4:6], 16))


def rgb_to_hex(r, g, b):
    return '{:02x}{:02x}{:02x}'.format(r, g, b)


def is_base64(text):
    try:

        image_bytes = b64decode(text)
        return image_bytes
    except Exception:

        return False


def time_use(fn):
    @wraps(fn)
    def cc(*args, **kwargs):
        f_time = time.time()
        res = fn(*args, **kwargs)

        cha = round(time.time() - f_time, 3)
        if cha > 0.3:
            print(f'函数:{fn.__name__} 一共用时', cha, '秒')
        return res

    return cc


def qrcode_maker(id_text=None, full_url=None) -> BytesIO:
    """
    :param id_text: id_text 提供id，二维码地址是春笋筛查表的地址；http://dcjxb.yunzhixue.cn/link?type=scanpage&id=999；
    :param full_url: 如果提供，直接使用这个文本来生成二维码的地址
    :return: 二维码图的IO对象，记得关闭
    """
    text = f"{address}/link?type=scanpage&id={id_text}"
    if full_url:
        text = full_url
    qr = qrcode.QRCode(
        version=1,
        error_correction=qrcode.constants.ERROR_CORRECT_L,
        box_size=12,
        border=4,
    )
    qr.add_data(text)
    qr.make(fit=True)

    img = qr.make_image(fill_color="black", back_color="white")
    img_byte_arr = io.BytesIO()
    img.save(img_byte_arr, format='PNG')
    img_byte_arr.seek(0)

    return img_byte_arr


def get_weekday():
    today = datetime.date.today()
    weekday_index = today.weekday()
    weekdays_chinese = ['星期一', '星期二', '星期三', '星期四', '星期五', '星期六', '星期日']
    weekday_chinese = weekdays_chinese[weekday_index]
    return weekday_chinese


def make_chart(x_axis_data, y_axis_datas, title, sub_title_list, x_axis_label=None, y_axis_label=None):
    """
    :param sub_title_list: 小标题集合，放在右上角，用来标记每个y轴的数据标题
    :param y_axis_label:Y轴文本
    :param x_axis_label:X轴文本
    :param title:图标标题
    :param x_axis_data: X数据列表,单列表数据，一般就日期。
    :param y_axis_datas: Y数据列表[合集],是个大列表。因为可能会有多个y线
    :return:
    """
    x_len = len(x_axis_data)

    image_io = BytesIO()

    font1 = {'family': 'SimSun', 'weight': 'normal', 'size': 14}
    plt.rc('font', **font1)
    plt.rcParams["axes.unicode_minus"] = False
    try:
        for y in y_axis_datas:
            if len(y) != x_len:
                logger.error("x轴的y轴的数据个数不一致")
            plt.plot(x_axis_data, y, marker='o', label="zxs")

        plt.title(title)
        if x_axis_label:
            plt.xlabel(x_axis_label)
        if y_axis_label:
            plt.ylabel(y_axis_label)
        plt.grid(True)

        for index, sub_title in enumerate(sub_title_list):
            plt.text(0.95, 0.9 - index * 0.15, sub_title, transform=plt.gca().transAxes, fontsize=10, va='top', ha='right', backgroundcolor='w')
        with lock:
            plt.savefig(image_io, format='png', bbox_inches='tight')
            image_io.seek(0)

        return image_io
    except Exception as e:
        log_err_e(e, "折线图生成错误")
        image_io.close()
        return None


if __name__ == '__main__':
    t = time.time()
    io = qrcode_maker('', "http://111.231.167.191:8001/mp3")
    with open("1.jpg", 'wb') as f:
        f.write(io.read())
    print(time.time() - t)