[[416887]]

本文轉(zhuǎn)載自微信公眾號(hào)「AirPython」，作者星安果。轉(zhuǎn)載本文請(qǐng)聯(lián)系A(chǔ)irPython公眾號(hào)。

丹徒網(wǎng)站建設(shè)公司創(chuàng)新互聯(lián),丹徒網(wǎng)站設(shè)計(jì)制作，有大型網(wǎng)站制作公司豐富經(jīng)驗(yàn)。已為丹徒成百上千家提供企業(yè)網(wǎng)站建設(shè)服務(wù)。企業(yè)網(wǎng)站搭建\外貿(mào)網(wǎng)站建設(shè)要多少錢，請(qǐng)找那個(gè)售后服務(wù)好的丹徒做網(wǎng)站的公司定做！

1. 前言

PgSQL，全稱為 PostgreSQL，是一款免費(fèi)開源的關(guān)系型數(shù)據(jù)庫

相比最流行的 Mysql 數(shù)據(jù)庫，PgSQL 在可靠性、數(shù)據(jù)完整性、擴(kuò)展性方面具有絕對(duì)的優(yōu)勢(shì)

本篇文章將聊聊如何使用 Python 操作 PgSQL 數(shù)據(jù)庫

2. PgSQL 使用

Python 操作 PgSQL，需要先安裝依賴包「 psycopg2 」

 
 
 

  
  
  
# 安裝依賴包 
  
  
  
pip3 install psycopg2 
 
 
 

接下來，就可以使用 Python 來操作數(shù)據(jù)庫了

2-1 數(shù)據(jù)庫連接及游標(biāo)對(duì)象

使用 psycopg2 中的「 connect() 」方法連接數(shù)據(jù)庫，創(chuàng)建數(shù)據(jù)庫連接對(duì)象及游標(biāo)對(duì)象

 
 
 

  
  
  
import psycopg2 
  
  
  
 
  
  
  
# 獲得連接對(duì)象 
  
  
  
# database：數(shù)據(jù)庫名稱 
  
  
  
# user：用戶名 
  
  
  
# password：密碼 
  
  
  
# host：數(shù)據(jù)庫ip地址 
  
  
  
# port：端口號(hào)，默認(rèn)為5432 
  
  
  
conn = psycopg2.connect(database="db_name", user="postgres", password="pwd", host="127.0.0.1", port="5432") 
  
  
  
 
  
  
  
# 獲取游標(biāo)對(duì)象 
  
  
  
cursor = conn.cursor() 
 
 
 

獲取游標(biāo)對(duì)象后，就可以執(zhí)行 SQL，進(jìn)而操作數(shù)據(jù)庫了

2-2 插入數(shù)據(jù)

首先，編寫插入數(shù)據(jù)的 SQL 語句及參數(shù)( 可選 )

 
 
 

  
  
  
# 構(gòu)建SQL語句 
  
  
  
# 方式一：直帶參數(shù) 
  
  
  
 sql = "INSERT INTO student (name,age) \ 
  
  
  
                        VALUES (%s, '%s')" % \ 
  
  
  
          ('xag',23) 
  
  
  
 
  
  
  
# 方式二：參數(shù)分離 
  
  
  
sql = """INSERT INTO student (name,age) VALUES (%s, %s)""" 
  
  
  
# 參數(shù) 
  
  
  
params = ('xag',23) 
 
 
 

然后，使用游標(biāo)對(duì)象執(zhí)行 SQL

 
 
 

  
  
  
# 執(zhí)行sql 
  
  
  
# 注意：params可選，根據(jù)上面的參數(shù)方式來選擇設(shè)置 
  
  
  
cursor.execute(sql,[params]) 
 
 
 

接著，使用連接對(duì)象提交事務(wù)

 
 
 

  
  
  
# 事務(wù)提交 
  
  
  
onn.commit() 
 
 
 

最后，釋放游標(biāo)對(duì)象及數(shù)據(jù)庫連接對(duì)象

 
 
 

  
  
  
# 釋放游標(biāo)對(duì)象及數(shù)據(jù)庫連接對(duì)象 
  
  
  
cursor.close() 
  
  
  
conn.close() 
 
 
 

2-3 查詢數(shù)據(jù)

游標(biāo)對(duì)象的 fetchone()、fetchmany(size)、fetchall() 這 3個(gè)函數(shù)即可以實(shí)現(xiàn)單條數(shù)據(jù)查詢、多條數(shù)據(jù)查詢、全部數(shù)據(jù)查詢

 
 
 

  
  
  
# 獲取一條記錄 
  
  
  
one_data = cursor.fetchone() 
  
  
  
print(one_data) 
  
  
  
 
  
  
  
# 獲取2條記錄 
  
  
  
many_data = cursor.fetchmany(2) 
  
  
  
print(many_data) 
  
  
  
 
  
  
  
# 獲取全部數(shù)據(jù) 
  
  
  
all_data = cursor.fetchall() 
  
  
  
print(all_data) 
 
 
 

需要注意的是，條件查詢與上面的插入操作類似，條件語句可以將參數(shù)分離出來

 
 
 

  
  
  
# 條件查詢 SQL語句 
  
  
  
sql = """SELECT * FROM student where id = %s;""" 
  
  
  
 
  
  
  
# 對(duì)應(yīng)參數(shù)，參數(shù)結(jié)尾以逗號(hào)結(jié)尾 
  
  
  
params = (1,) 
  
  
  
 
  
  
  
# 執(zhí)行SQL 
  
  
  
cursor.execute(sql, params) 
  
  
  
 
  
  
  
# 獲取所有數(shù)據(jù) 
  
  
  
datas = cursor.fetchall() 
  
  
  
print(datas) 
 
 
 

2-4 更新數(shù)據(jù)

更新操作和上面操作一樣，唯一不同的是，執(zhí)行完 SQL 后，需要使用連接對(duì)象提交事務(wù)，才能將數(shù)據(jù)真實(shí)更新到數(shù)據(jù)庫中

 
 
 

  
  
  
def update_one(conn, cursor): 
  
  
  
    """更新操作""" 
  
  
  
    # 更新語句 
  
  
  
    sql = """update student set name = %s where id = %s  """ 
  
  
  
    params = ('AirPython', 1,) 
  
  
  
 
  
  
  
    # 執(zhí)行語句 
  
  
  
    cursor.execute(sql, params) 
  
  
  
 
  
  
  
    # 事務(wù)提交 
  
  
  
    conn.commit() 
  
  
  
 
  
  
  
    # 關(guān)閉數(shù)據(jù)庫連接 
  
  
  
    cursor.close() 
  
  
  
    conn.close() 
 
 
 

2-5 刪除數(shù)據(jù)

刪除數(shù)據(jù)同更新數(shù)據(jù)操作類似

 
 
 

  
  
  
def delete_one(conn, cursor): 
  
  
  
    """刪除操作""" 
  
  
  
    # 語句及參數(shù) 
  
  
  
    sql = """delete from  student where id = %s  """ 
  
  
  
    params = (1,) 
  
  
  
 
  
  
  
    # 執(zhí)行語句 
  
  
  
    cursor.execute(sql, params) 
  
  
  
 
  
  
  
    # 事物提交 
  
  
  
    conn.commit() 
  
  
  
 
  
  
  
    # 關(guān)閉數(shù)據(jù)庫連接 
  
  
  
    cursor.close() 
  
  
  
    conn.close() 
 
 
 

3. 最后

通過上面操作，可以發(fā)現(xiàn) Python 操作 PgSQl 與 Mysql 類似，但是在原生 SQL 編寫上兩者還是有很多差異性

更多關(guān)于 PgSQL 的操作可以參考下面鏈接：

https://www.postgresql.org/docs/14/index.html

網(wǎng)頁題目：聊聊Python數(shù)據(jù)處理全家桶（PgSQL篇）
分享網(wǎng)址：http://uogjgqi.cn/article/cdphede.html