基于研究生管理信息系统的数据下载与实现

import pymysql

def fetch_data():

# 连接数据库

connection = pymysql.connect(

host='localhost',

user='root',

password='password',

database='grad_management'

)

try:

with connection.cursor() as cursor:

sql = "SELECT id, name, grade FROM students"

cursor.execute(sql)

results = cursor.fetchall()

return results

finally:

connection.close()

data = fetch_data()

print(data)

]]>

from openpyxl import Workbook

def export_to_excel(data):

wb = Workbook()

ws = wb.active

# 设置表头

headers = ['ID', 'Name', 'Grade']

ws.append(headers)

# 写入数据

for row in data:

ws.append(row)

# 保存文件

wb.save("students.xlsx")

export_to_excel(data)

]]>