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 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 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 listUser(self):
'''
打印所有用户信息
:return:
'''
# 查询数据库全部信息
user_data = student.select()
# 将数据转化为列表
user_list = [
user_data[values].__data__.items()
for values in range(len(user_data))
]
xtb = PrettyTable()
xtb.field_names = FIELDS
# 设置字段左对齐
xtb.align = 'l'
for v in user_list:
# 强制转化为字典
xtb.add_row(dict(v).values())
print(xtb)
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