Python 内置了 sqlite3 模块,可以方便地调用 SQLite 数据库。
import sqlite3
conn = sqlite3.connect('test.db')
cur = conn.cursor()
cur.execute('CREATE TABLE students (id BIGINT PRIMARY KEY NOT NULL, name VARCHAR(10) NOT NULL)')
conn.commit()
cur.close()
conn.close()
以上代码有一个问题,就是如果出现异常,那么,数据库不会关闭,所以,更完善的写法如下:
import sqlite3
conn = sqlite3.connect('test.db')
try:
cur = conn.cursor()
cur.execute('CREATE TABLE students (id BIGINT PRIMARY KEY NOT NULL, name VARCHAR(10) NOT NULL)')
conn.commit()
except (sqlite3.Warning, sqlite3.Error) as e:
print(e)
exit(1)
finally:
cur.close()
conn.close()
但是,这么写不免有些繁琐,我们知道, Python 里面有个with ... as ... 语句可以简化代码:
with open('example.txt') as f:
pass