diff --git a/app/api_bp.py b/app/api_bp.py index 45cd5ea..59e2113 100644 --- a/app/api_bp.py +++ b/app/api_bp.py @@ -1,8 +1,7 @@ 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/' @@ -16,7 +15,7 @@ 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': + if type(res[str(item)]) == 'list': for i in range(len(res[str(item)])): helper(res[str(item)][i], result) result.append(str(item)) @@ -29,11 +28,11 @@ def api_bp(username): result = [] user_freq_record = path_prefix + 'static/frequency/' + 'frequency_%s.pickle' % (username) s = pickle_idea2.load_record(user_freq_record) - wordlist = helper(s,result) + wordlist = helper(s, result) print(json.dumps(s)) results = {} for word in wordlist: results[word] = len(s[word]) - return results \ No newline at end of file + return results