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)