在Python中,你可以使用pymysql
模块来初始化MySQL连接并实现函数化。下面是一个简单的示例:
首先,确保你已经安装了pymysql
模块:
pip install pymysql
然后,可以创建一个db_utils.py
文件,并添加以下代码:
import pymysql
def get_connection():
# 创建数据库连接
conn = pymysql.connect(
host='localhost', # 数据库主机地址
user='your_username', # 用户名
password='your_password', # 密码
database='your_database' # 数据库名称
)
return conn
def close_connection(conn):
# 关闭数据库连接
conn.close()
def execute_query(sql):
try:
conn = get_connection()
cursor = conn.cursor()
cursor.execute(sql)
result = cursor.fetchall()
close_connection(conn)
return result
except Exception as e:
print(f"Error executing query: {e}")
接下来,你可以在其他脚本中导入这个函数库,并使用其中的函数进行MySQL连接和查询操作。例如:
from db_utils import execute_query
# 示例查询语句:获取所有用户信息
query = "SELECT * FROM users"
result = execute_query(query)
# 打印结果
for row in result:
print(row)
请注意,在上述示例中,需要替换相应的数据库主机地址、用户名、密码和数据库名称。
这样,你就可以通过调用execute_query()
函数来执行自定义的SQL查询语句并获取结果了。
内容由零声教学AI助手提供,问题来源于学员提问