diff --git a/app/main.py b/app/main.py index 0c747a3..19e82f2 100644 --- a/app/main.py +++ b/app/main.py @@ -12,6 +12,7 @@ from account_service import accountService from admin_service import adminService, ADMIN_NAME import os from translate import * +from wordCMD import show_bp app = Flask(__name__) app.secret_key = os.urandom(32) @@ -20,6 +21,7 @@ app.secret_key = os.urandom(32) app.register_blueprint(userService) app.register_blueprint(accountService) app.register_blueprint(adminService) +app.register_blueprint(show_bp) path_prefix = '/var/www/wordfreq/wordfreq/' path_prefix = './' # comment this line in deployment diff --git a/app/wordCMD.py b/app/wordCMD.py new file mode 100644 index 0000000..dd579da --- /dev/null +++ b/app/wordCMD.py @@ -0,0 +1,39 @@ +from flask import * +from flask_httpauth import HTTPTokenAuth +from Article import load_freq_history +from wordfreqCMD import sort_in_descending_order +import pickle_idea + +auth = HTTPTokenAuth(scheme='Bearer') + +path_prefix = '/var/www/wordfreq/wordfreq/' +path_prefix = './' # comment this line in deployment + +show_bp = Blueprint( + 'site', + __name__, +) + +tokens = { + "token": "token" +} + + +@auth.verify_token +def verify_token(token): + if token in tokens: + return tokens[token] + + +@show_bp.route('/show//') # set route for show page means the var name to search +@auth.login_required +def show(name): + 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)) + 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)