def init_users():
root = ElementTree.Element("USERS")
for k, info in users.items():
u = ElementTree.SubElement(root, "USER", attrib={"UNAME": k})
upass = ElementTree.SubElement(u, "UPASS")
ugid = ElementTree.SubElement(u, "GID")
urole = ElementTree.SubElement(u, "UROLE")
uname = ElementTree.SubElement(u, "NAME")
upass.text = common.encry_sha(info["UPASS"])
ugid.text = info["GID"]
urole.text = info["UROLE"]
uname.text = info["NAME"]
xmlfile = ElementTree.ElementTree(root)
xmlfile.write("users.xml", encoding="utf-8", xml_declaration=True)
def user_auth(self, password):
"""
登录验证
:param password: 用户输入密码
:return:
"""
try:
if not self.exists:
auth_status = False
else:
encry_passwd = common.encry_sha(password)
if self.passwd == encry_passwd:
auth_status = True
else:
auth_status = False
return auth_status
except Exception as e:
common.write_log(e, "error")
def user_auth(self, password):
"""
登录验证
:param password: 用户输入密码
:return:
"""
try:
if not self.exists:
auth_status = False
else:
encry_passwd = common.encry_sha(password)
if self.passwd == encry_passwd:
auth_status = True
else:
auth_status = False
return auth_status
except Exception as e:
common.write_log(e, "error")
if command == "2":
try:
tmpflag = False
while not tmpflag:
username = common.input_msg("输入用户名[q返回]: ")
if username == "q":
tmpflag = True
continue
# 创建用户对象
new_user = Users(username)
if not new_user.exists:
userpasswd = common.input_msg("设置初始密码[默认12345]: ", default="12345")
totalspace = common.input_msg("设置磁盘配额[默认500M]: ", default=str(settings.HOME_QUOTA))
print("\n 正在初始化用户，请稍等.........\n")
new_user.password = common.encry_sha(userpasswd)
new_user.islocked = 0
new_user.isdel = 0
new_user.totalspace = int(totalspace) * 1024 * 1024
new_user.usedspace = 0
new_user.create_user()
print("初始化成功!")
else:
print("\033[1;30m用户已经存在!\033[0m;\n")
continue
except Exception as e:
common.writelog("start - main - type2 - {0}".format(e), "error")
tmpflag = False
while not tmpflag:
username = common.input_msg("输入用户名[q返回]: ")
if username == "q":
tmpflag = True
continue
# 创建用户对象
new_user = Users(username)
if not new_user.exists:
userpasswd = common.input_msg("设置初始密码[默认12345]: ",
default="12345")
totalspace = common.input_msg("设置磁盘配额[默认500M]: ",
default=str(
settings.HOME_QUOTA))
print("\n 正在初始化用户，请稍等.........\n")
new_user.password = common.encry_sha(userpasswd)
new_user.islocked = 0
new_user.isdel = 0
new_user.totalspace = int(totalspace) * 1024 * 1024
new_user.usedspace = 0
new_user.create_user()
print("初始化成功!")
else:
print("\033[1;30m用户已经存在!\033[0m;\n")
continue
except Exception as e:
common.writelog("start - main - type2 - {0}".format(e),
"error")
