跳转至

SQL 数据库

Python 提供了一系列标准的数据库的 API,这里我们介绍 sqlite 数据库的用法,其他的数据库的用法大同小异:

In [1]:

import sqlite3 as db

首先我们要建立或者连接到一个数据库上:

In [2]:

connection = db.connect("my_database.sqlite")

不同的数据库有着不同的连接方法,例如 cx-oracle 数据库的链接方式为:

connection = db.connect(username, password, host, port,  'XE') 

一旦建立连接,我们可以利用它的 cursor() 来执行 SQL 语句:

In [3]:

cursor = connection.cursor()
cursor.execute("""CREATE TABLE IF NOT EXISTS orders(
 order_id TEXT PRIMARY KEY,
 date TEXT,
 symbol TEXT,
 quantity INTEGER,
 price NUMBER)""")
cursor.execute("""INSERT INTO orders VALUES
 ('A0001', '2013-12-01', 'AAPL', 1000, 203.4)""")
connection.commit()

不过为了安全起见,一般不将数据内容写入字符串再传入,而是使用这样的方式:

In [4]:

orders = [
          ("A0002","2013-12-01","MSFT",1500,167.5),
          ("A0003","2013-12-02","GOOG",1500,167.5)
]
cursor.executemany("""INSERT INTO orders VALUES
 (?, ?, ?, ?, ?)""", orders)
connection.commit()

cx-oracle 数据库使用不同的方式:

cursor.executemany("""INSERT INTO orders VALUES
(:order_id, :date, :symbol, :quantity, :price)""",
orders) 

查看支持的数据库格式:

In [5]:

db.paramstyle

Out[5]:

'qmark'

query 语句执行之后,我们需要进行 commit,否则数据库将不会接受这些变化,如果想撤销某个 commit,可以使用 rollback() 方法撤销到上一次 commit() 的结果:

try:
    ... # perform some operations
except:
    connection.rollback()
    raise
else:
    connection.commit() 

使用 SELECT 语句对数据库进行查询:

In [6]:

stock = 'MSFT'
cursor.execute("""SELECT *
 FROM orders
 WHERE symbol=?
 ORDER BY quantity""", (stock,))
for row in cursor:
    print row
(u'A0002', u'2013-12-01', u'MSFT', 1500, 167.5)

cursor.fetchone() 返回下一条内容, cursor.fetchall() 返回所有查询到的内容组成的列表(可能非常大):

In [7]:

stock = 'AAPL'
cursor.execute("""SELECT *
 FROM orders
 WHERE symbol=?
 ORDER BY quantity""", (stock,))
cursor.fetchall()

Out[7]:

[(u'A0001', u'2013-12-01', u'AAPL', 1000, 203.4)]

关闭数据库:

In [8]:

cursor.close()
connection.close()


回到顶部