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


import json
from flask import Flask
from flask import request
from flask import Response
from flask import make_response
from flask import render_template
from flask import send_from_directory
from bin.b64img import re_sort_number_image

app = Flask(__name__)


def is_new_user(name: str) -> str:
    """
    判断填入的参数是否为已有本地已有
    如果有则该名称值加一
    如果没有则新建一个
    :param name:
    :return:
    """
    with open('./static/data.json', 'r', encoding='utf-8') as fp:
        try:
            data = json.load(fp)
        except json.decoder.JSONDecodeError:
            with open('./static/data.json', 'w', encoding='utf-8') as init_fp:
                json.dump({}, init_fp)
        all_user = data.keys()
        try:
            if name in all_user:  # 加次数
                data[name] += 1
                return str(data[name])

            else:
                data[name] = 0  # 加用户
                return str(0)
        finally:
            with open('./static/data.json', 'w', encoding='utf-8') as dump:
                json.dump(data, dump)


def build_page(name: str) -> str:
    """
    渲染最终的页面
    :param name:
    :return:
    """
    count = is_new_user(name)

    if count == 0:  # 新id生成十个0
        return '0000000000'
    else:
        if len(str(count)) <= 10:
            zero_counter = 10 - len(str(count))
            final_number = '0' * zero_counter + str(count)
            sorted_image = re_sort_number_image(final_number)
            return render_template('index.html',
                                   title=name,
                                   svg_img_0=f'{sorted_image[0]}',
                                   svg_img_1=f'{sorted_image[1]}',
                                   svg_img_2=f'{sorted_image[2]}',
                                   svg_img_3=f'{sorted_image[3]}',
                                   svg_img_4=f'{sorted_image[4]}',
                                   svg_img_5=f'{sorted_image[5]}',
                                   svg_img_6=f'{sorted_image[6]}',
                                   svg_img_7=f'{sorted_image[7]}',
                                   svg_img_8=f'{sorted_image[8]}',
                                   svg_img_9=f'{sorted_image[9]}'
                                   )


def arg_not_be_full() -> Response:
    """
    参数不完整
    :return:
    """
    response = make_response('<h2 style="text-align: center">参数填写不完整 -> owner</h2>')
    response.status_code = 200
    return response


@app.route('/API', methods=['GET', 'POST'])
def api_page() -> Response or str:
    """
     API页面主函数
    :return:
    """
    if 'owner' in request.args.keys():
        owner = request.args['owner']
        if owner != '':
            html_template = build_page(owner)
            response = make_response(html_template)
            response.headers['Content-Type'] = 'image/svg+xml'
            # response.headers['Content-Type'] = 'text/html'
            return response
        else:
            return arg_not_be_full()
    else:
        arg_not_be_full()


@app.route('/', methods=['GET', 'POST'])
def index():
    """
    索引页面
    :return:
    """
    return send_from_directory('./static', 'index.html')


if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000, threaded=True)