technical_analysis: minor bug fixes, return df

src/technical_analysis.py

@@ -32,10 +32,6 @@ class TechnicalAnalysis():
         debug : bool, optional, default: False
             whether run in debug mode, so that logging should be produced at debug level
         """
-        self.ticker = ticker
-        self.fetchperiodinweeks = fetchperiodinweeks
-        self.plot_ta = plot_ta
-        
         # set up logging
         if debug:
             self.logger_level = logging.DEBUG
@@ -44,6 +40,11 @@ class TechnicalAnalysis():
         self.logger = logging.getLogger(__name__)
         logging.basicConfig(level=self.logger_level)  # filename='TechnicalAnalysis.log', 
 
+        self.ticker = ticker
+        self.fetchperiodinweeks = fetchperiodinweeks
+        self.plot_ta = plot_ta
+        self.logger.info(f'Initialized TechnicalAnalysis object for ticker: {ticker}')
+
     def run(
             self
             ) -> None:
@@ -69,7 +70,7 @@ class TechnicalAnalysis():
                         'Indices': ['RSI', 'StochOsc'],
                         'Trend': ['MACD', 'MACDsig', 'MACDdif']}
                         )
-        
+        return self.df
 
     def fetch_data(
             self
@@ -93,7 +94,7 @@ class TechnicalAnalysis():
         """
 
         period_start = dt.datetime.now() - dt.timedelta(weeks=self.fetchperiodinweeks)
-        self.logger.info(f'Fetching price data for {ticker}')
+        self.logger.info(f'Fetching price data for {self.ticker}')
         try:
             df = yf.download(
                 self.ticker,
@@ -110,9 +111,9 @@ class TechnicalAnalysis():
             # get rid of the redundant ticker column
             df.columns = df.columns.droplevel('Ticker')
             self.logger.debug(df.head(10))
-            self.logger.info(f'Fetched {df.shape[0]} rows for {ticker}')
+            self.logger.info(f'Fetched {df.shape[0]} rows for {self.ticker}')
         else:
-            raise Exception(f'Failed to fetch data for {ticker}')
+            raise Exception(f'Failed to fetch data for {self.ticker}')
         
         return df
     
@@ -213,6 +214,7 @@ class TechnicalAnalysis():
                 colstoplot)
         plt.tight_layout()
         plt.savefig(os.path.join('plots',  f'{self.ticker}.png'))
+        plt.close()
 
     def plot_stock_metrics_ax(
             self,
@@ -272,6 +274,7 @@ class TechnicalAnalysis():
 
 if __name__ == '__main__':
     ticker = 'GOOG'
-    ta = TechnicalAnalysis(ticker, debug=False).run()
+    df = TechnicalAnalysis(ticker, debug=False).run()
+    print(f'columns: {df.columns}')