def add_job(): sess = DbSession().get_session() user = get_user(sess, session["username"]) post = request.form quantity = extract_field(post, 'qty_adj', 1) for job_num in range(quantity): new_job = Job() new_job.OwnerId = user.Id if not new_job.populate(post): return jsonify({"status":0, "error":"Double check fields. Integers only please."}) if job_num > 0: new_job.Name += " %d" % (job_num+1) if job_num == 0: if user.GoldPool < (new_job.GoldCost*quantity): return jsonify({"status":0, "error":"Insufficient Gold. Gold remaining: %d. Needed: %d." % (user.GoldPool, new_job.GoldCost*quantity)}) if user.XpPool < (new_job.XpCost*quantity): return jsonify({"status":0, "error":"Insufficient Experience. Experience remaining: %d. Needed: %d." % (user.XpPool, new_job.XpCost*quantity)}) user.GoldPool -= new_job.GoldCost user.XpPool -= new_job.XpCost sess.add(new_job) sess.commit() session["next_tab"] = 2 return jsonify({"status":1})
def add_wright(): sess = DbSession().get_session() new_wright = Wright() #user = sess.query(User).filter(User.Username == session["username"]) user = get_user(sess, session["username"]) new_wright.OwnerId = user.Id new_wright.Name = request.form["wright_name"] sess.add(new_wright) sess.commit() session["next_tab"] = 2 return redirect(url_for("calc"))
def create_acct(): if request.method == "POST": username, pw1, pw2 = request.form["username"], request.form["password1"], request.form["password2"] if pw1 != pw2: return render_template("login.html", create_style="active", error="Passwords do not match. Please try again.") sess = DbSession().get_session() new_user = create_user(User, username, pw1) new_user.XpPool = 0 new_user.GoldPool = 0 sess.add(new_user) sess.commit() flash("Account %s created!" % (username)) return redirect(url_for("login")) else: return render_template("create_acct.html", create_style="active")