#! python3
# -*- coding: utf-8 -*-
from flask import Flask, jsonify
from flask_restful import reqparse, Api, Resource
from elasticsearch import Elasticsearch


es = Elasticsearch([{'host': 'localhost', 'port': 9200}])

app = Flask(__name__)
api = Api(app)

parser = reqparse.RequestParser()
parser.add_argument('query', required=False)
parser.add_argument('pageSize', type=int, default=10, required=False)
parser.add_argument('currentPage', type=int, default=1, required=False)


class Search(Resource):
    def get(self):
        """
        发送get请求去查询els引擎并返回结果
        :param query: 查询字段
        :param size: 单页结果数
        :param offset: 从哪个结果开始
        :return:
        """
        args = parser.parse_args()
        query = args['query']  # 查询字段
        pageSize = args['pageSize']  # 一页有多少结果，默认是5
        currentPage = args['currentPage']  # 当前的页数,搜索结果从哪里开始，默认是0
        if not query:
            return {'list': [], 'pagination': {'total': 0,
                                               'pageSize': 5,
                                               'current': 0}}, 200
        search_res = es.search(index='frommysql', body={'query': {'multi_match': {'query': query,
                                                                                  'fields': ['title', 'content']}},
                                                        "highlight": {"fields": {"title": {}, 'content': {}}}},
                                                   size=pageSize, from_=currentPage)
        total_els_res_length = int(search_res['_shards']['total'])  # els搜索返回的结果有多少条
        if total_els_res_length == 0:
            return {'list': [], 'pagination': {'total': 0,
                                               'pageSize': 5,
                                               'current': 0}}, 200
        else:
            # 如果返回的是正常内容，要对els的数据重新组织一下结构
            hit_res = search_res['hits']['hits']
            search_list = []
            for item in hit_res:
                _source = item['_source']
                object_id = _source['object_id']
                title = _source['title']
                content = _source['content']
                binary_sentiment = _source['binary_sentiment']
                source_name = _source['source']
                source_href = _source['url']
                updated_at = _source['occur_time']
                keywords = _source['keyword']
                tag_list = []
                if keywords:
                    keyword_list = keywords.split(' ')
                    for keyword_index, keyword_item in enumerate(keyword_list):
                        tag_list_item = {'id': 'tag-%s' % keyword_index, 'name': keyword_item}
                        tag_list.append(tag_list_item)
                search_dict = {'id': object_id, 'negative': binary_sentiment, 'title': title, 'sourceName': source_name,
                               'sourceHref': source_href, 'content': content, 'updatedAt': updated_at, 'tagList': tag_list}
                search_list.append(search_dict)
            front_receive_data = {'list': search_list, 'pagination': {'total': total_els_res_length,
                                                                      'pageSize': pageSize,
                                                                      'current': currentPage}}
            return front_receive_data, 200


class SearchHotNews(Resource):
    def get(self):
        """
        利用q普通查询
        搜索热门新闻
        :return:
        """
        keywords = [{'keyword': '区块链'}, {'keyword': '概念股'}, {'keyword': '人工智能'}, {'keyword': 'Google'}, {'keyword': '量子计算机'}, {'keyword': '无人车'},
                    {'keyword': '无人超市'}, {'keyword': 'SpaceX'}, {'keyword': '量子通信'}, {'keyword': '比特币'}]
        return jsonify(keywords)


# api.add_resource(SearchList, '/searchlist')  # 返回结果列表
api.add_resource(Search, '/api/search/query')
api.add_resource(SearchHotNews, '/api/search/hot')


if __name__ == '__main__':
    app.run(host='0.0.0.0', port=9001, debug=True)

#  api.add_resource(SearchHotNews, '/hot')
# 'http://localhost:5000/hot?query=google&size=10&offset=0'  # 启动服务后, 在网址中发送get请求
# 启动0.0.0.0端口后，通过ipconfig查看ipv4地址，然后去访问192.168.1.191:1080即可