最全總結 | 聊聊 Python 資料處理全家桶（PgSQL篇）

AirPython發表於2021-08-12
PythonSQL

1. 前言

大家好，我是安果！

Python 資料處理全家桶，截止到現在，一共寫過 6 篇文章，有興趣的小夥伴可以去了解一下！

最全總結 | 聊聊 Python 資料處理全家桶（Mysql 篇）

最全總結 | 聊聊 Python 資料處理全家桶（Sqlite篇）

最全總結 | 聊聊 Python 資料處理全家桶（MongoDB 篇）

最全總結 | 聊聊 Python 資料處理全家桶（Redis篇）

最全總結 | 聊聊 Python 資料處理全家桶（Memcached篇）

最全總結 | 聊聊 Python 資料處理全家桶（儲存過程篇）

PgSQL，全稱為 PostgreSQL，是一款免費開源的關係型資料庫

相比最流行的 Mysql 資料庫，PgSQL 在可靠性、資料完整性、擴充套件性方面具有絕對的優勢

本篇文章將聊聊如何使用 Python 操作 PgSQL 資料庫

2. PgSQL 使用

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

# 安裝依賴包
pip3 install psycopg2

接下來，就可以使用 Python 來運算元據庫了

2-1  資料庫連線及遊標物件

使用 psycopg2 中的「 connect() 」方法連線資料庫，建立資料庫連線物件及遊標物件

import psycopg2

# 獲得連線物件
# database：資料庫名稱
# user：使用者名稱
# password：密碼
# host：資料庫ip地址
# port：埠號，預設為5432
conn = psycopg2.connect(database="db_name", user="postgres", password="pwd", host="127.0.0.1", port="5432")

# 獲取遊標物件
cursor = conn.cursor()

獲取遊標物件後，就可以執行 SQL，進而運算元據庫了

2-2  插入資料

首先，編寫插入資料的 SQL 語句及引數（ 可選 ）

# 構建SQL語句
# 方式一：直帶引數
​ sql = "INSERT INTO student (name,age) \
                        VALUES (%s, '%s')" % \
          ('xag',23)

# 方式二：引數分離
sql = """INSERT INTO student (name,age) VALUES (%s, %s)"""
# 引數
params = ('xag',23)

然後，使用遊標物件執行 SQL

# 執行sql
# 注意：params可選，根據上面的引數方式來選擇設定
​cursor.execute(sql,[params])

接著，使用連線物件提交事務

 # 事務提交
conn.commit()

最後，釋放遊標物件及資料庫連線物件

# 釋放遊標物件及資料庫連線物件
cursor.close()
conn.close()

2-3  查詢資料

遊標物件的 fetchone()、fetchmany(size)、fetchall() 這 3個函式即可以實現單條資料查詢、多條資料查詢、全部資料查詢

# 獲取一條記錄
one_data = cursor.fetchone()
print(one_data)

# 獲取2條記錄
many_data = cursor.fetchmany(2)
print(many_data)

# 獲取全部資料
all_data = cursor.fetchall()
print(all_data)

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

# 條件查詢 SQL語句
sql = """SELECT * FROM student where id = %s;"""

# 對應引數，引數結尾以逗號結尾
​params = (1,)

# 執行SQL
cursor.execute(sql, params)

# 獲取所有資料
datas = cursor.fetchall()
print(datas)

2-4  更新資料

更新操作和上面操作一樣，唯一不同的是，執行完 SQL 後，需要使用連線物件提交事務，才能將資料真實更新到資料庫中

def update_one(conn, cursor):
    """更新操作"""
    # 更新語句
    sql = """update student set name = %s where id = %s  """
    params = ('AirPython', 1,)

    # 執行語句
    cursor.execute(sql, params)

    # 事務提交
    conn.commit()

    # 關閉資料庫連線
    cursor.close()
    conn.close()

2-5  刪除資料

刪除資料同更新資料操作類似

​def delete_one(conn, cursor):
    """刪除操作"""
    # 語句及引數
    sql = """delete from  student where id = %s  """
    params = (1,)

    # 執行語句
    cursor.execute(sql, params)

    # 事物提交
    conn.commit()

    # 關閉資料庫連線
    cursor.close()
    conn.close()

3. 最後

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

更多關於 PgSQL 的操作可以參考下面連結：

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

如果你覺得文章還不錯，請大家 點贊、分享、留言 下，因為這將是我持續輸出更多優質文章的最強動力！

相關文章