def get_friends(username):
with Database('master.db') as db:
rough = db.get_friends(username)
friends = []
for row in rough:
friends.append(row[0])
return friends
def buy_holdings(user_name,ticker_symbol,trade_volume,price):
with Database('master.db') as db:
try:
current = db.check_holdings(user_name,ticker_symbol)
new_current = float(trade_volume) + current
db.update_holding(user_name,ticker_symbol,new_current)
except:
db.new_holding(user_name,ticker_symbol,trade_volume,(price*float(trade_volume)))
def check_username(username):
with Database('master.db') as db:
usernames = db.usernames()
usernames1 = []
print(usernames1)
for row in usernames:
usernames1.append(row[0])
if username not in usernames1:
return True
else:
return False
def check_user(looker, lookie):
with Database('master.db') as db:
users = db.get_users(looker)
users1 = []
for row in users:
users1.append(row[0])
print(users1)
if lookie in users1:
return True
else:
return False
def sell(user_name,ticker_symbol,trade_volume):
price = quote(ticker_symbol)
balance = check_balance(user_name)
owned = float(check_holdings(user_name,ticker_symbol))
if owned >= float(trade_volume):
with Database('master.db') as db:
new_owned = owned - float(trade_volume)
db.update_holding(user_name,ticker_symbol,new_owned)
new_balance = balance + (float(trade_volume)*float(price)) - 6.25
db.update_cash(user_name,new_balance)
db.insert_4('orders','username','ticker_symbol','trade_volume','execution_price',user_name,ticker_symbol,trade_volume,price)
return "Congratulations toy just sold {} stocks of {}".format(trade_volume,ticker_symbol)
else:
return "You don't have that many to sell"
def dump():
with Database('master.db') as db:
return db.dump()
def tweet(username, tweet):
with Database('master.db') as db:
db.insert_3('tweets', 'username', 'tweet', 'date', username, tweet,
datetime.now())
def add_friend(username, friend):
with Database('master.db') as db:
return db.add_friend(username, friend)
def buy_execute(user_id,ticker_symbol,trade_volume,price):
with Database('master.db') as db:
db.insert_4('orders','username','ticker_symbol','trade_volume','execution_price',user_id,ticker_symbol,float(trade_volume),float(price))
def get_tweets(username):
with Database('master.db') as db:
return db.get_tweets(username)
def check_holdings(user_name,ticker_symbol):
with Database('master.db') as db:
try:
return db.check_holdings(user_name,ticker_symbol)
except:
return False
def update_cash(username,balance):
with Database('master.db') as db:
return db.update_cash(username,balance)
def check_balance(user_id):
with Database('master.db') as db:
return db.check_balance(user_id)
def user_check(password):
with Database('master.db') as db:
username = db.user_check(password)
return username
def newsfeed(username):
with Database('master.db') as db:
return db.newsfeed(username)
def delete_user(username):
with Database('master.db') as db:
return db.delete_user(username)
def get_users():
with Database('master.db') as db:
return db.get_users()
def get_holdings(user_name):
with Database('master.db') as db:
return db.get_holdings(user_name)
def user_check(username):
with Database('master.db') as db:
username = db.user_check(username)
return username
def sign_up(username, password):
with Database('master.db') as db:
db.sign_up(username, password)
def holdings_volumes(user_name):
with Database('master.db') as db:
return db.holdings_volumes(user_name)
