summaryrefslogtreecommitdiff
path: root/django/db/backends/sqlite3/introspection.py
diff options
context:
space:
mode:
authorAnv3sh <anveshgreat11@gmail.com>2021-09-21 22:49:58 +0530
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2021-10-26 09:50:23 +0200
commit69af4d09bae7782f733e01b19cf69fa6f36d7165 (patch)
tree871d2664f1aa42ae91e19635417d522435764a81 /django/db/backends/sqlite3/introspection.py
parentd446f8ba0820c8a4a923614611ae53582d38bc8e (diff)
downloaddjango-69af4d09bae7782f733e01b19cf69fa6f36d7165.tar.gz
Fixed #32672 -- Fixed introspection of primary key constraints on SQLite.
Thanks Simon Charette for the implementation idea.
Diffstat (limited to 'django/db/backends/sqlite3/introspection.py')
-rw-r--r--django/db/backends/sqlite3/introspection.py21
1 files changed, 4 insertions, 17 deletions
diff --git a/django/db/backends/sqlite3/introspection.py b/django/db/backends/sqlite3/introspection.py
index f109b5d5ee..ce097dbd72 100644
--- a/django/db/backends/sqlite3/introspection.py
+++ b/django/db/backends/sqlite3/introspection.py
@@ -201,25 +201,12 @@ class DatabaseIntrospection(BaseDatabaseIntrospection):
def get_primary_key_column(self, cursor, table_name):
"""Return the column name of the primary key for the given table."""
- # Don't use PRAGMA because that causes issues with some transactions
cursor.execute(
- "SELECT sql, type FROM sqlite_master "
- "WHERE tbl_name = %s AND type IN ('table', 'view')",
- [table_name]
+ 'PRAGMA table_info(%s)' % self.connection.ops.quote_name(table_name)
)
- row = cursor.fetchone()
- if row is None:
- raise ValueError("Table %s does not exist" % table_name)
- create_sql, table_type = row
- if table_type == 'view':
- # Views don't have a primary key.
- return None
- fields_sql = create_sql[create_sql.index('(') + 1:create_sql.rindex(')')]
- for field_desc in fields_sql.split(','):
- field_desc = field_desc.strip()
- m = re.match(r'(?:(?:["`\[])(.*)(?:["`\]])|(\w+)).*PRIMARY KEY.*', field_desc)
- if m:
- return m[1] if m[1] else m[2]
+ for _, name, *_, pk in cursor.fetchall():
+ if pk:
+ return name
return None
def _get_foreign_key_constraints(self, cursor, table_name):