Add ability to plot saved queries using plotly, need to check performance in production, also need to improve generate_plot function
This commit is contained in:
9
db.py
9
db.py
@@ -6,6 +6,7 @@ from datetime import datetime
|
||||
from dateutil.relativedelta import relativedelta
|
||||
from urllib.parse import urlparse
|
||||
from flask import g
|
||||
import pandas as pd
|
||||
from features.calendar import Calendar
|
||||
from features.exercises import Exercises
|
||||
from features.stats import Stats
|
||||
@@ -60,6 +61,14 @@ class DataBase():
|
||||
cur.close()
|
||||
|
||||
return (rv[0] if rv else None) if one else rv
|
||||
|
||||
def read_sql_as_df(self, query, params=None):
|
||||
conn = self.getDB()
|
||||
try:
|
||||
df = pd.read_sql(query, conn, params=params)
|
||||
return df
|
||||
except Exception as e:
|
||||
raise e
|
||||
|
||||
def get_exercise(self, exercise_id):
|
||||
exercise = self.execute(
|
||||
|
||||
Reference in New Issue
Block a user