Python與SQL的連接可以通過多種方式實現,具體取決于你使用的數據庫類型。以下是一些常見的方法:
import sqlite3
# 連接到SQLite數據庫(如果不存在,則創建)
conn = sqlite3.connect('example.db')
# 創建一個游標對象
cursor = conn.cursor()
# 執行SQL命令
cursor.execute('CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT)')
cursor.execute("INSERT INTO users (name) VALUES ('Alice')")
cursor.execute("INSERT INTO users (name) VALUES ('Bob')")
# 提交更改
conn.commit()
# 查詢數據
cursor.execute('SELECT * FROM users')
rows = cursor.fetchall()
for row in rows:
print(row)
# 關閉連接
conn.close()
首先,你需要安裝這個模塊:
pip install mysql-connector-python
然后,你可以使用以下代碼連接到MySQL數據庫:
import mysql.connector
# 連接到MySQL數據庫
conn = mysql.connector.connect(
host='localhost',
user='your_username',
password='your_password',
database='your_database'
)
# 創建一個游標對象
cursor = conn.cursor()
# 執行SQL命令
cursor.execute('CREATE TABLE IF NOT EXISTS users (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255))')
cursor.execute("INSERT INTO users (name) VALUES ('Alice')")
cursor.execute("INSERT INTO users (name) VALUES ('Bob')")
# 提交更改
conn.commit()
# 查詢數據
cursor.execute('SELECT * FROM users')
rows = cursor.fetchall()
for row in rows:
print(row)
# 關閉連接
conn.close()
首先,你需要安裝這個模塊:
pip install psycopg2
然后,你可以使用以下代碼連接到PostgreSQL數據庫:
import psycopg2
# 連接到PostgreSQL數據庫
conn = psycopg2.connect(
host='localhost',
user='your_username',
password='your_password',
dbname='your_database'
)
# 創建一個游標對象
cursor = conn.cursor()
# 執行SQL命令
cursor.execute('CREATE TABLE IF NOT EXISTS users (id SERIAL PRIMARY KEY, name VARCHAR(255))')
cursor.execute("INSERT INTO users (name) VALUES ('Alice')")
cursor.execute("INSERT INTO users (name) VALUES ('Bob')")
# 提交更改
conn.commit()
# 查詢數據
cursor.execute('SELECT * FROM users')
rows = cursor.fetchall()
for row in rows:
print(row)
# 關閉連接
conn.close()
注意:在實際應用中,請確保不要將敏感信息(如數據庫憑據)硬編碼到腳本中。最好使用環境變量或其他安全方法來存儲這些信息。