def company():
if "auth_field" in session:
ticker = request.args['ticker']
dbcursor.execute("SELECT * FROM company WHERE ticker = %s", (ticker, ))
dbres = dbcursor.fetchone()
value = value_tracker(ticker)
return render_template('company.info.html', dbres=dbres, value=value)
def deal():
if "auth_field" in session:
form = BidForm(user_id=session['_id'])
if form.validate_on_submit():
deal_id = request.args['deal_id']
acc_id = request.args['acc_id']
number = request.form['number']
symbol = request.form['symbol']
return redirect(
url_for('request_bid',
deal_id=deal_id,
acc_id=acc_id,
number=number,
symbol=symbol))
else:
if 'id' in request.args:
_id = request.args['id']
session['deal_id'] = _id
dbcursor.execute("SELECT * FROM deals WHERE _id = %s", (_id, ))
dbres = dbcursor.fetchone()
return render_template("deal.room.html",
dbres=dbres,
form=form,
acc_id=session["_id"])
else:
dbcursor.execute("SELECT * FROM deals WHERE _id = %s",
(session['deal_id'], ))
dbres = dbcursor.fetchone()
return render_template("deal.room.html",
dbres=dbres,
form=form,
acc_id=session["_id"])
def db_login(form_data):
dbcursor.execute("SELECT password FROM users WHERE email = %s OR uname = %s", (form_data["form_field"], form_data["form_field"]))
dbres = dbcursor.fetchone()
if dbres and bcrypt.check_password_hash(dbres[0], form_data["password"]):
return True
else:
return False
def validate_number(self, number):
dbcursor.execute("SELECT amount FROM stock_tracker WHERE symbol = %s AND user_id = %s", (self.symbol.data, self.user_id.data))
dbres = dbcursor.fetchone()
if dbres:
if dbres[0] <= number.data:
raise ValidationError("You don't own enough stocks for this symbol")
else:
raise ValidationError("You don't own enough stocks for this symbol")
def value_tracker(ticker_name):
now = datetime.datetime.now()
start = now.replace(hour=0, minute=0, second=0, microsecond=1)
sum = 0
num = 0
avg = None
data_tuple = (start, now, ticker_name)
dbcursor.execute("SELECT * FROM stock_transactions WHERE date >= %s AND date <= %s AND give = %s", data_tuple)
dbres = dbcursor.fetchall()
if dbres:
for res in dbres:
if res[8] == "DEAL":
if res[4] == "money":
num += res[3]
sum += num*res[5]
avg = sum/num
return avg
def login():
if "auth-field" not in session:
form = LoginForm()
if form.validate_on_submit():
if db_login(request.form):
session["auth_field"] = request.form["form_field"]
dbcursor.execute(
"SELECT _id FROM users WHERE email = %s OR uname = %s",
(session["auth_field"], session["auth_field"]))
dbres = dbcursor.fetchone()
session["_id"] = dbres[0]
return redirect(url_for('dashboard'))
else:
return render_template(
'login.html',
form=form,
auth_error="Incorrect username, email, or password")
else:
return render_template('login.html', form=form, auth_error="")
else:
return redirect(url_for('dashboard'))
def request_bid():
if "auth_field" in session:
deal_id = request.args['deal_id']
acc_id = request.args['acc_id']
bid_type = request.args['symbol']
bid_amount = request.args["number"]
dbcursor.execute("SELECT dealer_id FROM deals WHERE _id = %s",
(deal_id, ))
dbres = dbcursor.fetchone()
dbcursor.execute("SELECT email FROM users WHERE _id = %s",
(dbres[0], ))
dbres1 = dbcursor.fetchone()
if str(acc_id) == str(dbres[0]):
flash('You cannot bid on your own deal', 'danger')
return redirect(url_for('deal', id=deal_id))
data_tuple = (acc_id, deal_id, bid_type, bid_amount)
dbcursor.execute(
"INSERT INTO proposals (proposer_id, deal_id, proposal_type, bid_amount) VALUES (%s, %s, %s, %s)",
data_tuple)
db.commit()
flash('Bid requested successfully', 'success')
notify_tuple = (dbres1[0],
session['auth_field'] + ' made a bid for your deal',
session['auth_field'] +
' made a bid for your deal. Go check it out now!')
n = threading.Thread(target=notify.notification, args=notify_tuple)
n.start()
return redirect(url_for('deal', id=deal_id))
def my_account():
if "auth_field" in session:
dbcursor.execute(
"SELECT amount FROM stock_tracker WHERE user_id = %s AND symbol = %s",
(session['_id'], 'money'))
dbres = dbcursor.fetchone()
form = DealForm(user_id=session['_id'])
if form.validate_on_submit():
place_deal(request.form, session["_id"])
dbcursor.execute(
"UPDATE stock_tracker SET amount = amount - %s WHERE symbol = %s AND user_id = %s",
(request.form['number'], request.form['symbol'],
session['_id']))
db.commit()
flash('Deal placed successfully', 'success')
return render_template('my.account.html',
auth_name=session["auth_field"],
funds=dbres[0],
form=form)
else:
return render_template('my.account.html',
auth_name=session["auth_field"],
funds=dbres[0],
form=form)
def list_proposals():
if "auth_field" in session:
dbcursor.execute("SELECT _id FROM deals WHERE dealer_id = %s",
(session['_id'], ))
dbres = dbcursor.fetchall()
data_list = []
for _id in dbres:
dbcursor.execute("SELECT * FROM proposals WHERE deal_id = %s",
(_id[0], ))
dbres = dbcursor.fetchall()
for data in dbres:
dbcursor.execute("SELECT uname FROM users WHERE _id = %s",
(data[1], ))
dbres = dbcursor.fetchone()
data_list.append([data, dbres[0]])
return render_template('proposals.html', data_list=data_list)
def transactions():
if "auth_field" in session:
dbcursor.execute(
"SELECT * FROM stock_transactions WHERE taker_id = %s OR giver_id = %s",
(session['_id'], session['_id']))
dbres = dbcursor.fetchall()
data_list = []
if dbres:
for transaction in dbres:
dbcursor.execute("SELECT uname FROM users WHERE _id = %s",
(transaction[7], ))
dbres1 = dbcursor.fetchone()
if transaction[8] == "IPO":
dbcursor.execute("SELECT name FROM company WHERE _id = %s",
(transaction[6], ))
dbres2 = dbcursor.fetchone()
else:
dbcursor.execute("SELECT uname FROM users WHERE _id = %s",
(transaction[6], ))
dbres2 = dbcursor.fetchone()
data_list.append([transaction, dbres1[0], dbres2[0]])
return render_template('transactions.html', data_list=data_list)
else:
return render_template('transactions.html', data_list=data_list)
def proposal():
if "auth_field" in session:
_id = request.args['id']
dbcursor.execute("SELECT * FROM proposals WHERE _id = %s", (_id, ))
dbres = dbcursor.fetchone()
dbcursor.execute(
"SELECT name, symbol, symbol_amount, total_amount, deal_type FROM deals WHERE _id = %s",
(dbres[2], ))
dbres1 = dbcursor.fetchone()
dbcursor.execute("SELECT uname, _id FROM users WHERE _id = %s",
(dbres[1], ))
dbres2 = dbcursor.fetchone()
return render_template('proposal.html',
dbres=dbres,
dbres1=dbres1,
dbres2=dbres2,
proposal_id=_id)
def validate_uname(self, uname):
dbcursor.execute("SELECT uname FROM users WHERE uname = %s",
(uname.data, ))
dbres = dbcursor.fetchone()
if dbres and dbres[0] is not None:
raise ValidationError('This username is already taken')
def validate_email(self, email):
dbcursor.execute("SELECT email FROM users WHERE email = %s",
(email.data, ))
dbres = dbcursor.fetchone()
if dbres:
raise ValidationError("This email is already taken")
def validate_symbol(self, symbol):
dbcursor.execute("SELECT symbol FROM stock_tracker WHERE symbol = %s AND user_id = %s", (symbol.data, self.user_id.data))
dbres = dbcursor.fetchone()
if not dbres:
raise ValidationError("Symbol doesn't exists")
def feed_tracker_data(data, user_id):
data_tuple = (data["symbol"], data["amount"], user_id)
dbcursor.execute("INSERT INTO stock_tracker (symbol, amount, user_id) VALUES (%s, %s, %s)", data_tuple)
db.commit()
def db_signup(form_data):
hashed_pass = bcrypt.generate_password_hash(form_data["password"]).decode('utf-8')
data_tuple = (form_data["fname"], form_data["lname"], form_data["email"], form_data["uname"], hashed_pass)
dbcursor.execute("INSERT INTO users (fname, lname, email, uname, password) values (%s, %s, %s, %s, %s)", data_tuple)
db.commit()
return dbcursor.lastrowid
def tickers():
if "auth_field" in session:
dbcursor.execute("SELECT * FROM company")
dbres = dbcursor.fetchall()
return render_template('ticker.list.html', dbres=dbres)
def make_deal():
if "auth_field" in session:
deal_id = request.args['deal_id']
user_id = request.args['uid']
take = request.args['take']
take_amt = request.args['take_amt']
give = request.args['give']
give_amt = request.args['give_amt']
deal_type = request.args['type']
deal_date = datetime.datetime.today()
if deal_type == "IPO":
dbcursor.execute("SELECT dealer_id FROM deals WHERE _id = %s",
(deal_id, ))
dbres = dbcursor.fetchone()
giver_id = dbres[0]
data_tuple = (deal_date, give, give_amt, take, take_amt, giver_id,
user_id, deal_type)
dbcursor.execute(
"INSERT INTO stock_transactions (date, give, give_amount, take, take_amount, giver_id, taker_id, type) VALUES (%s, %s, %s, %s, %s, %s, %s, %s)",
data_tuple)
dbcursor.execute("DELETE FROM deals WHERE _id = %s", (deal_id, ))
dbcursor.execute(
"UPDATE stock_tracker SET amount = amount - %s WHERE user_id = %s AND symbol = %s",
(take_amt, user_id, take))
dbcursor.execute(
"SELECT symbol FROM stock_tracker WHERE user_id = %s AND symbol = %s",
(user_id, give))
dbres = dbcursor.fetchone()
if dbres:
dbcursor.execute(
"UPDATE stock_tracker SET amount = amount + %s WHERE user_id = %s AND symbol = %s",
(give_amt, user_id, give))
else:
dbcursor.execute(
"INSERT INTO stock_tracker (user_id, amount, symbol) VALUES (%s, %s, %s)",
(user_id, give_amt, give))
db.commit()
flash('Deal made successfully', 'success')
return redirect(url_for('market'))
else:
giver_id = session['_id']
data_tuple = (deal_date, give, give_amt, take, take_amt, giver_id,
user_id, deal_type)
dbcursor.execute(
"INSERT INTO stock_transactions (date, give, give_amount, take, take_amount, giver_id, taker_id, type) VALUES (%s, %s, %s, %s, %s, %s, %s, %s)",
data_tuple)
dbcursor.execute("DELETE FROM deals WHERE _id = %s", (deal_id, ))
dbcursor.execute(
"UPDATE stock_tracker SET amount = amount - %s WHERE user_id = %s AND symbol = %s",
(take_amt, user_id, take))
dbcursor.execute(
"SELECT symbol FROM stock_tracker WHERE user_id = %s AND symbol = %s",
(giver_id, take))
dbres = dbcursor.fetchone()
if dbres:
dbcursor.execute(
"UPDATE stock_tracker SET amount = amount + %s WHERE user_id = %s AND symbol = %s",
(take_amt, giver_id, take))
else:
dbcursor.execute(
"INSERT INTO stock_tracker (user_id, amount, symbol) VALUES (%s, %s, %s)",
(giver_id, take_amt, take))
dbcursor.execute(
"SELECT symbol FROM stock_tracker WHERE user_id = %s AND symbol = %s",
(user_id, give))
dbres = dbcursor.fetchone()
if dbres:
dbcursor.execute(
"UPDATE stock_tracker SET amount = amount + %s WHERE user_id = %s AND symbol = %s",
(give_amt, user_id, give))
else:
dbcursor.execute(
"INSERT INTO stock_tracker (user_id, amount, symbol) VALUES (%s, %s, %s)",
(user_id, give_amt, give))
db.commit()
flash('Deal made successfully', 'success')
dbcursor.execute("SELECT email FROM users WHERE _id = %s",
(user_id, ))
dbres1 = dbcursor.fetchone()
notify_tuple = (dbres1[0],
'Congratulations! your request was accepted',
session['auth_field'] +
' Approved your deal! You now own ' +
str(give_amt) + ' stocks of ' + str(give))
n = threading.Thread(target=notify.notification, args=notify_tuple)
n.start()
return redirect(url_for('list_proposals'))
def place_deal(form_data, dealer_id):
data_tuple = (form_data["name"], form_data["description"], form_data["symbol"], form_data["number"], form_data["amount"], dealer_id, "DEAL")
dbcursor.execute("INSERT INTO deals (name, description, symbol, symbol_amount, total_amount, dealer_id, deal_type) VALUES (%s, %s, %s, %s, %s, %s, %s)", data_tuple)
db.commit()
def market():
if "auth_field" in session:
dbcursor.execute("SELECT * FROM deals")
dbres = dbcursor.fetchall()
return render_template('marketplace.html', dbres=dbres)
