Skip to content

Commit d840a53

Browse files
1 parent 1a0d8d9 commit d840a53

File tree

1 file changed

+25
-12
lines changed

1 file changed

+25
-12
lines changed

‎django_spanner/introspection.py‎

Lines changed: 25 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -167,8 +167,12 @@ def get_relations(self, cursor, table_name):
167167
rc.UNIQUE_CONSTRAINT_NAME = ccu.CONSTRAINT_NAME
168168
WHERE
169169
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+
)
172176
return {
173177
column: (referred_column, referred_table)
174178
for (column, referred_column, referred_table) in results
@@ -199,8 +203,12 @@ def get_primary_key_column(self, cursor, table_name):
199203
ccu ON tc.CONSTRAINT_NAME = ccu.CONSTRAINT_NAME
200204
WHERE
201205
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+
)
204212
return results[0][0] if results else None
205213

206214
def get_constraints(self, cursor, table_name):
@@ -221,13 +229,13 @@ def get_constraints(self, cursor, table_name):
221229

222230
# Firstly populate all available constraints and their columns.
223231
constraint_columns = cursor.run_sql_in_snapshot(
224-
'''
232+
"""
225233
SELECT
226234
CONSTRAINT_NAME, COLUMN_NAME
227235
FROM
228236
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},
231239
)
232240
for constraint, column_name in constraint_columns:
233241
if constraint not in constraints:
@@ -246,14 +254,14 @@ def get_constraints(self, cursor, table_name):
246254

247255
# Add the various constraints by type.
248256
constraint_types = cursor.run_sql_in_snapshot(
249-
'''
257+
"""
250258
SELECT
251259
CONSTRAINT_NAME, CONSTRAINT_TYPE
252260
FROM
253261
INFORMATION_SCHEMA.TABLE_CONSTRAINTS
254262
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},
257265
)
258266
for constraint, constraint_type in constraint_types:
259267
already_added = constraint in constraints
@@ -299,7 +307,8 @@ def get_constraints(self, cursor, table_name):
299307
idx.TABLE_NAME=@table AND idx.TABLE_SCHEMA=@schema_name
300308
ORDER BY
301309
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},
303312
)
304313
for (
305314
index_name,
@@ -357,7 +366,11 @@ def get_key_columns(self, cursor, table_name):
357366
rc.CONSTRAINT_NAME = ccu.CONSTRAINT_NAME
358367
WHERE
359368
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+
},
361374
)
362375
key_columns.extend(cursor.fetchall())
363376
return key_columns

0 commit comments

Comments
 (0)