mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-11-07 07:15:35 +00:00
Code Issues 32 Releases Wiki Activity

Fixed #32234 -- Made inspectdb inform about composite primary keys.

Browse Source
This commit is contained in:
Anv3sh
2022-05-12 01:00:13 +05:30
committed by Mariusz Felisiak
parent ebf25555bb
commit 295249c901
10 changed files with 88 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
django/db/backends/sqlite3/introspection.py
View File

@@ -156,15 +156,11 @@ class DatabaseIntrospection(BaseDatabaseIntrospection):
) in cursor.fetchall()
}
def get_primary_key_column(self, cursor, table_name):
"""Return the column name of the primary key for the given table."""
def get_primary_key_columns(self, cursor, table_name):
cursor.execute(
"PRAGMA table_info(%s)" % self.connection.ops.quote_name(table_name)
)
for _, name, *_, pk in cursor.fetchall():
if pk:
return name
return None
return [name for _, name, *_, pk in cursor.fetchall() if pk]
def _parse_column_or_constraint_definition(self, tokens, columns):
token = None
@@ -372,14 +368,14 @@ class DatabaseIntrospection(BaseDatabaseIntrospection):
if orders is not None:
constraints[index]["orders"] = orders
# Get the PK
pk_column = self.get_primary_key_column(cursor, table_name)
if pk_column:
pk_columns = self.get_primary_key_columns(cursor, table_name)
if pk_columns:
# SQLite doesn't actually give a name to the PK constraint,
# so we invent one. This is fine, as the SQLite backend never
# deletes PK constraints by name, as you can't delete constraints
# in SQLite; we remake the table with a new PK instead.
constraints["__primary__"] = {
"columns": [pk_column],
"columns": pk_columns,
"primary_key": True,
"unique": False, # It's not actually a unique constraint.
"foreign_key": None,