在Python中加载数据库通常涉及以下步骤:
导入模块
对于MySQL,可以使用`pymysql`或`mysql.connector`。
对于SQLite,可以直接使用Python内置的`sqlite3`模块。
建立数据库连接
使用连接函数,如`connect()`,传入数据库地址、用户名、密码等参数。
创建游标对象
使用连接对象的`cursor()`方法创建一个游标对象,用于执行SQL语句。
执行SQL语句
使用游标对象的`execute()`方法执行SQL语句,如查询、插入、删除等操作。
获取数据结果
使用游标对象的`fetchall()`、`fetchmany()`或`fetchone()`方法获取SQL语句执行的结果集。
提交或回滚事务
若执行的是更新、插入、删除等操作,需要使用连接对象的`commit()`方法提交事务。
若执行中出现错误,可以使用`rollback()`方法回滚事务。
关闭游标和连接
使用游标对象的`close()`方法关闭游标。
使用连接对象的`close()`方法关闭数据库连接。
import pymysql
建立数据库连接
db = pymysql.connect(
host='localhost',
user='username',
password='password',
db='database_name',
port=3306,
charset='utf8mb4'
)
创建游标对象
cursor = db.cursor()
创建数据库(如果不存在)
cursor.execute('CREATE DATABASE IF NOT EXISTS database_name')
插入数据
sql = "INSERT INTO table_name (column1, column2) VALUES (%s, %s)"
values = ('value1', 'value2')
cursor.execute(sql, values)
提交事务
db.commit()
关闭游标和连接
cursor.close()
db.close()
请根据你的具体需求调整数据库连接参数和SQL语句。如果你使用的是SQLite或其他数据库,步骤类似,只是导入的模块和连接参数会有所不同