technical analysis: better error handling, fetchperiodinweeks=12

src/technical_analysis.py

@@ -16,7 +16,7 @@ class TechnicalAnalysis():
     def __init__(
             self,
             ticker:str,
-            fetchperiodinweeks:int=8,
+            fetchperiodinweeks:int=12,
             plot_ta:bool=True,
             debug=False):
         # input arguments
@@ -40,9 +40,11 @@ class TechnicalAnalysis():
         self.logger = logging.getLogger(__name__)
         logging.basicConfig(level=self.logger_level)  # filename='TechnicalAnalysis.log', 
 
+        # input arguments
         self.ticker = ticker
         self.fetchperiodinweeks = fetchperiodinweeks
         self.plot_ta = plot_ta
+        # done initializing
         self.logger.info(f'Initialized TechnicalAnalysis object for ticker: {ticker}')
 
     def run(
@@ -58,18 +60,19 @@ class TechnicalAnalysis():
         """
         self.df = self.fetch_data()
         # add the features based on technical analysis
-        self.df = self.tech_analysis()
-        # plot the results
-        if self.plot_ta:
-            os.makedirs('plots', exist_ok=True)
-            self.plot_stock_metrics(
-                self.df,
-                datasets={
-                        'Volume': ['Volume'],
-                        'Prices': ['Close', 'VWAP'],  # 'High','Low', 
-                        'Indices': ['RSI', 'StochOsc'],
-                        'Trend': ['MACD', 'MACDsig', 'MACDdif']}
-                        )
+        if self.df.shape[0] > 0:
+            self.df = self.tech_analysis()
+            # plot the results
+            if self.plot_ta:
+                os.makedirs('plots', exist_ok=True)
+                self.plot_stock_metrics(
+                    self.df,
+                    datasets={
+                            'Volume': ['Volume'],
+                            'Prices': ['Close', 'VWAP'],  # 'High','Low', 
+                            'Indices': ['RSI', 'StochOsc'],
+                            'Trend': ['MACD', 'MACDsig', 'MACDdif']}
+                            )
         return self.df
 
     def fetch_data(
@@ -112,8 +115,6 @@ class TechnicalAnalysis():
             df.columns = df.columns.droplevel('Ticker')
             self.logger.debug(df.head(10))
             self.logger.info(f'Fetched {df.shape[0]} rows for {self.ticker}')
-        else:
-            raise Exception(f'Failed to fetch data for {self.ticker}')
         
         return df