From 2edb9a1d57d9be01f15d67077b9f2b5bb3d79d05 Mon Sep 17 00:00:00 2001 From: Ethan Dalool Date: Sun, 25 Mar 2018 16:52:23 -0700 Subject: [PATCH] Centralize cursors to PDB.sql_execute. This gives me a nice common place to do logging if I want. --- etiquette/photodb.py | 29 +++++++++++++---------------- 1 file changed, 13 insertions(+), 16 deletions(-) diff --git a/etiquette/photodb.py b/etiquette/photodb.py index 9a333ab..6ce91d5 100644 --- a/etiquette/photodb.py +++ b/etiquette/photodb.py @@ -736,31 +736,32 @@ class PDBSQLMixin: return savepoint_id def sql_delete(self, table, pairs, *, commit=False): - cur = self.sql.cursor() - (qmarks, bindings) = sqlhelpers.delete_filler(pairs) query = 'DELETE FROM %s %s' % (table, qmarks) - #self.log.debug(query) - cur.execute(query, bindings) + self.sql_execute(query, bindings) if commit: self.commit() + def sql_execute(self, query, bindings=[]): + if bindings is None: + bindings = [] + cur = self.sql.cursor() + cur.execute(query, bindings) + return cur + def sql_insert(self, table, data, *, commit=False): column_names = constants.SQL_COLUMNS[table] - cur = self.sql.cursor() - (qmarks, bindings) = sqlhelpers.insert_filler(column_names, data) + query = 'INSERT INTO %s VALUES(%s)' % (table, qmarks) - #self.log.debug(query) - cur.execute(query, bindings) + self.sql_execute(query, bindings) if commit: self.commit() def sql_select(self, query, bindings=None): - cur = self.sql.cursor() - cur.execute(query, bindings) + cur = self.sql_execute(query, bindings) while True: fetch = cur.fetchone() if fetch is None: @@ -768,17 +769,13 @@ class PDBSQLMixin: yield fetch def sql_select_one(self, query, bindings=None): - cur = self.sql.cursor() - cur.execute(query, bindings) + cur = self.sql_execute(query, bindings) return cur.fetchone() def sql_update(self, table, pairs, where_key, *, commit=False): - cur = self.sql.cursor() - (qmarks, bindings) = sqlhelpers.update_filler(pairs, where_key=where_key) query = 'UPDATE %s %s' % (table, qmarks) - #self.log.debug(query) - cur.execute(query, bindings) + self.sql_execute(query, bindings) if commit: self.commit()