import random
import qrcode
import os


# 生成二维码图片
def gen_qr_code(data):
    qr = qrcode.QRCode(
        version=1,
        error_correction=qrcode.constants.ERROR_CORRECT_L,
        box_size=10,
        border=4,
    )
    qr.add_data(data)
    qr.make(fit=True)
    return qr.make_image(fill_color="black", back_color="white")


def gen_file_url(file, url):
    file_name = str(random.randint(10000, 99999))
    # file_name = file.filename
    file_ext = os.path.splitext(file.filename)[1]
    path = 'uploads/'
    if file_ext == '.json':
        path += 'json/' + file_name + file_ext
        url += "json/" + file_name
    elif file_ext == '.html':
        path += 'html/' + file_name + file_ext
        url += "html/" + file_name
    else:
        path += file_name + file_ext
    file.save(path)
    return url