使用Python的pymssql库实现多条数据插入的示例代码
发布时间:2023-12-11 13:57:23
pymssql是一个用于连接和操作Microsoft SQL Server的Python库。它提供了多个方法来实现数据插入,包括单条插入和批量插入。以下是一个示例代码,演示了如何使用pymssql库实现多条数据插入。
首先,你需要安装pymssql库。你可以使用以下命令来安装:
pip install pymssql
接下来,让我们来看一个实例,演示如何使用pymssql库实现多条数据插入。
import pymssql
def insert_records():
# 连接到数据库
conn = pymssql.connect(host='localhost', user='username', password='password', database='dbname')
cursor = conn.cursor()
# 定义要插入的数据
data = [
('John Doe', 'john.doe@example.com'),
('Jane Smith', 'jane.smith@example.com'),
('Bob Johnson', 'bob.johnson@example.com')
]
try:
# 执行批量插入
cursor.executemany("INSERT INTO users (name, email) VALUES (%s, %s)", data)
conn.commit()
print("数据插入成功!")
except Exception as e:
conn.rollback()
print("数据插入失败:", e)
# 关闭数据库连接
conn.close()
if __name__ == '__main__':
insert_records()
在上面的代码中,我们首先导入了pymssql库,然后定义了一个insert_records函数来执行数据插入。
在函数内部,我们首先使用pymssql.connect函数连接到数据库。你需要根据你的数据库配置修改连接参数。
然后,我们定义了一个包含多条数据的列表,即data。每个元组表示一条数据的字段值。
接下来,我们使用cursor.executemany方法执行批量插入。该方法接受两个参数:SQL语句和数据列表。SQL语句中的占位符%s将替换为数据列表中的值。
在try块中,我们使用conn.commit方法提交事务,将数据保存到数据库中。
如果发生异常,我们使用conn.rollback方法回滚事务,撤销之前的操作。
最后,我们使用conn.close方法关闭数据库连接。
运行上面的代码,将会将包含多条数据的data列表中的数据插入到数据库中。
这是使用pymssql库实现多条数据插入的一个示例。你可以根据自己的需求修改代码,来适应不同的数据插入场景。
