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)