WebTrader/webcontroller.py at master · MikeBloom914/WebTrader · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
#!/usr/bin/env python3

import time

from flask import Flask, redirect, render_template, request, session, url_for

import model


app = Flask(__name__)
app.config['SECRET_KEY'] = 'secret'


@app.route('/', methods=['GET', 'POST'])
def login():
    if request.method == 'GET':
        return render_template('login.html')
    else:
        submitted_email = request.form['email']
        submitted_password = request.form['password']
        if submitted_email and submitted_password:
            import sqlite3
            connection = sqlite3.connect('master.db', check_same_thread=False)
            cursor = connection.cursor()
            cursor.execute('SELECT email FROM users WHERE email=?;', (submitted_email,))
            objectified_email = cursor.fetchall()
            if len(objectified_email) == 0:
                return render_template('login.html', message='BAD CREDENTIALS...Please try again...If you are a new user please register')

            else:
                objectified_email = objectified_email[0][0]
                cursor.execute('SELECT password FROM users WHERE email=?;', (submitted_email,))
                objectified_password = cursor.fetchall()[0][0]
                if submitted_password == objectified_password:
                    import sqlite3
                    connection = sqlite3.connect('master.db', check_same_thread=False)
                    cursor = connection.cursor()
                    cursor.execute('SELECT pk FROM users WHERE email=?;', (submitted_email,))
                    session['guid'] = cursor.fetchall()[0][0]
                    return redirect(url_for('homepage'))
                else:
                    return render_template('login.html', message='BAD CREDENTIALS...Please try again')


@app.route('/registration', methods=['GET', 'POST'])
def registration():
    if request.method == 'GET':
        return render_template('registration.html')
    else:
        firstname = request.form['firstname']
        lastname = request.form['lastname']
        email = request.form['email']
        password = request.form['password']
        balance = float(100000)
        x = model.registration(firstname, lastname, email, password, balance)
        return render_template('login.html')


@app.route('/homepage', methods=['GET', 'POST'])
def homepage():
    if request.method == 'GET':
        return render_template('homepage.html')
    else:
        return render_template('homepage.html')


@app.route('/buy', methods=['GET', 'POST'])
def buy():
    if request.method == 'GET':
        return render_template('buy.html')
    else:
        ticker_symbol = request.form['thingone']
        trade_volume = request.form['thingtwo']
        x = model.buy(ticker_symbol, trade_volume, session['guid'])
        return render_template('buy.html', message=x)


@app.route('/sell', methods=['GET', 'POST'])
def sell():
    if request.method == 'GET':
        return render_template('sell.html')
    else:
        ticker_symbol = request.form['sellone']
        trade_volume = request.form['selltwo']
        x = model.sell(ticker_symbol, trade_volume, session['guid'])
        return render_template('sell.html', message=x)


@app.route('/lookup', methods=['GET', 'POST'])
def lookup():
    if request.method == 'GET':
        return render_template('lookup.html')
    else:
        company_name = request.form['coname']
        x = model.lookup(company_name)
        return render_template('lookup.html', message=x)


@app.route('/quote', methods=['GET', 'POST'])
def quote():
    if request.method == 'GET':
        return render_template('quote.html')
    else:
        ticker_symbol = request.form['copsymb']
        x = model.quote(ticker_symbol)
        return render_template('quote.html', message=x)


@app.route('/portfolio', methods=['GET'])
def portfolio():
    x = model.portfolio(session['guid'])
    return render_template('portfolio.html', message=x)


@app.route('/pl', methods=['GET'])
def pl():
    x = model.pl(session['guid'])
    return render_template('pl.html', message=x)


if __name__ == '__main__':
    app.run(host='127.0.0.1', port=5000, debug=True)