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")