在Python中关联数据库通常需要以下步骤:
安装数据库驱动程序
对于MySQL,可以使用`mysql-connector-python`或`pymysql`。
对于PostgreSQL,可以使用`psycopg2`。
对于SQLite,Python的标准库中已经内置了所需的驱动程序。
导入数据库模块
import pymysql 对于MySQL
或者
import psycopg2 对于PostgreSQL
或者
import sqlite3 对于SQLite
创建数据库连接
MySQL示例
conn = pymysql.connect(
host='localhost',
user='username',
password='password',
database='database_name',
charset='utf8mb4'
)
PostgreSQL示例
conn = psycopg2.connect(
host='localhost',
user='username',
password='password',
dbname='database_name',
port='5432'
)
SQLite示例
conn = sqlite3.connect('database.db')
创建游标对象
cursor = conn.cursor()
执行SQL语句
创建表
cursor.execute('CREATE TABLE IF NOT EXISTS users (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255), email VARCHAR(255))')
插入数据
cursor.execute("INSERT INTO users (name, email) VALUES (%s, %s)", ('John Doe', ''))
查询数据
cursor.execute('SELECT * FROM users')
更新数据
cursor.execute("UPDATE users SET email = %s WHERE id = %s", ('', 1))
删除数据
cursor.execute('DELETE FROM users WHERE id = %s', (1,))
提交或回滚事务 (如果适用):
提交事务
conn.commit()
回滚事务
conn.rollback()
关闭游标和连接
cursor.close()
conn.close()
请根据您使用的数据库类型选择合适的模块和连接参数。