用python代码做一个公共数据库处理函数
时间: 2024-02-01 08:04:08 浏览: 137
用Python实现数据库编程
好的,以下是一个用Python处理公共数据库的函数示例:
```python
import mysql.connector
def process_database(query):
# 连接到数据库
mydb = mysql.connector.connect(
host="localhost",
user="username",
password="password",
database="database_name"
)
# 创建游标对象
mycursor = mydb.cursor()
# 执行查询语句
mycursor.execute(query)
# 提交更改
mydb.commit()
# 获取结果
result = mycursor.fetchall()
# 关闭游标和数据库连接
mycursor.close()
mydb.close()
return result
```
该函数接受一个 SQL 查询语句作为参数,并返回查询结果。需要注意的是,该函数需要先安装 MySQL Connector Python 模块。可以使用以下命令进行安装:
```python
pip install mysql-connector-python
```
使用该函数的示例代码如下:
```python
query = "SELECT * FROM customers WHERE country = 'China'"
result = process_database(query)
for row in result:
print(row)
```
该示例代码将查询 `customers` 表中来自中国的客户,并将结果打印出来。
阅读全文