def register():
if request.method == 'POST':
if "" in request.form.values():
return render_template("register.html")
if request.form['username'] in list(User.query.values(User.name)):
flash('Please enter a password.', 'error')
return render_template("register.html")
if request.form['email'] in list(User.query.values(User.email)):
flash('Please enter a valid email.', 'error')
return render_template("register.html")
if request.form['password'] != request.form['passwordconfirm']:
flash('Passwords do not match.', 'error')
return render_template("register.html")
# TODO: error for when they try to register when logged in already
u = User(
request.form['username'],
request.form['email'],
generate_password_hash(
request.form['password'].strip()))
db_session.add(u)
db_session.commit()
"""for currency in config.get_currencies():
addr = generate_deposit_address(currency)
a = Address(currency, addr, u.id)
db_session.add(a)
db_session.commit()
if not send_confirm_email(u.id):
flash('An error occured during registration. Please contact the administrator.', 'danger')
return home_page("ltc_btc")"""
flash('Successfully registered. Please check your email and confirm your account before logging in.', 'dismissable')
return home_page("ltc_btc")
if request.method == 'GET':
return render_template("register.html")
def register():
if request.method == 'POST':
if "" in request.form.values():
return render_template("register.html")
if request.form['username'] in list(User.query.values(User.name)):
flash('Please enter a password.', 'error')
return render_template("register.html")
if request.form['email'] in list(User.query.values(User.email)):
flash('Please enter a valid email.', 'error')
return render_template("register.html")
if request.form['password'] != request.form['passwordconfirm']:
flash('Passwords do not match.', 'error')
return render_template("register.html")
# TODO: error for when they try to register when logged in already
u = User(request.form['username'], request.form['email'],
generate_password_hash(request.form['password'].strip()))
db_session.add(u)
db_session.commit()
"""for currency in config.get_currencies():
addr = generate_deposit_address(currency)
a = Address(currency, addr, u.id)
db_session.add(a)
db_session.commit()
if not send_confirm_email(u.id):
flash('An error occured during registration. Please contact the administrator.', 'danger')
return home_page("ltc_btc")"""
flash(
'Successfully registered. Please check your email and confirm your account before logging in.',
'dismissable')
return home_page("ltc_btc")
if request.method == 'GET':
return render_template("register.html")
def send_confirm_email(uid):
user = User.query.filter(User.id==uid).first()
if user:
if not user.activated:
code = generate_password_hash(str(random.random()))
redis.hset("activation_keys", code, str(uid))
msg = Message('Activation Code', sender="*****@*****.**", recipients=[user.email])
msg.body = "Thank you for signing up at OpenExchange. Activate your account at http://localhost:5000/activate/{}".format(code)
mail.send(msg)
return True
return False
def cancelorder(old_order_id):
if not is_logged_in(session):
flash("Please log in to perform that action.", "error")
return home_page("ltc_btc")
uid = session['userid']
if old_order_id not in redis.smembers(str(uid) + "/orders"):
flash("Unable to cancel the specified order!", "error")
else:
orderid = generate_password_hash(str(random.random()))
redis.hmset(orderid, {"ordertype": 'cancel', "uid": uid, 'old_order_id': old_order_id})
redis.rpush("order_queue", orderid)
flash("Cancelled order!", "dismissable")
return home_page("ltc_btc")
def send_confirm_email(uid):
user = User.query.filter(User.id == uid).first()
if user:
if not user.activated:
code = generate_password_hash(str(random.random()))
redis.hset("activation_keys", code, str(uid))
msg = Message('Activation Code',
sender="*****@*****.**",
recipients=[user.email])
msg.body = "Thank you for signing up at OpenExchange. Activate your account at http://localhost:5000/activate/{}".format(
code)
mail.send(msg)
return True
return False
def addorder():
""" Checks balance and essential stuff, generates an order ID then adds order to a redis queue. """
instrument = request.form['currency_pair']
if not is_logged_in(session):
flash("Please log in to perform that action.", "error")
return home_page(instrument)
# They shouldn't be able to modify the trade pair, if it isnt valid either
# I messed up somewhere or they are trying to do something wrong
if not config.is_valid_instrument(instrument):
flash("Unknown Error, contact the administrator!", "error")
return home_page("ltc_btc")
base_currency = request.form['currency_pair'].split("_")[0]
quote_currency = request.form['currency_pair'].split("_")[1]
try:
rprice = Decimal(request.form['price'])
ramount = string_to_currency_unit(
request.form['amount'],
config.get_multiplier(base_currency))
print(ramount)
except Exception as e:
print(e)
flash("Please enter numerical values for price and amount!", "error")
return home_page(instrument)
if ramount < 1: # TODO: find a good amount for this
flash("Transaction amount too low!", "error")
return home_page(instrument)
if rprice <= 0:
flash("Price must be greater than 0!", "error")
return home_page(instrument)
getcontext().prec = 6
whole, dec = ExtendedContext.divmod(
rprice * ramount / config.get_multiplier(base_currency), Decimal(1))
total = int(
whole *
config.get_multiplier(base_currency) +
dec *
config.get_multiplier(base_currency))
print("total: " + str(total))
uid = session['userid']
orderid = generate_password_hash(str(random.random()))
instrument = request.form['currency_pair']
bidtable = instrument + "/bid"
asktable = instrument + "/ask"
if request.form['ordertype'] == 'buy':
currency = quote_currency
if check_balance(currency, session['userid']) < total:
flash("Balance too low to execute order!", "error")
return home_page(instrument)
else:
adjustbalance(currency, session['userid'], -1 * total)
elif request.form['ordertype'] == 'sell':
currency = base_currency
if check_balance(currency, uid) < ramount:
flash("Balance too low to execute order!", "error")
return home_page(instrument)
else:
adjustbalance(currency, uid, -1 * ramount)
else:
# invalid order type, they must have been messing around
flash("Unknown Error, contact the administrator!", "error")
return home_page(instrument)
redis.hmset(orderid,
{"ordertype": request.form['ordertype'],
"instrument": request.form['currency_pair'],
"amount": ramount,
"uid": uid,
"price": rprice})
redis.rpush("order_queue", orderid)
redis.sadd(str(uid) + "/orders", orderid)
flash("Order placed successfully!","dismissable")
return home_page(instrument)
