import mysql.connector
# 创建连接
conn = mysql.connector.connect(
host="localhost",
user="root",
password="password",
database="grad_system"
)
cursor = conn.cursor()
# 创建表
cursor.execute("""
CREATE TABLE IF NOT EXISTS students (
id INT AUTO_INCREMENT PRIMARY KEY,
name VARCHAR(100) NOT NULL,
major VARCHAR(100),
year INT
)
""")
]]>
def add_student(name, major, year):
sql = "INSERT INTO students (name, major, year) VALUES (%s, %s, %s)"
val = (name, major, year)
cursor.execute(sql, val)
conn.commit()
print(cursor.rowcount, "record inserted.")
]]>
add_student("Alice", "Computer Science", 2023)
]]>
def view_students():
cursor.execute("SELECT * FROM students")
result = cursor.fetchall()
for row in result:
print(row)
]]>
view_students()
]]>
def update_major(student_id, new_major):
sql = "UPDATE students SET major = %s WHERE id = %s"
val = (new_major, student_id)
cursor.execute(sql, val)
conn.commit()
print(cursor.rowcount, "record(s) affected")
]]>
update_major(1, "Data Science")
]]>
