from openpyxl import load_workbook

fn = r'd:\tongxunlu.xlsx'


# 打印操作菜單
def showMenu():
? ? print("========歡迎使用通訊錄管理系統(tǒng)========")
? ? print(" ? ? 1.添加")
? ? print(" ? ? 2.修改")
? ? print(" ? ? 3.刪除")
? ? print(" ? ? 4.查詢")
? ? print(" ? ? 5.排序")
? ? print(" ? ? 6.退出")
? ? print("==================================")


# 打開文件
def openFile():
? ? wb = load_workbook(fn)
? ? ws = wb.active
? ? return ws, wb


# 判斷手機(jī)號(hào)是否合法
def isLegal(s):
? ? if len(s) == 11 and s.isdigit():
? ? ? ? return True
? ? else:
? ? ? ? return False


# 添加手機(jī)號(hào)
def addData():
? ? data = input("輸入姓名和號(hào)碼（空格分開）:").split(" ")
? ? for i in range(len(data), 1, -1):?? ??? ??? ??? ?# 從最后一個(gè)手機(jī)號(hào)開始
? ? ? ? s = isLegal(data[i - 1])
? ? if not s:
? ? ? ? print("手機(jī)號(hào)碼必須為11位數(shù)字！")
? ? ? ? addData()
? ? else:
? ? ? ? w1, w2 = openFile()
? ? ? ? w1.append(data)
? ? ? ? print("添加成功！")
? ? ? ? w2.save(fn)


# 更改手機(jī)號(hào)
def changeData():
? ? flag = False
? ? data = input("請(qǐng)輸入需要修改的姓名和新的手機(jī)號(hào)碼（空格分開）:").split(" ")
? ? if not isLegal(data[1]):
? ? ? ? print("手機(jī)號(hào)碼必須為11位數(shù)字！")
? ? ? ? changeData()
? ? else:
? ? ? ? w1, w2 = openFile()
? ? ? ? for row in w1.rows:
? ? ? ? ? ? if row[0].value == data[0]:
? ? ? ? ? ? ? ? flag = True
? ? ? ? ? ? ? ? row[1].value = data[1]
? ? ? ? if not flag:
? ? ? ? ? ? print("通訊錄中沒有此人！")
? ? ? ? ? ? changeData()
? ? ? ? else:
? ? ? ? ? ? print("修改成功！")
? ? ? ? ? ? w2.save(fn)


# 刪除人
def delete():
? ? flag = False
? ? x = 0
? ? name = input("請(qǐng)輸入要?jiǎng)h除的姓名:")
? ? w1, w2 = openFile()
? ? for row in w1.rows:
? ? ? ? x += 1
? ? ? ? if row[0].value == name:
? ? ? ? ? ? flag = True
? ? ? ? ? ? w1.delete_rows(x)
? ? if not flag:
? ? ? ? print("通訊錄中沒有此人！")
? ? else:
? ? ? ? print("刪除成功！")
? ? ? ? w2.save(fn)


# 查找手機(jī)號(hào)或者姓名
def search():
? ? flag = False
? ? data = input("請(qǐng)輸入姓名或手機(jī)號(hào)碼:")
? ? w1, w2 = openFile()
? ? for row in w1.rows:
? ? ? ? if row[0].value == data:
? ? ? ? ? ? flag = True
? ? ? ? ? ? print(row[1].value)
? ? ? ? ? ? if row[2].value is not None:
? ? ? ? ? ? ? ? print(row[2].value)
? ? ? ? elif row[1].value == data or row[2].value == data:
? ? ? ? ? ? flag = True
? ? ? ? ? ? print(row[0].value)
? ? if not flag:
? ? ? ? print("通訊錄中沒有此人或手機(jī)號(hào)！")
? ? ? ? search()


# 排序
def sort():
? ? s = input("請(qǐng)輸入正序或逆序(正序輸入F,逆序輸入T):")
? ? if s == "F":
? ? ? ? flag = False
? ? elif s == "T":
? ? ? ? flag = True
? ? data = []
? ? w1, w2 = openFile()
? ? for row in w1.rows:
? ? ? ? data1 = []
? ? ? ? for cell in row:
? ? ? ? ? ? data1.append(cell.value)
? ? ? ? data.append(data1)
? ? data.sort(key=lambda x: x[0], reverse=flag)
? ? w2.remove(w1)
? ? w1 = w2.create_sheet()
? ? for i in data:
? ? ? ? w1.append(i)
? ? w2.save(fn)


# 循環(huán)函數(shù)
while True:
? ? showMenu()
? ? num = int(input("請(qǐng)輸入操作序號(hào):"))
? ? if num == 1:
? ? ? ? addData()
? ? elif num == 2:
? ? ? ? changeData()
? ? elif num == 3:
? ? ? ? delete()
? ? elif num == 4:
? ? ? ? search()
? ? elif num == 5:
? ? ? ? sort()
? ? elif num == 6:
? ? ? ? print("再見！")
? ? ? ? break
? ? else:
? ? ? ? print("輸入的序號(hào)不在范圍內(nèi)！")