def insert_refresh_token(username: str, refresh_token: str, expiration_time):
db, db_cursor = get_cursor()
query = f"""
INSERT INTO access_tokens (username, refresh_token, expiry, expired)
VALUES ('{username}', '{refresh_token}', '{expiration_time}', false )
"""
db_cursor.execute(query)
commit_to_database(db, db_cursor)
def insert_security(username: str, security_symbol: str, exchange_name: str):
db, db_cursor = get_cursor()
query = f"""
INSERT INTO watchlist (username, security_symbol, exchange_name)
VALUES ('{username}', '{security_symbol}', '{exchange_name}')
"""
db_cursor.execute(query)
commit_to_database(db, db_cursor)
return
def revoke_refresh_tokens(username: str):
db, db_cursor = get_cursor()
query = f"""
UPDATE access_tokens
SET expired = 'true'
WHERE
access_tokens.username LIKE '{username}'
"""
db_cursor.execute(query)
commit_to_database(db, db_cursor)
def insert_transaction(params):
db, db_cursor = get_cursor()
timestamp = datetime.utcnow()
query = f"""
INSERT INTO transactions (username, security_symbol, exchange_name, price, quantity, transaction_time, transaction_type)
VALUES ('{params['username']}', '{params['symbol']}', '{params['exchange']}', {params['price']}, {params['quantity']}, '{timestamp}', '{params['transaction_type']}' )
"""
db_cursor.execute(query)
commit_to_database(db, db_cursor)
return
def delete_security(table_name: str, username: str, symbol: str,
exchange: str):
db, db_cursor = get_cursor()
query = f"""
DELETE FROM {table_name}
WHERE
{table_name}.username = '******' AND
{table_name}.security_symbol = '{symbol}' AND
{table_name}.exchange_name = '{exchange}'
"""
db_cursor.execute(query)
commit_to_database(db, db_cursor)
return
def update_quantity(table_name: str, transaction_id, username: str, symbol: str, exchange: str, new_quantity: int):
db, db_cursor = get_cursor()
query = f"""
UPDATE {table_name}
SET quantity = {new_quantity}
WHERE
{table_name}.username = '******' AND
{table_name}.security_symbol = '{symbol}' AND
{table_name}.exchange_name = '{exchange}' AND
{table_name}.transaction_id = {transaction_id}
"""
db_cursor.execute(query)
commit_to_database(db, db_cursor)
return