diff --git a/app/api_bp.py b/app/api_bp.py new file mode 100644 index 0000000..311af02 --- /dev/null +++ b/app/api_bp.py @@ -0,0 +1,44 @@ +import json +from flask import Blueprint +from wordfreqCMD import * +from Article import get_today_article, load_freq_history +import pickle_idea +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/', methods=['GET']) +def api_bp(username): + # 获取session里的用户名 + result = [] + user_freq_record = path_prefix + 'static/frequency/' + 'frequency_%s.pickle' % (username) + s = pickle_idea2.load_record(user_freq_record) + wordlist = helper(s,result) + results = "{" + i=0 + for word in wordlist: + if i!=len(wordlist)-1: + results +='"' + word +'":' + str(len(s[word])) +"," + i=i+1 + else: + results += '"' + word + '":' + str(len(s[word])) + results+="}" + + + return results diff --git a/app/main.py b/app/main.py index d903bf4..2a667e0 100644 --- a/app/main.py +++ b/app/main.py @@ -11,12 +11,14 @@ from Article import * import Yaml from user_service import userService from account_service import accountService +from api_bp import api_blue app = Flask(__name__) app.secret_key = 'lunch.time!' # 将蓝图注册到Lab app app.register_blueprint(userService) app.register_blueprint(accountService) +app.register_blueprint(api_blue) path_prefix = '/var/www/wordfreq/wordfreq/' path_prefix = './' # comment this line in deployment