| OLD | NEW |
| (Empty) |
| 1 # coding: utf-8 | |
| 2 | |
| 3 # This file is part of the Adblock Plus web scripts, | |
| 4 # Copyright (C) 2006-2016 Eyeo GmbH | |
| 5 # | |
| 6 # Adblock Plus is free software: you can redistribute it and/or modify | |
| 7 # it under the terms of the GNU General Public License version 3 as | |
| 8 # published by the Free Software Foundation. | |
| 9 # | |
| 10 # Adblock Plus is distributed in the hope that it will be useful, | |
| 11 # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 13 # GNU General Public License for more details. | |
| 14 # | |
| 15 # You should have received a copy of the GNU General Public License | |
| 16 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | |
| 17 | |
| 18 import json | |
| 19 import os | |
| 20 import traceback | |
| 21 from urlparse import parse_qsl | |
| 22 | |
| 23 import MySQLdb | |
| 24 | |
| 25 from sitescripts.web import url_handler | |
| 26 from sitescripts.utils import cached, setupStderr | |
| 27 from sitescripts.filterhits import db | |
| 28 from sitescripts.filterhits.web import common | |
| 29 | |
| 30 | |
| 31 def query(domain=None, filter=None, skip=0, take=20, order="DESC", order_by="fre
quency", **_): | |
| 32 """ | |
| 33 Returns the SQL and parameters needed to perform a query of the filterhits d
ata. | |
| 34 """ | |
| 35 sql = """SELECT SQL_CALC_FOUND_ROWS domain, filter, frequency | |
| 36 FROM frequencies as freq | |
| 37 LEFT JOIN filters as f ON f.sha1=freq.filter_sha1 | |
| 38 %s | |
| 39 ORDER BY %s | |
| 40 LIMIT %%s, %%s""" | |
| 41 | |
| 42 where = zip(*[("%s LIKE %%s" % s, "%%%s%%" % p) for s, p in (("domain", doma
in), | |
| 43 ("filter", filt
er)) if p]) | |
| 44 if where: | |
| 45 where_fields, params = where | |
| 46 where_sql = "WHERE " + " AND ".join(where_fields) | |
| 47 else: | |
| 48 where_sql = "" | |
| 49 params = [] | |
| 50 | |
| 51 order = order.upper() if order.upper() in ("ASC", "DESC") else "ASC" | |
| 52 if order_by not in ["filter", "domain", "frequency"]: | |
| 53 order_by = "frequency" | |
| 54 order_by_sql = "`%s` %s" % (order_by, order) | |
| 55 | |
| 56 params = list(params) + [int(skip), int(take)] | |
| 57 return [sql % (where_sql, order_by_sql)] + params | |
| 58 | |
| 59 | |
| 60 @url_handler("/query") | |
| 61 def query_handler(environ, start_response): | |
| 62 setupStderr(environ["wsgi.errors"]) | |
| 63 params = dict(parse_qsl(environ.get("QUERY_STRING", ""))) | |
| 64 | |
| 65 try: | |
| 66 db_connection = db.connect() | |
| 67 try: | |
| 68 results = db.query(db_connection, *query(**params), dict_result=True
) | |
| 69 total = db.query(db_connection, "SELECT FOUND_ROWS()")[0][0] | |
| 70 finally: | |
| 71 db_connection.close() | |
| 72 except MySQLdb.Error: | |
| 73 traceback.print_exc() | |
| 74 return common.show_error("Failed to query database!", start_response, | |
| 75 "500 Database error") | |
| 76 | |
| 77 try: | |
| 78 echo = int(params["echo"]) | |
| 79 except (ValueError, KeyError): | |
| 80 echo = 0 | |
| 81 | |
| 82 response_headers = [("Content-type", "application/json; charset=utf-8")] | |
| 83 start_response("200 OK", response_headers) | |
| 84 return [json.dumps({"results": results, "echo": echo, | |
| 85 "total": total, "count": len(results)}, | |
| 86 ensure_ascii=False).encode("utf-8")] | |
| OLD | NEW |