@@ -167,8 +167,12 @@ def get_relations(self, cursor, table_name):
167
167
rc.UNIQUE_CONSTRAINT_NAME = ccu.CONSTRAINT_NAME
168
168
WHERE
169
169
tc.TABLE_SCHEMA=@schema_name AND tc.TABLE_NAME=@view_name
170
- """ , params = {"schema_name" : schema_name , "view_name" : self .connection .ops .quote_name (table_name )}
171
- )
170
+ """ ,
171
+ params = {
172
+ "schema_name" : schema_name ,
173
+ "view_name" : self .connection .ops .quote_name (table_name ),
174
+ },
175
+ )
172
176
return {
173
177
column : (referred_column , referred_table )
174
178
for (column , referred_column , referred_table ) in results
@@ -199,8 +203,12 @@ def get_primary_key_column(self, cursor, table_name):
199
203
ccu ON tc.CONSTRAINT_NAME = ccu.CONSTRAINT_NAME
200
204
WHERE
201
205
tc.TABLE_NAME=@table_name AND tc.CONSTRAINT_TYPE='PRIMARY KEY' AND tc.TABLE_SCHEMA=@schema_name
202
- """ , params = {"schema_name" : schema_name , "table_name" : self .connection .ops .quote_name (table_name )}
203
- )
206
+ """ ,
207
+ params = {
208
+ "schema_name" : schema_name ,
209
+ "table_name" : self .connection .ops .quote_name (table_name ),
210
+ },
211
+ )
204
212
return results [0 ][0 ] if results else None
205
213
206
214
def get_constraints (self , cursor , table_name ):
@@ -221,13 +229,13 @@ def get_constraints(self, cursor, table_name):
221
229
222
230
# Firstly populate all available constraints and their columns.
223
231
constraint_columns = cursor .run_sql_in_snapshot (
224
- '''
232
+ """
225
233
SELECT
226
234
CONSTRAINT_NAME, COLUMN_NAME
227
235
FROM
228
236
INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE
229
- WHERE TABLE_NAME=@table AND TABLE_SCHEMA=@schema_name''' ,
230
- params = {"table" : quoted_table_name , "schema_name" : schema_name }
237
+ WHERE TABLE_NAME=@table AND TABLE_SCHEMA=@schema_name""" ,
238
+ params = {"table" : quoted_table_name , "schema_name" : schema_name },
231
239
)
232
240
for constraint , column_name in constraint_columns :
233
241
if constraint not in constraints :
@@ -246,14 +254,14 @@ def get_constraints(self, cursor, table_name):
246
254
247
255
# Add the various constraints by type.
248
256
constraint_types = cursor .run_sql_in_snapshot (
249
- '''
257
+ """
250
258
SELECT
251
259
CONSTRAINT_NAME, CONSTRAINT_TYPE
252
260
FROM
253
261
INFORMATION_SCHEMA.TABLE_CONSTRAINTS
254
262
WHERE
255
- TABLE_NAME=@table AND TABLE_SCHEMA=@schema_name''' ,
256
- params = {"table" : quoted_table_name , "schema_name" : schema_name }
263
+ TABLE_NAME=@table AND TABLE_SCHEMA=@schema_name""" ,
264
+ params = {"table" : quoted_table_name , "schema_name" : schema_name },
257
265
)
258
266
for constraint , constraint_type in constraint_types :
259
267
already_added = constraint in constraints
@@ -299,7 +307,8 @@ def get_constraints(self, cursor, table_name):
299
307
idx.TABLE_NAME=@table AND idx.TABLE_SCHEMA=@schema_name
300
308
ORDER BY
301
309
idx_col.ORDINAL_POSITION
302
- """ , params = {"table" : quoted_table_name , "schema_name" : schema_name }
310
+ """ ,
311
+ params = {"table" : quoted_table_name , "schema_name" : schema_name },
303
312
)
304
313
for (
305
314
index_name ,
@@ -357,7 +366,11 @@ def get_key_columns(self, cursor, table_name):
357
366
rc.CONSTRAINT_NAME = ccu.CONSTRAINT_NAME
358
367
WHERE
359
368
tc.TABLE_NAME=@table AND tc.TABLE_SCHEMA=@schema_name
360
- """ , params = {"table" : self .connection .ops .quote_name (table_name ), "schema_name" : schema_name }
369
+ """ ,
370
+ params = {
371
+ "table" : self .connection .ops .quote_name (table_name ),
372
+ "schema_name" : schema_name ,
373
+ },
361
374
)
362
375
key_columns .extend (cursor .fetchall ())
363
376
return key_columns
0 commit comments