def get_dftable(table_name):
#gets the table from the db
from fup.utils.dbwrap import connection
import pandas as pd
conn = connection()
query = "SELECT * FROM {}".format(table_name)
df = pd.read_sql_query(query, conn)
return df
def df2sql(df, dfname):
import pandas as pd
from fup.utils.dbwrap import stringifyDF, connection
connection = connection()
if connection == False:
print("Connection to db failed")
df = stringifyDF(df)
df.to_sql(dfname, connection, if_exists="replace", index=False)
connection.commit()
def sql2df(table_name):
import pandas as pd
from fup.utils.dbwrap import connection
connection = connection()
if connection == False:
print("Connection to db failed")
query = "SELECT * FROM {}".format(table_name)
df = pd.read_sql_query(query, connection)
return df
def execute_query(query, keepConn=False):
#Execute query, commit and close query
from fup.utils.dbwrap import connection
try: #execute query in database
conn = connection()
if conn == False:
print("Connection to db failed")
return False #if conn fails
else:
with conn:
conn.execute(query)
if keepConn:
return True
else:
conn.close()
return True
except Exception as e: #query was not executed, Try again
print("Got error: ", e)
return False