python学生管理系统代码实现

class Student:  stuID = ""  name = ""  sex = "M"  classID = "NULL"    #set ID  def setStuID(self,stuID):   self.stuID = stuID    def setName(self,name):   self.name = name    def setSex(self,sex):   self.sex = sex     def setClassID(self,classID):   self.classID = classID    def getStuId(self):   return self.stuID    def getName(self):   return self.name    def getSex(self):   return self.sex    def getClassID(self):   return self.classID 

import os import re import sys import string import student  #save file global FILEPATH FILEPATH = "student.db" #temporary file global TEMPFILE TEMPFILE = "temp.db"  #This is menu def menu():  while True :   print "1.Add a student information"   print "2.Query student information"   print "3.Delete a student information"   print "0.Quit"   opt = raw_input("Select:")   if opt == "1":    while True:     addStudent()     opt2 = raw_input("Continue Add(Y/N)?:")     if opt2 == "Y" or opt2 == "y" or opt2 == "":      continue     else:      break   elif opt == "2":    while True:     query()     opt2 = raw_input("Continue Query(Y/N)?:")     if opt2 == "Y" or opt2 == "y" or opt2 == "":      continue     else:      break      elif opt == "3":    while True:     delMenu()     opt2 = raw_input("Continue Delete(Y/N)?:")     if opt2 == "Y" or opt2 == "y" or opt2 == "":      continue     else:      break      elif opt == "0" :    exitProgram()     break   else:    print "Error input"      #Add a student def addStudent() :  stu = student.Student()  while True:   stuID = raw_input("ID(001-999):")   #match ID 001-999   p = re.match("^[0-9]{3}$", stuID)   if p :    if stuID == "000":     print "ID must be 001-999"     continue    if isIDExist(stuID):     print "ID = %s already exist!" % stuID     continue    else :      stu.setStuID(stuID)    break   else:    print "ID must be 001-999"    while True:     stuName = raw_input("Name(a-z,A-Z,5 char):")   #match name a-z A-Z 5 char   p = re.match("^[a-zA-Z]{1,5}$",stuName)   if p :    stu.setName(stuName)    break   else :    print "Name format error must a-z,A-Z,within 5 char"    while True:    stuSex = raw_input("Sex(default is M):")   #default value   if stuSex == "":    stu.setSex("M")    print "Sex:M"    break   if stuSex =="M" or stuSex == "m":    stu.setSex(string.upper(stuSex))    break      #if stuSex == "M" or stuSex == "m" || stuSex == "F" stuSex == "f":   p = re.match("^M|m|F|f$",stuSex)   if p :    stu.setSex(string.upper(stuSex))    break   else :    print "Sex(M/f)"     while True:   stuClass = raw_input("Class(01-99):")   #default value   if stuClass == "":    stu.setClassID("NULL")    print "CLASS:NULL"    break   #match 00-99   p = re.match("^[0-9]{2}$",stuClass)   if p :    #get rid of 00    if stuClass == "00":     print "Class must 01-99"     continue    stu.setClassID(stuClass)    break   else:    print "Class must 01-99"  #save to file    file1 = open(FILEPATH,"a")  print "ID/tNAME/tSEX/tCLASS"  print stu.getStuId(),"/t",stu.getName(),"/t",stu.getSex(),"/t",stu.getClassID()  file1.write(stu.getStuId()+"/t"+stu.getName()+"/t"+stu.getSex()+"/t"+stu.getClassID()+"/n")  print "Add student success!"  file1.close()  #Delete student menu  def delMenu():  print "1.Delete by ID"  print "2.Delete contains ID"  opt = raw_input("Select:")  if opt == "1":   delStudentByID()  elif opt == "2":   delStudentContainsID()  else:   print "Error input"     #Delete contains id   def delStudentContainsID():  contID = raw_input("ID:")  if getInfoContainsID(contID)==0 :   print "Can't find ID contains /"%s/" student" % contID   return  opt = raw_input("Are you sure delete all(Y/N):")  if not opt == "y" or opt == "Y":   return   f = open(FILEPATH,"r")  tmp = open(TEMPFILE,"a")  i=0  for eachLine in f:   items = eachLine.split("/t") #  if not re.match(contID, items[0]):   if items[0].count(contID) ==0:    tmp.write(eachLine)   else:    i+=1   f.close()  tmp.close()  os.remove(FILEPATH)  os.rename(TEMPFILE, FILEPATH)  print "Deleted %d data" % i  #get contains ID information     def getInfoContainsID(stuID):  f = open(FILEPATH)  i=0  for eachLine in f:   items = eachLine.split("/t")      if not items[0].count(stuID) ==0: #  if re.match(stuID,items[0]):    i+=1    if i==1:     print "ID/tNAME/tSEX/tCLASS"    print eachLine,  if i==0:   return 0  else :   return i  f.close()    #Delete student by ID  def delStudentByID():  delID = raw_input("Delete ID:")  if not isIDExist(delID) :   print "Can't find ID = %s student information" % delID   return    getInfoByID(delID)  opt = raw_input("Are you sure delete it(Y/N):")  if not (opt == "Y" or opt == "y"):   return    f = open(FILEPATH,"r")  tmp = open(TEMPFILE,"a")    for eachLine in f:   split = eachLine.split("/t")     if not delID == split[0]:    tmp.write(eachLine)       tmp.close()  f.close()  os.remove(FILEPATH)  os.rename(TEMPFILE, FILEPATH)  print "Delete success!"  #Query menu  def query():  print "1.Query student by ID"  print "2.Query all students"  opt = raw_input("Select:")  if opt == "1":   queryByID()  elif opt == "2":   queryAll()  #default is query all   elif opt == "":   queryAll()   else :   print "Error Input!"     #query ID exist def isIDExist(ID):  f = open(FILEPATH)  flag = 0  for eachline in f:   temp = eachline.split("/t")   if temp[0] == ID:    flag+=1  f.close()  if flag == 0 :   return False  else:   return True   #get information by ID for delete student  def getInfoByID(stuID):  f = open(FILEPATH)  i=0  for eachLine in f:   items = eachLine.split("/t")   if items[0] == stuID:    i+=1    if i==1:     print "ID/tNAME/tSEX/tCLASS"    print eachLine   f.close()     #Query student by ID def queryByID():  queryID = raw_input("ID:")  f = open(FILEPATH) # lines = f.readlines() # print lines[0].strip() # for line in lines: #  a = line.split() #  if queryID == a[1].strip(): #   print line.strip()  flag = 0  for eachline in f:   #split by "/t" get as C array temp[4]   temp = eachline.split("/t") #  print temp[0] , temp[1] , temp[2] , temp[3]   if temp[0] == queryID:    flag+=1    if flag == 1:     print "ID/tNAME/tSEX/tCLASS"    print eachline,  if flag == 0 :   print "Can't find ID = %s student information" % queryID    f.close() # f = open(FILEPATH) # readLines = f.readlines()   # for eachLine in f: #  eachLine #     #  if eachLine == queryID: #   print eachLine # f.close()  #Query all students  def queryAll():  f = open(FILEPATH,"r")  i=0  for eachLine in f:   i+=1   if i==1:    print "ID/tNAME/tSEX/tCLASS"   print eachLine,  f.close()  if i==0:   print "No data!"  def exitProgram():  print "Thank you Bye!" # sys.exit()   def init():  if os.path.exists(FILEPATH) :   print "Load file successful"  else:   try:    f = open(FILEPATH,"w")   except Exception:    print "Can't open file"    sys.exit()   finally:    f.close()   if __name__ == '__main__':  init()  menu()