def register():
if request.method == "POST":
try:
account = request.form["account"]
username = request.form["username"]
password = request.form["password"]
user = User()
user.createUser(
account=account,
password=generate_password_hash(password),
role="USER_LOW",
username=username,
)
db.session.add(user)
db.session.commit()
baseLink = BaseUser()
baseLink.createInfo(level=1, userAc=account)
db.session.add(baseLink)
db.session.commit()
except Exception as e:
print(e)
flash("注册失败")
return redirect(url_for("auth.register"))
return redirect(url_for("auth.login"))
else:
return render_template("auth/register.html")
def addUser():
if request.method == "POST":
try:
account = request.form["account"]
username = request.form["username"]
password = request.form["password"]
level = request.form["level"]
if level == "MANAGE_LOW":
user = User()
user.createUser(
account=account,
password=generate_password_hash(password),
role=level,
username=username,
)
db.session.add(user)
db.session.commit()
sha = hashlib.md5()
sha.update(account.encode("utf-8"))
loveworker = LoveWorker()
loveworker.id = sha.hexdigest()
loveworker.level = 8
loveworker.manId = g.user.manLink.first().id
loveworker.userAc = account
db.session.add(loveworker)
db.session.commit()
except Exception as e:
flash("用户已存在")
return redirect(url_for("manage.index"))
else:
user = g.user
return render_template("manage/user/add.html", user=user)
def addUser():
if request.method == "POST":
account = request.form["account"]
username = request.form["username"]
password = request.form["password"]
level = request.form["level"]
print(level)
if account and username and password and level:
try:
user = User()
user.createUser(
account=account,
password=generate_password_hash(password),
role=level,
username=username,
)
db.session.add(user)
# db.session.commit()
levelNum = User.LEVEL.get(level)
print(levelNum, type(levelNum))
if levelNum > 8:
loveman = LoveManage()
loveman.userAc = account
loveman.level = User.LEVEL.get(level)
md = md5()
md.update(f"{account}".encode("utf-8"))
loveman.id = md.hexdigest()
db.session.add(loveman)
db.session.commit()
elif levelNum == 8:
loveworker = LoveWorker()
loveworker.userAc = account
loveworker.level = User.LEVEL.get(level)
md = md5()
md.update(f"{account}".encode("utf-8"))
loveworker.id = md.hexdigest()
db.session.add(loveworker)
db.session.commit()
elif levelNum == 4:
lineWorker = LoveLineWorker()
md = md5()
md.update(f"{account}".encode("utf-8"))
lineWorker.id = md.hexdigest()
lineWorker.level = User.LEVEL.get(level)
lineWorker.userAc = account
db.session.add(lineWorker)
db.session.commit()
else:
baseuser = BaseUser()
baseuser.userAc = account
baseuser.level = User.LEVEL.get(level)
md = md5()
md.update(f"{account}".encode("utf-8"))
baseuser.id = md.hexdigest()
db.session.add(baseuser)
db.session.commit()
except sqlalchemy.exc.IntegrityError:
flash("添加失败")
return redirect(url_for("admin.index"))
return redirect(url_for("admin.index"))
else:
return render_template("admin/user/add.html")
