Pre code clean-up

src/submissions/frontend/visualizations/__init__.py

@@ -10,9 +10,12 @@ from frontend.widgets.functions import select_save_file
 
 class CustomFigure(Figure):
 
+    df = None
+
     def __init__(self, df: pd.DataFrame, modes: list, ytitle: str | None = None, parent: QWidget | None = None,
                  months: int = 6):
         super().__init__()
+        self.df = df
 
     def save_figure(self, group_name: str = "plotly_output", parent: QWidget | None = None):
         """
@@ -28,6 +31,11 @@ class CustomFigure(Figure):
         output = select_save_file(obj=parent, default_name=group_name, extension="png")
         self.write_image(output.absolute().__str__(), engine="kaleido")
 
+    def save_data(self, group_name: str = "plotly_export", parent:QWidget|None=None):
+        output = select_save_file(obj=parent, default_name=group_name, extension="xlsx")
+        self.df.to_excel(output.absolute().__str__(), engine="openpyxl", index=False)
+
+
     def to_html(self) -> str:
         """
         Creates final html code from plotly

src/submissions/frontend/visualizations/pcr_charts.py

@@ -21,6 +21,7 @@ class PCRFigure(CustomFigure):
     def __init__(self, df: pd.DataFrame, modes: list, ytitle: str | None = None, parent: QWidget | None = None,
                  months: int = 6):
         super().__init__(df=df, modes=modes)
+        logger.debug(f"DF: {self.df}")
         self.construct_chart(df=df)
         # self.generic_figure_markers(modes=modes, ytitle=ytitle, months=months)