定义一个函数,用来实现 连接数据库并执行SQL语句
时间: 2024-02-20 11:02:02 浏览: 30
好的,以下是一个示例函数,可以连接 MySQL 数据库,并执行一条 SQL 语句:
```python
import pymysql
def execute_sql(sql):
# 连接数据库
conn = pymysql.connect(
host='localhost',
port=3306,
user='root',
password='password',
database='test_db'
)
# 创建游标
cursor = conn.cursor()
try:
# 执行 SQL 语句
cursor.execute(sql)
# 提交事务
conn.commit()
# 获取结果
result = cursor.fetchall()
return result
except Exception as e:
# 发生错误时回滚
conn.rollback()
print(f"执行 SQL 时发生错误:{e}")
finally:
# 关闭游标和连接
cursor.close()
conn.close()
```
你可以把要执行的 SQL 语句作为参数传递给这个函数,它会返回执行结果。当然,你需要根据实际情况修改函数中的数据库连接参数。