index.py

# coding:utf8
import json
from time import time
from datetime import datetime
from search import Search
from settings import handler, DEBUG
from util import gen_pages, pretty_date
from flask import Flask, request, render_template, jsonify

app = Flask(__name__)
app.debug = DEBUG
app.logger.addHandler(handler)


@app.route("/", methods=['GET'])
@app.route("/api", methods=['GET'])
def index():

    try:
        _from = int(request.args.get('from', '0'))
        limit = int(request.args.get('limit', '10'))
        q = unicode(request.args.get('q', ''))
        s = request.args.get('s', 'sumup')
        raw = bool(request.args.get('raw', ''))

    except Exception, e:
        result = {
            "status_code": 400,
            "error": u"(Ｔ▽Ｔ) params error "
        }
        return jsonify(**result)

    if len(q) == 0 and request.path == '/':
        return render_template("index.html")

    #: build search
    search = Search(index='v2', doc_type='topic')
    search['body']['query'] = \
        {
            "multi_match":
                {
                    "query": "%s" % q,
                    "fields": ["title", "content", "rcontent"]
                }
        }
    search['size'] = limit
    search['from_'] = _from
    if request.path == '/api':
        #: field exclude for api
        search['_source_exclude'] = ['content_rendered', 'rcontent']

    #: choose a sort method from ['sumup','replies','created','match']
    if s == 'match':
        search['sort'] = '_score'

    if s in ["replies", "created"]:
        search['sort'] = "%s:desc" % s

    if s == 'sumup':
        if search['sort']:
            del search['sort']
        search['body']['sort'] = {
            "_script": {
                "script": "(doc['created'].value-1272124800000) * \
                    log10(doc['replies'].value+1)* log10(doc.score)",
                "type": "number",
                "params": {
                    "factor": 0
                },
                "order": "desc"
            }
        }

    #: do search in es
    try:
        time0 = time()
        result = search.exe()
        time1 = time()
    except Exception, e:
        msg = str(datetime.now()) + '\n' + \
            unicode(request.url) + '\n' + str(e) + '\r\n'
        app.logger.error(msg)
        result = {
            "status_code": 500,
            "error": "Sorry,server error T_T"
        }
        return jsonify(**result)

    # [return] raw json
    if raw and app.debug:
        return jsonify(**result)

    # [return] api request
    if request.path == '/api':
        # remove some keys from result
        del_key = ['_shards', 'timed_out', 'took']
        for k in del_key:
            if k in result:
                del result[k]
        result['cost_ms'] = int((time1 - time0) * 1000)

        return jsonify(**result)

    # data for template
    total = result['hits']['total']
    current = _from / 10
    max_page = total / 10
    pages = gen_pages(current, max_page)

    sort_by = {
        "sumup": u"综合",
        "match": u"精确匹配",
        "created": u"创建时间",
        "replies": u"回复数"
        }

    return render_template(
        'result.html', res=result, pages=pages,
        current=current, q=q, s=s, cost=time1-time0, pretty_date=pretty_date,
        enumerate=enumerate, int=int, sort_by=sort_by, route="")


if __name__ == "__main__":
    app.run(host="0.0.0.0", port=8000)