试用研究生信息管理系统的功能与实现

import sqlite3

# 连接数据库

conn = sqlite3.connect('graduate.db')

cursor = conn.cursor()

# 创建表

cursor.execute('''

CREATE TABLE IF NOT EXISTS students (

id INTEGER PRIMARY KEY AUTOINCREMENT,

name TEXT NOT NULL,

major TEXT NOT NULL,

year INTEGER NOT NULL

)

''')

]]>

def add_student(name, major, year):

cursor.execute('''

INSERT INTO students (name, major, year)

VALUES (?, ?, ?)

''', (name, major, year))

conn.commit()

]]>

def get_student(student_id):

cursor.execute('SELECT * FROM students WHERE id = ?', (student_id,))

return cursor.fetchone()

]]>