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()