git-interface/db.py

   1 import mysql.connector
   2
   3 import config
   4
   5
   6 class Connection:
   7     _conn = None
   8
   9     def __init__(self):
  10         aur_db_host = config.get('database', 'host')
  11         aur_db_name = config.get('database', 'name')
  12         aur_db_user = config.get('database', 'user')
  13         aur_db_pass = config.get('database', 'password')
  14         aur_db_socket = config.get('database', 'socket')
  15
  16         self._conn = mysql.connector.connect(host=aur_db_host,
  17                                              user=aur_db_user,
  18                                              passwd=aur_db_pass,
  19                                              db=aur_db_name,
  20                                              unix_socket=aur_db_socket,
  21                                              buffered=True)
  22
  23     def execute(self, query, params=()):
  24         query = query.replace('%', '%%').replace('?', '%s')
  25
  26         cur = self._conn.cursor()
  27         cur.execute(query, params)
  28
  29         return cur
  30
  31     def commit(self):
  32         self._conn.commit()
  33
  34     def close(self):
  35         self._conn.close()