import json

from flask import Blueprint, session, jsonify

import pickle_idea2

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

# 创建api蓝图
api_blue = Blueprint('api', __name__, url_prefix='/api')


def helper(res, result):
    for item in res:
        if type(res[str(item)]) == 'dict':
            helper(res[str(item)], result)
        if type(res[str(item)]) == 'list':
            for i in range(len(res[str(item)])):
                helper(res[str(item)][i], result)
        result.append(str(item))
    return result


@api_blue.route('/json/<username>', methods=['GET'])
def api_bp(username):
    # 获取session里的用户名,必须携带token
    token = session.get("token")
    if token == "70620F32A9DC965FCCF0447B674AA161":
        result = []
        user_freq_record = path_prefix + 'static/frequency/' + 'frequency_%s.pickle' % (username)
        s = pickle_idea2.load_record(user_freq_record)
        wordlist = helper(s, result)
        print(json.dumps(s))
        results = {}

        for word in wordlist:
            results[word] = len(s[word])

        return jsonify(results)

    else:
        print("无效的token")
        return jsonify({"error": "无效的token"})