external/aurweb
0
0
Fork 0
mirror of https://gitlab.archlinux.org/archlinux/aurweb.git synced 2025-02-03 10:43:03 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

git-interface: Support SQLite as database backend

Browse source 
In addition to MySQL, add support for SQLite to the database abstraction
layer. Also, add a new configuration option to select the DBMS.

Signed-off-by: Lukas Fleischer <lfleischer@archlinux.org>
This commit is contained in:
Lukas Fleischer 2016-08-03 20:28:22 +02:00
parent f2a6bd207d
commit baf8a220ab
2 changed files with 29 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
conf/config.proto
View file

@ -1,5 +1,5 @@
[database] [database]
dsn_prefix = mysql backend = mysql
host = localhost host = localhost
socket = /var/run/mysqld/mysqld.sock socket = /var/run/mysqld/mysqld.sock
name = AUR name = AUR

18
git-interface/db.py
View file

@ -1,27 +1,43 @@
import mysql.connector import mysql.connector
import sqlite3
import config import config
class Connection: class Connection:
_conn = None _conn = None
_paramstyle = None
def __init__(self): def __init__(self):
aur_db_backend = config.get('database', 'backend')
if aur_db_backend == 'mysql':
aur_db_host = config.get('database', 'host') aur_db_host = config.get('database', 'host')
aur_db_name = config.get('database', 'name') aur_db_name = config.get('database', 'name')
aur_db_user = config.get('database', 'user') aur_db_user = config.get('database', 'user')
aur_db_pass = config.get('database', 'password') aur_db_pass = config.get('database', 'password')
aur_db_socket = config.get('database', 'socket') aur_db_socket = config.get('database', 'socket')
self._conn = mysql.connector.connect(host=aur_db_host, self._conn = mysql.connector.connect(host=aur_db_host,
user=aur_db_user, user=aur_db_user,
passwd=aur_db_pass, passwd=aur_db_pass,
db=aur_db_name, db=aur_db_name,
unix_socket=aur_db_socket, unix_socket=aur_db_socket,
buffered=True) buffered=True)
self._paramstyle = mysql.connector.paramstyle
elif aur_db_backend == 'sqlite':
aur_db_name = config.get('database', 'name')
self._conn = sqlite3.connect(aur_db_name)
self._paramstyle = sqlite3.paramstyle
else:
raise ValueError('unsupported database backend')
def execute(self, query, params=()): def execute(self, query, params=()):
if self._paramstyle == 'format':
query = query.replace('%', '%%').replace('?', '%s') query = query.replace('%', '%%').replace('?', '%s')
elif self._paramstyle == 'qmark':
pass
else:
raise ValueError('unsupported paramstyle')
cur = self._conn.cursor() cur = self._conn.cursor()
cur.execute(query, params) cur.execute(query, params)