@@ -224,39 +224,53 @@ def key(self):
224
224
def _set_value (self , key , value ):
225
225
self ._init_db ()
226
226
value = json .dumps (value , sort_keys = True )
227
- query = """
227
+ cr = self ._data_table_cr
228
+ query = util .format_query (
229
+ cr ,
230
+ """
228
231
INSERT INTO {} (key, value) VALUES (%s, %s)
229
232
ON CONFLICT (key) DO UPDATE SET value=EXCLUDED.value
230
- """ .format (DATA_TABLE )
231
- self ._data_table_cr .execute (query , (key , value ))
232
- self ._data_table_cr ._cnx .commit ()
233
+ """ ,
234
+ DATA_TABLE ,
235
+ )
236
+ cr .execute (query , (key , value ))
237
+ cr ._cnx .commit ()
233
238
234
239
def _get_value (self , key ):
235
240
self ._init_db ()
236
- query = "SELECT value FROM {} WHERE key = %s" .format (DATA_TABLE )
237
- self ._data_table_cr .execute (query , [key ])
238
- result = self ._data_table_cr .fetchone ()
241
+ cr = self ._data_table_cr
242
+ query = util .format_query (cr , "SELECT value FROM {} WHERE key = %s" , DATA_TABLE )
243
+ cr .execute (query , [key ])
244
+ result = cr .fetchone ()
239
245
if not result :
240
246
raise KeyError (key )
241
247
return result [0 ]
242
248
243
249
def _key_exists (self , key ):
244
250
self ._init_db ()
245
- query = "SELECT 1 FROM {} WHERE key = %s" .format (DATA_TABLE )
246
- self ._data_table_cr .execute (query , [key ])
247
- return bool (self ._data_table_cr .rowcount )
251
+ cr = self ._data_table_cr
252
+ query = util .format_query (cr , "SELECT 1 FROM {} WHERE key = %s" , DATA_TABLE )
253
+ cr .execute (query , [key ])
254
+ return bool (cr .rowcount )
248
255
249
256
def _init_db (self ):
250
257
if not UpgradeCommon .__initialized :
251
- self ._data_table_cr .execute ("SELECT 1 FROM pg_class WHERE relname=%s" , [DATA_TABLE ])
252
- if not self ._data_table_cr .rowcount :
258
+ cr = self ._data_table_cr
259
+ cr .execute ("SELECT 1 FROM pg_class WHERE relname=%s" , [DATA_TABLE ])
260
+ if not cr .rowcount :
253
261
_logger .info ("Creating table %s" , DATA_TABLE )
254
- query = """ CREATE TABLE {} (
255
- key VARCHAR(255) PRIMARY KEY,
256
- value JSONB NOT NULL
257
- )""" .format (DATA_TABLE )
258
- self ._data_table_cr .execute (query )
259
- self ._data_table_cr ._cnx .commit ()
262
+ query = util .format_query (
263
+ cr ,
264
+ """
265
+ CREATE TABLE {} (
266
+ key VARCHAR(255) PRIMARY KEY,
267
+ value JSONB NOT NULL
268
+ )
269
+ """ ,
270
+ DATA_TABLE ,
271
+ )
272
+ cr .execute (query )
273
+ cr ._cnx .commit ()
260
274
UpgradeCommon .__initialized = True
261
275
262
276
def _setup_registry (self ):
0 commit comments