Update 学生管理.py

学生管理.py

@@ -59,13 +59,14 @@ def read_someone_by_phone():
             print_message("【{}】学生信息查询成功！".format(stu_phone))
             display(stu)
             break
-# 通过姓名/电话查询某个学生信息
+# 通过姓名/电话/学号查询某个学生信息
 def read_stu():
     stus = read_stu_csv()
     stu_names = stus['姓名'].values.tolist()
     stu_phones = stus['电话'].values.tolist()
+    stu_IDs = stus['学号ID'].values.tolist()
     while True:
-        stu_input = input("请输入学生姓名/电话：")
+        stu_input = input("请输入学生姓名/电话/学号：")
         if stu_input in stu_names:
             stu = stus.loc[(stus['姓名'] == stu_input),:]
             display(stu)
@@ -73,6 +74,10 @@ def read_stu():
         if int(stu_input) in stu_phones:
             stu = stus.loc[(stus['电话'] == int(stu_input)),:] 
             display(stu)
+            break      
+        if int(stu_input) in stu_IDs:
+            stu = stus.loc[(stus['学号ID'] == int(stu_input)),:] 
+            display(stu)
             break
         else:
             print_message("未查询到【{}】信息，请重新输入".format(stu_input))
@@ -129,6 +134,21 @@ def add_stu():
         print_message("添加学生【{}】成功".format(stu_name))
         display(stu_df)
         break
+#通过学号查询学生信息
+def read_stu_by_number():
+    stus = read_stu_csv()
+
+    stu_IDs = stus['学号ID'].values.tolist()
+    while True:
+        stu_input = input("请输入学生学号：")
+        
+        if int(stu_input) in stu_IDs:
+            stu = stus.loc[(stus['学号ID'] == int(stu_input)),:] 
+            display(stu)
+            break
+        else:
+            print_message("未查询到【{}】信息，请重新输入".format(stu_input))
+            continue
 #通过姓名修改某个学生姓名
 def update_by_name():
     stus = read_stu_csv()
@@ -298,35 +318,39 @@ def del_stu():
             break          
         else:
             print('重新输入姓名')
-print('你想查询的内容')
+print('   你想查询的内容')
 print('0.查询全部学生信息')
 print('1.通过姓名查找学生信息')
 print('2.通过电话查找学生信息')
-print('3.通过姓名/电话查找学生信息')
-print('4.通过姓名修改学生信息')
-print('4.通过电话修改学生信息')
-print('6.添加学生信息')
-print('7.更新学生信息')
-print('8.删除学生信息')
-x=eval(input('请输入：'))
-if x==1:
-    read_someone_by_name()
-elif x==2:
-    read_someone_by_phone()
-elif x==3:
-    read_stu()
-elif x==4:                
-    update_by_name()
-elif x==5:
-    update_by_phone()    
-elif x==6:
-    add_stu()
-elif x==7:
-    update_stu()
-elif x==0:
-    read_all_stus()
-elif x==8:
-    del_stu()
-else:
-    print('输入错误，重新输入')   
-
+print('3.根据学号查找学生信息')
+print('4.通过姓名/电话/学号查找学生信息')
+print('5.通过姓名修改学生信息')
+print('6.通过电话修改学生信息')
+print('7.添加学生信息')
+print('8.更新学生信息')
+print('9.删除学生信息')
+while True:
+    x=eval(input('请输入：'))
+    if x==1:
+        read_someone_by_name()
+    elif x==2:
+        read_someone_by_phone()
+    elif x==3:
+        read_stu()
+    elif x==4:                
+        update_by_name()
+    elif x==5:
+        update_by_phone()    
+    elif x==6:
+        add_stu()
+    elif x==7:
+        update_stu()
+    elif x==0:
+        read_all_stus()
+    elif x==8:
+        del_stu()
+    elif x==9:
+        read_stu_by_number()
+    else:
+        print('输入错误，重新输入')   
+