def ExportUser(self):
args = self.user_info_string
user_info_list = args.split(" ")
if args is '' or len(user_info_list) != 1:
user_expand.format_print(False,
"ExportUser failed, args length != 1")
return
# 配置标题
csv_user_list = [FIELDS]
# 查询数据库全部信息
user_data = student.select()
# 将数据转化为列表
user_list = [
user_data[values].__data__.items()
for values in range(len(user_data))
]
for j in range(len(user_list)):
db_user_data = dict(user_list[j])
db_user_list = [i for i in db_user_data.values()]
csv_user_list.append(db_user_list)
# 写入CSV文件
with open(args, 'w', newline='') as csv_file:
csv_writer = csv.writer(csv_file)
for list1 in csv_user_list:
csv_writer.writerow(list1)
user_expand.format_print(True, "文件导出完毕, 文件名为：{}".format(args))
def updateUser(self):
args = self.user_info_string
user_info_list = args.split()
if len(user_info_list) != 5:
user_expand.format_print(False,
"updateUser failed, args length != 5")
return
where = user_info_list[1]
symbol = user_info_list[-2]
if where != 'set' or symbol != '=':
user_expand.format_print(False, "syntax")
else:
username = user_info_list[0]
where_field = user_info_list[2]
update_value = user_info_list[-1]
result = self.findUser()
if not result:
return
try:
tag = student.update({
where_field: update_value
}).where(student.username == username).execute()
# 检查用户更新值是否与原值相同
if tag:
user_expand.format_print(
True, "user {} update secc".format(username))
else:
user_expand.format_print(False,
"兄弟你这也没更新啊！！".format(username))
except Exception as e:
print(e)
def main():
init_fail_count = 0
max_fail_count = 3
while init_fail_count < max_fail_count:
username = input("Please input your login username: "******"Please input your login password: "******"{} 登录成功".format(username))
#user_expand.format_print(True, "\n\tWelcome to user magage system.\n")
user_expand.user_help(username=username)
user_expand.logic()
else:
logs.save_log("{} 登录失败".format(username), tag='error')
user_expand.format_print(False,
"username or password valid failed.")
print()
init_fail_count += 1
user_expand.format_print(False, "Game Over.")
logs.save_log("密码错误次数过多，退出登录", tag='error')
def main():
init_fail_count = 0
max_fail_count = 3
while init_fail_count < max_fail_count:
try:
username = input("Please input your login username: "******"Please input your login password: "******"Please input your login password: "******"{} 登录成功".format(username))
user_expand.user_help(username=username, roles=roles)
user_expand.logic(roles)
else:
logs.save_log("{} 登录失败".format(username), tag='error')
user_expand.format_print(False, "username or password valid failed.")
print()
init_fail_count += 1
except Exception as e:
print(e)
user_expand.format_print(False, "操作异常，程序退出!!!")
exit(1)
user_expand.format_print(False,"Game Over.")
logs.save_log("密码错误次数过多，退出登录", tag='error')
def addUser(self):
args = self.user_info_string
'''
add monkey1 12 132xxx [email protected]
'''
user_info_list = args.split(" ")
if len(user_info_list) != 4:
logs.save_log("addUser failed, args length != 4", tag='error')
user_expand.format_print(False, "addUser failed, args length != 4")
return
username = user_info_list[0]
try:
student.select().where(student.username == username).get()
user_expand.format_print(False, "{} 已存在".format(username))
return
except:
RESULT = {
'username': username,
'age': user_info_list[1],
'phone': user_info_list[2],
'email': user_info_list[3],
}
try:
student.insert(RESULT).execute()
logs.save_log("add user {} secc.".format(username))
user_expand.format_print(True,
"add user {} secc.".format(username))
except Exception as e:
print(e)
def displayUser(self):
'''
display page 2 pagesize 5
:param args: page 2 pagesize 5 ;default pagesize = 5
page 1 -> 0-4
切片
slice
'''
args = self.user_info_string
try:
user_data = student.select()
values = [
user_data[values].__data__.items()
for values in range(len(user_data))
]
xtb = PrettyTable()
xtb.field_names = FIELDS
# 设置字段左对齐
xtb.align = 'l'
except Exception as e:
print(e)
return
user_info_list = args.split()
if len(user_info_list) == 2:
if user_info_list[0] != 'page':
user_expand.format_print(False, "eg: display page 2")
return
page_value = int(user_info_list[1]) - 1 # 1
pagesize = 5
start = page_value * pagesize
end = start + pagesize
for t_user_info in values[start:end]:
xtb.add_row(dict(t_user_info).values())
print(xtb)
elif len(user_info_list) == 4:
if user_info_list[0] != 'page' or user_info_list[-2] != 'pagesize':
user_expand.format_print(False,
"eg: display page 2 pagesize 3")
return
page_value = int(user_info_list[1]) - 1
try:
pagesize = int(user_info_list[-1])
except ValueError as e:
print("{} not a int number".format(user_info_list[-1]))
return False
start = page_value * pagesize
end = start + pagesize
for t_user_info in values[start:end]:
xtb.add_row(dict(t_user_info).values())
print(xtb)
else:
user_expand.format_print(False, "eg: display page 2")
def findUser(self):
# 有个bug 暂时没考虑模糊查询
username = self.user_info_string
tag = self.tag
if "=" in username:
username = username.split(" ")[0]
tag = True
user_data = student.select().where(student.username == username)
if len(user_data):
user_dict = dict(user_data[0].__data__.items())
xtb = PrettyTable()
xtb.field_names = FIELDS
xtb.add_row(user_dict.values())
# tag 非None才打印
if not tag:
print(xtb)
result = True
else:
user_expand.format_print(
False, "Username: {} not found.".format(username))
result = False
if tag:
return result
def deleteUser(self):
args = self.user_info_string
user_info_list = args.split(" ")
if args is '' or len(user_info_list) != 1:
logs.save_log("deleteUser failed, args length != 1", tag='error')
user_expand.format_print(False,
"deleteUser failed, args length != 1")
return
username = args
try:
tag = student.delete().where(
student.username == username).execute()
if tag:
user_expand.format_print(
True, "delete user {} secc.".format(username))
else:
user_expand.format_print(
False, "Username: {} not found.".format(username))
except Exception as e:
print(e)