import sqlite3

def init_db():

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

cursor = conn.cursor()

# 創(chuàng)建students表

cursor.execute('''

CREATE TABLE IF NOT EXISTS students (

id INTEGER PRIMARY KEY AUTOINCREMENT,

student_id TEXT UNIQUE,

name TEXT,

major TEXT

)

''')

# 創(chuàng)建courses表

cursor.execute('''

CREATE TABLE IF NOT EXISTS courses (

id INTEGER PRIMARY KEY AUTOINCREMENT,

course_id TEXT UNIQUE,

title TEXT,

credits INTEGER

)

''')

# 創(chuàng)建enrollments表

cursor.execute('''

CREATE TABLE IF NOT EXISTS enrollments (

id INTEGER PRIMARY KEY AUTOINCREMENT,

student_id TEXT,

course_id TEXT,

FOREIGN KEY(student_id) REFERENCES students(student_id),

FOREIGN KEY(course_id) REFERENCES courses(course_id)

)

''')

conn.commit()

conn.close()

if __name__ == "__main__":

init_db()

]]>

def add_student(student_id, name, major):

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

cursor = conn.cursor()

try:

cursor.execute('INSERT INTO students (student_id, name, major) VALUES (?, ?, ?)',

(student_id, name, major))

conn.commit()

print("Student added successfully!")

except sqlite3.IntegrityError:

print("Student ID already exists!")

智慧校園服務平臺

conn.close()

]]>