from flask import Flask, request, Blueprint, render_template, json, jsonify
from UseSqlite import InsertQuery, RecordQuery
import pickle
import difficulty
import pickle_idea2
from Article import load_freq_history
from app import pickle_idea
from app.wordfreqCMD import sort_in_descending_order

path_prefix = '/var/www/wordfreq/wordfreq/'
path_prefix = '../'  # comment this line in deployment

TKTK = 'token' # set token

show_bp = Blueprint(
    'site',
    __name__,
)


@show_bp.route('/show/<name>/')  # set route for show page <name> means the var name to search
def show(name):
    token = request.args.get("token")
    # when token is wrong
    if token != TKTK:
        return "token is wrong, please try again"
    user_freq_record = path_prefix + 'static/frequency/' + 'frequency_%s.pickle' % (name)
    d = load_freq_history(user_freq_record)
    freqlst = sort_in_descending_order(pickle_idea.dict2lst(d))
    print(freqlst)
    words_freq=[] # 存储单词表的数组，格式为 单词-词频
    for i in range(len(freqlst)):
        words_freq.append(str(freqlst[i][0])+"-"+str(len(freqlst[i][1])))
    t={}
    t[name]=words_freq
    return jsonify(t)