def get_atomic_kpi_static_data(rds_conn):
"""
This function extracts the static KPI data and saves it into one global data frame.
The data is taken from static.kpi / static.atomic_kpi / static.kpi_set.
"""
query = MarsUkQueries.get_atomic_kpi_data()
kpi_static_data = pd.read_sql_query(query, rds_conn)
return kpi_static_data
def commit_results_data(self):
"""
This function writes all KPI results to the DB, and commits the changes.
"""
cur = self._rds_conn.db.cursor()
delete_queries = MarsUkQueries.get_delete_session_results_query(
self._session_uid)
for query in delete_queries:
cur.execute(query)
for query in self._kpi_results_queries:
cur.execute(query)
self._rds_conn.db.commit()
def get_store_att10(self, store_fk):
query = MarsUkQueries.get_store_attribute_10(store_fk)
att10 = pd.read_sql_query(query, self.rds_conn.db)
return att10.values[0][0]