From d508c5e815677fef85571d4a8d3fe8fc0ec987b5 Mon Sep 17 00:00:00 2001
From: hjoaquim <h.joaquim@campus.fct.unl.pt>
Date: Wed, 14 Sep 2022 15:16:33 +0100
Subject: [PATCH] Added export to stocks/qa/beta (#2510)

* added export to beta

* added alpha and beta to filename and removed it from the actual file
---
 .../stocks/quantitative_analysis/beta_view.py        | 12 ++++++++++++
 .../stocks/quantitative_analysis/qa_controller.py    |  4 +++-
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/openbb_terminal/stocks/quantitative_analysis/beta_view.py b/openbb_terminal/stocks/quantitative_analysis/beta_view.py
index cf47d36a206..edd492a862b 100644
--- a/openbb_terminal/stocks/quantitative_analysis/beta_view.py
+++ b/openbb_terminal/stocks/quantitative_analysis/beta_view.py
@@ -2,12 +2,14 @@
 __docformat__ = "numpy"
 
 import logging
+import os
 
 import matplotlib.pyplot as plt
 import pandas as pd
 from openbb_terminal.decorators import log_start_end
 from openbb_terminal.rich_config import console
 from openbb_terminal.stocks.quantitative_analysis.beta_model import beta_model
+from openbb_terminal.helper_funcs import export_data
 
 logger = logging.getLogger(__name__)
 
@@ -18,6 +20,7 @@ def beta_view(
     ref_symbol: str,
     data: pd.DataFrame = None,
     ref_data: pd.DataFrame = None,
+    export: str = "",
 ) -> None:
     """Display the beta scatterplot + linear regression.
 
@@ -52,3 +55,12 @@ def beta_view(
     ax.text(0.9, 0.1, beta_text, horizontalalignment="right", transform=ax.transAxes)
     fig.show()
     console.print()
+
+    df = pd.DataFrame({"sr": sr, "rr": rr})
+
+    export_data(
+        export,
+        os.path.dirname(os.path.abspath(__file__)),
+        f"beta_alpha={alpha}_beta={beta}",
+        df,
+    )
diff --git a/openbb_terminal/stocks/quantitative_analysis/qa_controller.py b/openbb_terminal/stocks/quantitative_analysis/qa_controller.py
index 7480394795b..cd96d33b5dc 100644
--- a/openbb_terminal/stocks/quantitative_analysis/qa_controller.py
+++ b/openbb_terminal/stocks/quantitative_analysis/qa_controller.py
@@ -783,7 +783,9 @@ class QaController(StockBaseController):
             parser, other_args, EXPORT_ONLY_RAW_DATA_ALLOWED
         )
         if ns_parser:
-            beta_view(self.ticker, ns_parser.ref, data=self.stock)
+            beta_view(
+                self.ticker, ns_parser.ref, data=self.stock, export=ns_parser.export
+            )
 
     @log_start_end(log=logger)
     def call_var(self, other_args: List[str]):