#-*- coding:utf-8 -*-
import sqlite3
#打開(kāi)本地?cái)?shù)據(jù)庫(kù)用于存儲(chǔ)用戶(hù)信息
conn = sqlite3.connect('student.db')

#在該數(shù)據(jù)庫(kù)下創(chuàng)建學(xué)生信息表
conn.execute ('''CREATE TABLE StudentTable(
 ID INTEGER PRIMARY KEY AUTOINCREMENT,
 StuId INTEGER NOT NULL,
 NAME TEXT NOT NULL,
 CLASS INT NOT NULL);''')
print "Table created successfully";

#在該數(shù)據(jù)庫(kù)下創(chuàng)建課程信息表
conn.execute ('''CREATE TABLE CourseTable(
 ID INTEGER PRIMARY KEY AUTOINCREMENT,
 CourseId INT NOT NULL,
 Name TEXT NOT NULL,
 Teacher TEXT NOT NULL,
 Classroom TEXT NOT NULL,
 StartTime CHAR(11) NOT NULL,
 EndTime CHAR(11) NOT NULL);''')
print "Table created successfully";

#在該數(shù)據(jù)庫(kù)下創(chuàng)建選課情況信息表
conn.execute ('''CREATE TABLE XuankeTable(
 ID INTEGER PRIMARY KEY AUTOINCREMENT,
 StuId INT NOT NULL,
 CourseId INT NOT NULL,
 StudentNAME TEXT NULL,
 StudenCourse TEXT NULL);''')
print "Table created successfully";

#以上三個(gè)表創(chuàng)建完后，再次運(yùn)行程序時(shí)，需要把三個(gè)建表代碼注釋掉，否則會(huì)提示:該表已存在。即建表只需建一次。

def insert_stu(): #錄入學(xué)生信息
 conn = sqlite3.connect('student.db')
 stu_id = input("請(qǐng)輸入學(xué)生學(xué)號(hào):")
 cursor = conn.execute("SELECT StuId from StudentTable where StuId = '%s';"%stu_id)
 conn.commit()
 for row in cursor:
 if stu_id == row[0]:
 print "sorry,該學(xué)號(hào)已存在,請(qǐng)重新輸入"
 break
 else:
 stu_name = raw_input("請(qǐng)輸入學(xué)生姓名:")
 stu_class = input("請(qǐng)輸入學(xué)生班級(jí):")
 sql1 = "INSERT INTO StudentTable(StuId,NAME,CLASS)"
 sql1 += " VALUES(%d,'%s',%d);"%(stu_id,stu_name,stu_class)
 conn.execute(sql1)
 conn.commit()
 print "恭喜你,學(xué)生錄入成功!"

def xuanke(): #學(xué)生選課
 stu_id = input('請(qǐng)輸入要選課的學(xué)生學(xué)號(hào):')
 sql2 = "select StuId from StudentTable where StuId = %d;"%(stu_id)
 cursor1 = conn.execute(sql2)
 for row in cursor1:
 if stu_id == row[0]:
 sql3 = "select CourseId,Name,Teacher,Classroom, StartTime,EndTime from CourseTable"
 cursor2 = conn.execute(sql3)
 for row in cursor2:
 print "CourseId = ", row[0]
 print "Name = ", row[1]
 print "Teacher = ", row[2]
 print "Classroom = ",row[3]
 print "StartTime = ",row[4]
 print "EndTime = ",row[5], "\n"
 cou_id = input("請(qǐng)輸入要選的課程號(hào):")
 sql = "select StuId from XuankeTable where CourseId = %d;"%(cou_id)
 cursor3= conn.execute(sql)
 for row in cursor3:
 if stu_id == row[0]:
 print "該課程已選,請(qǐng)重新輸入要選課程!"
 else:
 sql3 = "insert into XuankeTable (StuId,CourseId) values (%d,%d)"%(stu_id,cou_id)
 cursor4= conn.execute(sql3)
 conn.commit()
 print "恭喜你，選課成功!"
 break
 break
 break
 else:
 print "sorry,沒(méi)有該學(xué)生號(hào)"

def stu_id_search():#按照學(xué)生學(xué)號(hào)查詢(xún)學(xué)生信息
 conn = sqlite3.connect('student.db')
 search_stu_id = input("請(qǐng)輸入要查詢(xún)的學(xué)號(hào):")
 sql4 = "SELECT StuId from StudentTable where StuId= %d;" % (search_stu_id)
 cursor1 = conn.execute(sql4)
 conn.commit()
 for row in cursor1:
 if search_stu_id == row[0]:
 sql10 = "select ID,StuId,NAME, CLASS from StudentTable where StuId = %d;"%(search_stu_id)
 cursor2 = conn.execute(sql10)
 conn.commit()
 for row in cursor2:
 print
 print "您要查詢(xún)的學(xué)生信息為:"
 print "ID = ", row[0]
 print "StuId = ", row[1]
 print "NAME = ", row[2]
 print "CLASS = ",row[3], "\n"
 break
 else:
 print "sorry,沒(méi)有該學(xué)生信息！"

def stu_id_cou(): #按照學(xué)生學(xué)號(hào)查詢(xún)?cè)搶W(xué)生所選課程
 stu_id = input("請(qǐng)輸入要查詢(xún)學(xué)生號(hào):")
 sql5 = "select StuId from StudentTable where StuId = %d;"%(stu_id)
 cursor = conn.execute(sql5)
 for row in cursor:
 if stu_id == row[0]:
 sql6 = "select CourseId from XuankeTable where StuId = %d;"%(stu_id)
 cursor = conn.execute(sql6)
 conn.commit()
 for row in cursor:
 print
 print "該學(xué)生所選課程號(hào)為:"
 print row
 print
 break
 else:
 print "sorry,沒(méi)有該學(xué)生選課信息!"

def cou_id_search(): #按照課程號(hào)查詢(xún)課程信息
 cou_id = input("請(qǐng)輸入要查詢(xún)的課程號(hào):")
 sql7 = "select CourseId ,Name,Teacher,Classroom,StartTime,EndTime from CourseTable "
 sql7 += "where CourseId = %d;"%(cou_id)
 cursor1 = conn.execute(sql7)
 conn.commit()
 for row in cursor1:
 print "您要查詢(xún)的課程信息為:"
 print "CourseId = ",row[0]
 print "Name = ", row[1]
 print "Teacher = ", row[2]
 print "Classroom = ",row[3]
 print "StartTime = " ,row[4]
 print "EndTime = ",row[5],"\n"
 break
 else:
 print "sorry,沒(méi)有該課程信息!"

def cou_id_stu():#按照課程號(hào)查詢(xún)選擇該課程的學(xué)生列表
 cou_id = input('請(qǐng)輸入課程號(hào):')
 sql8 = "select CourseId from XuankeTable where CourseId =%d;"%(cou_id)
 cursor1 = conn.execute(sql8)
 for row in cursor1:
 if cou_id == row[0]:
 sql9 = "select StuId from XuankeTable where CourseId =%d;"%(cou_id)
 cursor2 = conn.execute(sql9)
 conn.commit()
 for row in cursor2:
 print
 print "選擇該課程的學(xué)生為:"
 print row,"\n"
 break
 break
 else:
 print "sorry,沒(méi)有該課程信息!"

def menu():
 print '1.進(jìn)入學(xué)生信息系統(tǒng)(學(xué)生信息錄入)'
 print '2.進(jìn)入學(xué)生選課系統(tǒng)(學(xué)生選課操作)'
 print '3.進(jìn)入學(xué)生選課信息系統(tǒng)(學(xué)生信息查詢(xún)和選課情況查詢(xún))'
 print '4.退出程序'

def student():
 print '1.錄入學(xué)生信息'
 print '2.返回主菜單'
 print '3.退出程序'

def Course():
 print '1.開(kāi)始選課'
 print '2.返回主菜單'
 print '3.退出程序'

def information():
 print '1.按學(xué)號(hào)查詢(xún)學(xué)生信息'
 print '2.按學(xué)號(hào)查看學(xué)生選課課程列表'
 print '3.按課程號(hào)查看課程信息'
 print '4.按課程號(hào)查看選課學(xué)生列表'
 print '5.返回主菜單'
 print '6.退出程序'

while True:
 menu()
 print
 x = raw_input('請(qǐng)輸入您的選擇菜單號(hào):')
 if x == '1':
 #進(jìn)入學(xué)生信息系統(tǒng)
 student()
 stu = raw_input('您已進(jìn)入學(xué)生錄入系統(tǒng),請(qǐng)?jiān)俅屋斎脒x擇菜單:')
 if stu == '1':
 insert_stu()
 continue
 if stu == '2':
 menu()
 continue
 if stu == '3':
 print "謝謝使用！"
 exit()
 continue
 else:
 print "輸入的選項(xiàng)不存在，請(qǐng)重新輸入！"
 continue

 if x == '2':
 #進(jìn)入選課信息系統(tǒng)
 Course()
 cou = raw_input('您已進(jìn)入學(xué)生選課系統(tǒng),請(qǐng)?jiān)俅屋斎脒x擇菜單:')
 if cou == '1':
 xuanke()
 continue
 if cou == '2':
 menu()
 continue
 if cou == '3':
 print "謝謝使用！"
 exit()
 continue
 else:
 print "輸入的選項(xiàng)不存在，請(qǐng)重新輸入！"
 continue

 if x == '3':
 #進(jìn)入學(xué)生選課信息表
 information()
 inf = raw_input('您已進(jìn)入學(xué)生選課信息系統(tǒng),請(qǐng)?jiān)俅屋斎脒x擇菜單:')
 if inf == '1':
 stu_id_search()
 continue
 if inf == '2':
 stu_id_cou()
 continue
 if inf == '3':
 cou_id_search()
 continue
 if inf == '4':
 cou_id_stu()
 continue
 if inf == '5':
 menu()
 continue
 if inf == '6':
 print "謝謝使用！"
 exit()
 continue
 else:
 print "輸入的選項(xiàng)不存在，請(qǐng)重新輸入！"
 continue

 if x == '4':
 print "謝謝使用！"
 exit()
 else:
 print "輸入的選項(xiàng)不存在，請(qǐng)重新輸入！"
 continue