36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
import psycopg2
|
|
|
|
from common import *
|
|
import psycopg2.extras
|
|
|
|
def get_db_connection():
|
|
return psycopg2.connect(cursor_factory=psycopg2.extras.RealDictCursor,
|
|
database="nasfaq", user = "steaky", host = "127.0.0.1", port = "5432")
|
|
|
|
def fetch_history(cursor, ts_low, ts_high, coins = COINS, userids = None):
|
|
if coins == None: return {}
|
|
|
|
query = "SELECT * FROM HISTORY WHERE (TIMESTAMP > {} AND TIMESTAMP <= {}) AND (".format(ts_low, ts_high)
|
|
# Filter by coins
|
|
for i in range(len(coins)):
|
|
if i == len(coins) - 1:
|
|
query += "COIN = '{}' )".format(coins[i])
|
|
else:
|
|
query += "COIN = '{}' OR ".format(coins[i])
|
|
|
|
# Filter by userids
|
|
if userids: query += "AND ("
|
|
for i in range(len(userids)):
|
|
if i == len(userids) - 1:
|
|
query += "userid = '{}' )".format(userids[i])
|
|
else:
|
|
query += "userid = '{}' OR ".format(userids[i])
|
|
|
|
query += ";"
|
|
print(query)
|
|
|
|
cursor.execute(query)
|
|
res = cursor.fetchall()
|
|
|
|
return res
|