python訪問(wèn)PostgreSQL數(shù)據(jù)庫(kù)之連接庫(kù)Psycopg2

作者：傭工7001

由于要在python訪問(wèn)PostgreSQL數(shù)據(jù)庫(kù)，需要一個(gè)符合DB-API的連接庫(kù)。通過(guò)搜索，鎖定兩個(gè)候選庫(kù)：一個(gè)是py-postgresql，一個(gè)是Psycopg2。簡(jiǎn)單的看了一下py-postgresql的文檔，發(fā)現(xiàn)它并非基于DB-api接口的，而是使用postgresql的API，這可不是我想要的，只好排除。剩下就是這個(gè)名字很怪異的Psycopg，看了下，它遵循DB-API規(guī)范，好了就選它了，下載安裝，試用，老規(guī)矩，記錄下來(lái)要點(diǎn)備忘。

一、安裝

由于我開(kāi)發(fā)環(huán)境在windows下，所以我下載的是window下的一鍵安裝版本，安裝過(guò)程非常容易沒(méi)啥可說(shuō)的。

二、使用

面的簡(jiǎn)單使用流程是官網(wǎng)的文檔中的例子，看上去是這么的熟悉，是個(gè)不錯(cuò)的開(kāi)始。

# 引入psycopg2庫(kù)

>>> import psycopg2

# 連接到數(shù)據(jù)庫(kù)test

>>> conn = psycopg2.connect("dbname=test user=postgres")

# 建立Cursor對(duì)象

>>> cur = conn.cursor()

# 執(zhí)行sql命令：創(chuàng)建一個(gè)新表test

>>> cur.execute("CREATE TABLE test (id serial PRIMARY KEY, num integer, data varchar);")

# 傳遞數(shù)據(jù)到語(yǔ)句中形參中，讓Psycopg處理正確的轉(zhuǎn)換問(wèn)題(可以避免sql注入)

>>> cur.execute("INSERT INTO test (num, data) VALUES (%s, %s)",

... (100, "abc'def"))

# 插敘數(shù)據(jù)，并獲取結(jié)果

>>> cur.execute("SELECT * FROM test;")

>>> cur.fetchone()

(1, 100, "abc'def")

# 提交數(shù)據(jù)改變

>>> conn.commit()

# 關(guān)閉Cursor對(duì)象和連接對(duì)象

>>> cur.close()

>>> conn.close()

下面就要看psycopg是否能完全滿足我的需求：

(-)建立數(shù)據(jù)庫(kù)連接psycopg2中建立數(shù)據(jù)庫(kù)連接需要使用connect方法，方法形式如下：psycopg2.connect(dsn or params [, connection_factory] [, async=0])

返回值是一個(gè)Connection對(duì)象。1，可以使用DSN連接數(shù)據(jù)庫(kù)，也就是數(shù)據(jù)源名稱字符串，例如conn = psycopg2.connect("dbname=test user=postgres password=secret")2，也可以使用命名參數(shù)進(jìn)行連接。conn = psycopg2.connect(database="test", user="postgres", password="secret")

基本的連接參數(shù)有:

dbname – 數(shù)據(jù)庫(kù)名(僅在DSN中使用有效)

database – 數(shù)據(jù)庫(kù)名(僅在使用命名參數(shù)時(shí)可用)

user – 連接數(shù)據(jù)所用的數(shù)據(jù)庫(kù)用戶名

password – 用戶的密碼

host – 數(shù)據(jù)庫(kù)所在主機(jī)名或IP地址(默認(rèn)為本機(jī))

port – 端口號(hào)(默認(rèn)為5432 )

還有其他一些參數(shù)，請(qǐng)參見(jiàn)PostgreSQL官方文檔 list of supported parameters。另外需要注意的是，同樣的參數(shù)也可以通過(guò)環(huán)境變量的形式傳遞到客戶端庫(kù)。

3，使用connection_factory參數(shù)，可以指定自定義的Connection連接類(lèi)，詳情查看官方文檔 Connection and cursor factories 。

4，設(shè)置參數(shù)async=1，則會(huì)創(chuàng)建一個(gè)異步的連接對(duì)象。具體參見(jiàn)官方文檔Asynchronous support 了解異步的優(yōu)缺點(diǎn)。

(二)創(chuàng)建字典形式的Cursor對(duì)象

簡(jiǎn)單的創(chuàng)建Cursor對(duì)象十分簡(jiǎn)單，但是這不能滿足我的需求。DictCursor這個(gè)才是我所需要的，也就是在創(chuàng)建Cursor的時(shí)候給出參數(shù)cursor_factory

import psycopg2

#為了使用DictCursor

import psycopg2.extrascursor = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)使用過(guò)dbapi編程的，這個(gè)好處你懂的

(三)使用命名參數(shù)傳遞數(shù)據(jù)

我的另一個(gè)需求就是，執(zhí)行sql語(yǔ)句時(shí)，最好使用命名參數(shù)傳遞數(shù)據(jù)，幸運(yùn)的是psycopg2也是支持命名參數(shù)的！psycopg中命名參數(shù)的形式是：【%(argname)s】，注意在中括號(hào)中的內(nèi)容就是命名參數(shù)要求的形式，其中argname是你起的參數(shù)名稱，其他的百分號(hào)小括號(hào)以及小括號(hào)后面的s，照寫(xiě)就是。

使用命名參數(shù)的好處就是，你不用考慮傳遞數(shù)據(jù)的順序，可以使用字典傳遞數(shù)據(jù)(注意字典的key，應(yīng)該等于上面所說(shuō)的argname)：

>>> cur.execute(

... """INSERT INTO some_table (an_int, a_date, another_date, a_string)

... VALUES (%(int)s, %(date)s, %(date)s, %(str)s);""",

... {'int': 10, 'str': "O'Reilly", 'date': datetime.date(2005, 11, 18)})

總結(jié)

以上是生活随笔為你收集整理的python postgresql跨数据库查询_python访问PostgreSQL数据库之连接库Psycopg2的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。