python - 如何以编程方式确定列是否在 SQLite 中设置为 AUTOINCREMENT？

Question

使用 Python 2.7 的 sqlite3 库，如何确定表的哪些列是 AUTOINCREMENT？我知道我可以使用 SQLite 命令行实用程序，但我该如何以编程方式进行操作。我粗略地查看了 SQLite 文档，我能找到的最接近的 PRAGMA 命令是 table_info .

Answer 1

AUTOINCREMENT 仅适用于主键。因此对于给定的表，您可以使用 PRAGMA table_info([tablename]) 来确定哪一列是主键:

>>> import sqlite3
>>> conn = sqlite3.connect(':memory:')
>>> conn.execute('CREATE TABLE foo (bar INTEGER PRIMARY KEY AUTOINCREMENT, baz)')
<sqlite3.Cursor object at 0x10a124f50>
>>> c = conn.cursor()
>>> c.execute('PRAGMA table_info("foo")')
<sqlite3.Cursor object at 0x10a124ef8>
>>> for row in c: print row
... 
(0, u'bar', u'INTEGER', 0, None, 1)
(1, u'baz', u'', 0, None, 0)
>>> [col[0] for col in c.description]
['cid', 'name', 'type', 'notnull', 'dflt_value', 'pk']

所以行中的最后一列是 pk 行，对于设置为 1 的 bar。

要确定主键是否自动递增完全，您可以执行以下两项操作之一:

1. 查询sqlite_master table并检查架构中是否提到了 AUTOINCREMENT:

   >>> c.execute('SELECT 1 FROM sqlite_master WHERE tbl_name="foo" AND sql LIKE "%AUTOINCREMENT%"')
   <sqlite3.Cursor object at 0x10a124ef8>
   >>> c.fetchone()
   (1,)
   

2. 如果有插入数据，表名会出现在sqlite_sequence table中:

   >>> c.execute('insert into foo (baz) values (1)')
   <sqlite3.Cursor object at 0x10a124ef8>
   >>> c.execute('SELECT 1 FROM sqlite_sequence WHERE name="foo"')
   <sqlite3.Cursor object at 0x10a124ef8>
   >>> c.fetchone()
   (1,)