Archived
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -- coding:utf-8 --
|
||
|
|
# @Author: markushammered@gmail.com
|
||
|
|
# @Development Tool: PyCharm
|
||
|
|
# @Create Time: 2022/3/25
|
||
|
|
# @File Name: views.py
|
||
|
|
|
||
|
|
|
||
|
|
from flask import abort
|
||
|
|
from flask import jsonify
|
||
|
|
from flask import request
|
||
|
|
from flask import Response
|
||
|
|
from flask import make_response
|
||
|
|
from flask import render_template
|
||
|
|
from . import main
|
||
|
|
from ..db.db import SQLite as db
|
||
|
|
from ..utils.view import view
|
||
|
|
|
||
|
|
|
||
|
|
@main.route('/', methods=['GET', 'POST'])
|
||
|
|
def index():
|
||
|
|
"""
|
||
|
|
主页
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
return render_template('index.html', remote_address=request.remote_addr)
|
||
|
|
|
||
|
|
|
||
|
|
@main.route('/api/v1/<string:name>', methods=['GET', 'POST'])
|
||
|
|
@main.route('/count/<string:name>', methods=['GET', 'POST'])
|
||
|
|
def api_v1(name: str):
|
||
|
|
length = request.args.get('length', type=int)
|
||
|
|
theme = request.args.get('theme', type=str)
|
||
|
|
if not bool(length):
|
||
|
|
length = 7
|
||
|
|
if not bool(theme):
|
||
|
|
theme = 'lewd'
|
||
|
|
if not db().exists_name(name):
|
||
|
|
db().insert(name)
|
||
|
|
if not db().exists_table(theme):
|
||
|
|
abort(500)
|
||
|
|
if 7 <= length <= 10:
|
||
|
|
data = db().fetch(name)
|
||
|
|
number = '0' * (length - len(str(data[-1]))) + str(data[-1])
|
||
|
|
# 将设置的长度减去数据库内已有的数据库字符串长度剩下的结果为[0]的个数
|
||
|
|
# 再将这个字符串后拼接上数据库内已有次数的字符串, 就可以得到最终的结果
|
||
|
|
response = make_response(view(theme, int(length), name, number))
|
||
|
|
response.headers['Content-Type'] = 'image/svg+xml; charset=utf-8'
|
||
|
|
response.headers['cache-control'] = 'max-age=0, no-cache, no-store, must-revalidate'
|
||
|
|
return response
|
||
|
|
return abort(500)
|
||
|
|
|
||
|
|
|
||
|
|
@main.route('/exists-table', methods=['GET', 'POST'])
|
||
|
|
@main.route('/exists', methods=['GET', 'POST'])
|
||
|
|
def show_tables() -> Response:
|
||
|
|
"""
|
||
|
|
返回已有表
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
tables = db().show_tables
|
||
|
|
return jsonify({'code': 200, 'msg': '已有表', 'data': [tables]})
|