hot-words/app.py

import json
import time
import traceback
from gevent import monkey
monkey.patch_all()

from flask import Flask, request
from log import Log
from config import set_config
from words_func import get_words, update_wechat_score_data

app = Flask(__name__)
log_ = Log()
config_ = set_config()


@app.route('/healthcheck')
def health_check():
    return 'ok!'


# 获取热点词库中的所有词
@app.route('/hot/word/getAllWords', methods=['GET', 'POST'])
def get_all_words():
    try:
        start_time = time.time()
        request_data = json.loads(request.get_data())
        page_num = request_data.get('pageNum', 1)
        page_size = request_data.get('pageSize', 100)
        words = get_words(page_num=page_num, page_size=page_size)
        result = {'code': 200, 'message': 'success', 'data': {'words': words}}
        log_message = {
            'requestUri': '/hot/word/getAllWords',
            'logTimestamp': int(time.time() * 1000),
            'result': result,
            'executeTime': (time.time() - start_time) * 1000
        }
        log_.info(log_message)
        return json.dumps(result)
    except Exception as e:
        log_.error(traceback.format_exc())
        result = {'code': -1, 'message': 'fail'}
        return json.dumps(result)


# 更新热点词的微信指数
@app.route('/hot/word/updateWechatScore', methods=['GET', 'POST'])
def update_wechat_score():
    try:
        start_time = time.time()
        request_data = json.loads(request.get_data())
        data = request_data.get('data', None)
        update_wechat_score_data(data=data)
        result = {'code': 200, 'message': 'success'}
        log_.info(f"executeTime = {(time.time() - start_time) * 1000}")
        return json.dumps(result)
    except Exception as e:
        log_.error(traceback.format_exc())
        result = {'code': -1, 'message': 'fail'}
        return json.dumps(result)


if __name__ == '__main__':
    app.run()