app.py

# Import flask
from flask import Flask
from flask import render_template, request, make_response, redirect
from flask import current_app
from flask.ext.sqlalchemy import SQLAlchemy
import hashlib
# Library for bitmarket
from swaper import Yolo
# Make heroku happy and app secure
import os
import sys
import logging
app = Flask(__name__)

app.config['SQLALCHEMY_DATABASE_URI'] = os.environ['DATABASE_URL']
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)

app.logger.addHandler(logging.StreamHandler(sys.stdout))
app.logger.setLevel(logging.ERROR)


class Keys(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    public_key = db.Column(db.String(256), unique=False)
    private_key = db.Column(db.String(256), unique=True)
    url = db.Column(db.String(256), unique=True)

    def __init__(self, public_key, private_key):
        self.public_key = public_key.encode('utf8')
        self.private_key = private_key.encode('utf8')
        self.url = hashlib.sha256(
            "{0.public_key}{0.private_key}".format(self)
        ).hexdigest()

    def __repr__(self):
        return "<Public:  {public} Prv: {private}, {url}>".format(
            public=self.public_key, private=self.private_key, url=self.url)


class Rates(db.Model):
    id = db.Column(db.Integer, primary_key=True)

# Basic route for index


@app.route('/')
def index():
    return render_template("index.html")


@app.route('/login')
def login():
    return render_template("login.html")


@app.route('/handle', methods=['GET', 'POST'])
def handle():
    if request.method == 'POST':
        if request.form['pubkey'] and request.form['privkey']:
            keys = Keys(request.form['pubkey'], request.form['privkey'])
            db.session.rollback()
            try:
                db.session.add(keys)
                db.session.commit()
            except IOError:
                db.session.rollback()
            hashed_key = hashlib.sha256(
                "{public_key}{private_key}".format(
                    public_key=request.form['pubkey'],
                    private_key=request.form['privkey'])).hexdigest()
            return render_template("generated.html", hashed_key=hashed_key)
        else:
            return "Cannot add Key (duplicate or bad format)"
    else:
        return "Try moving to homepage and adding new key"


@app.route('/display/<special>')
def display(special):
    # Query database for keys
    record = Keys.query.filter(Keys.url == special).first()
    # Class initialisation for bitmarket API
    yolo = Yolo(record.public_key, record.private_key)
    balance = str(yolo.get_balance())
    positions = yolo.swap_list()
    # Setting cookie
    # cutoff level
    cut = yolo.get_cutoff()
    rsp = make_response(render_template("display.html",
                        balance=balance,
                        positions=positions,
                        cut=cut,
                        special=special))
    rsp.set_cookie('key', 'value')
    return rsp


@app.route('/make_best/<special>')
def create(special):

    record = Keys.query.filter(Keys.url == special).first()
    # Class initialisation for bitmarket API
    yolo = Yolo(record.public_key, record.private_key)
    yolo.cancel_all()
    yolo.make_best()
    return "Made best offer!"


@app.route('/set_cookie')
def cookie_insertion():
    redirect_to_index = redirect('/')
    response = current_app.make_response(redirect_to_index)
    response.set_cookie('cookie_name', value='values')
    return response


@app.route("/test")
def test():
    return render_template("test.html")

@app.route("/chart")
def chart():
    return render_template("chart.html")

@app.route("/data.csv")
def data_feed():
    return render_template("data.csv")