def updateUser(user_info): username = user_info[1] try: keyword = user_info[2] field = user_info[3] new_value = user_info[-1] mark = user_info[-2] if keyword != "set" or mark != "=": print("\033[0;31;1mupdate method error.\033[0m") else: result = loadUser.loadUser(USERFILE) usernames = result.keys() if username in usernames: if field == "age": result[username]["age"] = new_value saveUser.saveUser(result, USERFILE) print("\033[0;31;1mupdate {} age success.\033[0m".format( username)) elif field == "tel": result[username]["tel"] = new_value saveUser.saveUser(result, USERFILE) print("\033[0;31;1mupdate {} tel success.\033[0m".format( username)) elif field == "email": result[username]["email"] = new_value saveUser.saveUser(result, USERFILE) print("\033[0;31;1mupdate {} email success.\033[0m".format( username)) else: print( "\033[0;31;1mfield {} not found.\033[0m".format(field)) else: print("\033[0;31;1muser {} not found.\033[0m".format(username)) except Exception as e: print(e)
def deleteUser(user_info): username = user_info[1] result = loadUser.loadUser(USERFILE) usernames = result.keys() if username in usernames: result.pop(username) saveUser.saveUser(result, USERFILE) print("\033[0;31;1muser {} remove success.\033[0m".format(username)) msg = "user {} remove success.".format(username) logger.debug(msg) else: print("\033[0;31;1muser {} not found.\033[0m".format(username)) return result
def addUser(info_list): if len(info_list) != 5: print("Add info invaild, Please add again.") return False username = info_list[1] RESULT = loadUser.loadUser(USERFILE) if username in RESULT: print("Username {} already exists.".format(username)) return False else: RESULT[username] = dict(zip(FIELDS, info_list[1:])) print("Add {} success.".format(username)) saveUser.saveUser(RESULT, USERFILE) return True
def listUser(): try: result = loadUser.loadUser(USERFILE) xtb = PrettyTable() xtb.field_names = ["name", "age", "tel", "email"] for v in result.values(): name = v["name"] age = v["age"] tel = v["tel"] email = v["email"] xtb.add_row([name, age, tel, email]) print(xtb) except Exception as e: print(e)
def findUser(user_info): try: username = user_info[1] result = loadUser.loadUser(USERFILE) usernames = result.keys() xtb = PrettyTable() xtb.field_names = ["name", "age", "tel", "email"] if username in usernames: name = result[username]["name"] age = result[username]["age"] tel = result[username]["tel"] email = result[username]["email"] xtb.add_row([name, age, tel, email]) print(xtb) else: print("\033[0;31;1muser {} not found.\033[0m".format(username)) except Exception as e: print(e)
def displayUser(user_info): # dispaly page 2 pagesize 5 # default = 5 try: if len(user_info[1:]) >= 2 and len(user_info[1:]) <= 4: pagesize = 5 if len(user_info[1:]) == 2: if user_info[1] == "page": pagesize = 5 else: print("Display info invaild,Please input agein.") # continue else: if user_info[1] == "page" and user_info[3] == "pagesize": pagesize = int(user_info[-1]) else: print("Display info invaild,Please input agein.") # continue page = int(user_info[2]) - 1 RESULT = loadUser.loadUser(USERFILE) data = [] for k, v in RESULT.items(): data.append(v.values()) start = page * pagesize end = start + pagesize print("Start: {},End: {}".format(start, end)) xtb = PrettyTable() xtb.field_names = FIELDS for userinfo in data[start:end]: xtb.add_row(userinfo) print(xtb) else: print("Input info invaid,Please input again.") # continue except Exception as e: print(e)