# -*- coding:utf-8 -*-
# @Author len
# @Create 2023/11/29 14:11

import qrcode

def create_qr_code(data, fill_color="black", back_color="white", size=10, location="myqr.png"):
    """
    创建并保存一个二维码图像。

    :param data: 二维码包含的数据或链接。
    :param fill_color: 二维码的颜色。
    :param back_color: 二维码的背景色。
    :param size: 二维码每个小格的像素大小。
    """
    qr = qrcode.QRCode(
        version=1,
        error_correction=qrcode.constants.ERROR_CORRECT_L,
        box_size=size,
        border=4,
    )
    qr.add_data(data)
    qr.make(fit=True)

    img = qr.make_image(fill_color=fill_color, back_color=back_color)
    img.save(location)

# 内容
data = '''智巡卫士'''
fill_color = "red"  # 颜色
back_color = "white"  # 底色
size = 10  # 尺寸
location = r"./data/2024_1015_世界大赛训练/red_智巡卫士.png"  # 地址
# 使用函数生成二维码
create_qr_code(data, fill_color=fill_color, back_color=back_color, size=10, location=location)