Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Side by Side Diff: sitescripts/filterhits/db.py

Issue 4615801646612480: Issue 395 - Filter hits statistics backend (Closed)
Patch Set: Addressed Sebastian's and Wladimir's comments. Created March 27, 2015, 3:08 p.m.
Left:
Right:
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 # coding: utf-8
2
3 # This file is part of the Adblock Plus web scripts,
4 # Copyright (C) 2006-2015 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 itertools
19 import MySQLdb
20
21 from sitescripts.utils import get_config
22
23 testing = False
24
25 def connect():
26 config = get_config()
27 return MySQLdb.connect(
28 user=config.get("filterhitstats", "dbuser"),
29 passwd=config.get("filterhitstats", "dbpassword"),
30 db=config.get("filterhitstats", "test_database" if testing else "database"),
31 use_unicode=True, charset="utf8"
32 )
33
34 def query(db, sql, *params, **kwargs):
35 """
36 Executes the query given by the provided SQL and returns the results.
37 If dict_result keyword argument is provided + True the results will be
38 returned as a tuple of dictionaries, otherwise a tuple of tuples.
39 """
40 if kwargs.pop('dict_result', False):
41 cursor = db.cursor(MySQLdb.cursors.DictCursor)
42 else:
43 cursor = db.cursor()
44 try:
45 cursor.execute(sql, params)
46 db.commit()
47 return cursor.fetchall()
48 finally:
49 cursor.close()
50
51 def write(db, queries):
52 """
53 This writes a given iteratable object of tuples containing SQL
54 strings and any required parameters to the database. All queries will
55 be run as one transaction and rolled back on error.
56 """
57 try:
58 cursor = db.cursor()
59 try:
60 for query in queries:
61 sql, params = query[0], query[1:]
62 cursor.execute(sql, params)
63 db.commit()
64 finally:
65 cursor.close()
66 except MySQLdb.Error:
67 db.rollback()
68 raise
OLDNEW

Powered by Google App Engine
This is Rietveld