在Python中连接数据库通常遵循以下步骤:
导入模块
根据所使用的数据库类型,导入相应的Python库。例如,使用MySQL可以使用`pymysql`或`mysql.connector`,使用PostgreSQL可以使用`psycopg2`。
建立数据库连接
使用库提供的连接函数,传入数据库地址、用户名、密码等信息来建立连接。
创建游标对象
使用连接对象的`cursor()`方法创建一个游标对象,用于执行SQL语句。
执行SQL语句
使用游标对象的`execute()`方法执行SQL命令,如查询、插入、删除等。
获取数据结果
使用游标对象的`fetchone()`、`fetchall()`等方法获取SQL语句执行的结果集。
提交或回滚事务
对于更新、插入、删除等操作,使用连接对象的`commit()`方法提交事务。
若出现错误,可以使用`rollback()`方法回滚事务。
关闭游标和连接
使用游标对象的`close()`方法关闭游标,使用连接对象的`close()`方法关闭数据库连接。
```python
import pymysql
数据库连接信息
db_info = {
'host': 'localhost',
'user': 'username',
'password': 'password',
'database': 'database_name',
'port': 3306,
'charset': 'utf8mb4',
'autocommit': True
}
连接数据库
conn = pymysql.connect(db_info)
创建游标对象
cur = conn.cursor()
执行SQL语句
cur.execute("CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT, age INTEGER)")
插入数据
cur.execute("INSERT INTO users (name, age) VALUES (%s, %s)", ('Alice', 30))
提交事务
conn.commit()
关闭游标和连接
cur.close()
conn.close()
请根据你的具体需求选择合适的库和参数来连接数据库。