Exemple #1
0
def withdraw(currency):
    if not is_logged_in(session):
        flash("Please log in to perform that action.", "error")
        return home_page("ltc_btc")
    if not config.is_valid_currency(currency):
        flash("Invalid Currency!", "error")
        return account_page()
    if request.method == 'GET':
        return account_page(withdraw=currency)
    elif request.method == 'POST':
        if 'amount' not in request.form or 'address' not in request.form:
            flash("Please enter an address and an amount!", "error")
            return account_page()
        try:
            total = string_to_currency_unit(
                request.form['amount'],
                config.get_multiplier(currency))
        except:
            flash("Invalid amount!", "error")
            return account_page()
        if check_balance(currency, session['userid']) < total or total < 0:
            flash("Balance too low to execute withdrawal!", "error")
            return account_page()
        # TODO: add valid address checking
        adjustbalance(currency, session['userid'], -1 * total)
        co = CompletedOrder(
            currency +
            "_" +
            currency,
            "WITHDRAWAL",
            total,
            0,
            session['userid'],
            is_withdrawal=True,
            withdrawal_address=request.form['address'])
        db_session.add(co)
        db_session.commit()
        flash("Deposit to " + request.form['address'] + " completed!", "success")
        return account_page()
Exemple #2
0
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)