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