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行，为此bar设置为1.

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

1. 查询sqlite_master表并检查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表中：

   >>> 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,)