goldhand 15.5__tar.gz → 15.7__tar.gz

{goldhand-15.5 → goldhand-15.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: goldhand
-Version: 15.5
+Version: 15.7
 Summary: A package working with financial data
 Home-page: https://github.com/misrori/goldhand
 Author: Mihaly

{goldhand-15.5 → goldhand-15.7}/goldhand/backtest.py

@@ -6,6 +6,15 @@ import plotly.express as px
 
 class Backtest:
     def __init__(self, data, strategy_function, plot_title='', **kwargs):
+        """
+        Backtest class to test strategies on historical data to see how they would have performed.
+        
+        Parameters:
+        - data: pandas DataFrame with historical data
+        - strategy_function: function that takes in the data and returns a DataFrame of trades
+        - plot_title: title for the plot
+        - kwargs: additional parameters to be passed to the strategy function
+        """
         self.data = data
         self.plot_title = plot_title
         self.strategy_function = strategy_function
@@ -15,6 +24,9 @@ class Backtest:
 
 
     def add_trades(self):
+        """
+        Calculate the trades using the strategy function and the data provided
+        """
         self.trades = self.strategy_function(self.data, **self.additional_params)
         self.trades['ticker'] = self.data['ticker'].iloc[0]
         
@@ -26,6 +38,9 @@ class Backtest:
 
 
     def summary_of_trades(self):
+        """
+        Calculate the summary of the trades
+        """
         self.trades_summary = {
           'ticker' : self.data['ticker'].iloc[0],
           'number_of_trades' : self.trades.shape[0],
@@ -68,6 +83,9 @@ class Backtest:
         self.trades_summary.update(self.additional_params)
 
     def show_trades(self):
+        """
+        Plot the trades of the strategy on the data provided
+        """
         tdf = self.data
         fig = go.Figure(data=go.Ohlc(x=tdf['date'], open=tdf['open'], high=tdf['high'], low=tdf['low'],close=tdf['close']))
         fig.add_trace( go.Scatter(x=tdf['date'], y=tdf['sma_50'], opacity =0.5, line=dict(color='lightblue', width = 2) , name = 'SMA 50') )
@@ -121,6 +139,12 @@ class Backtest:
 
 
     def summarize_strategy(self):
+        """
+        Display the summary of the strategy:
+        - Summary of trades
+        - Trades in interactive plot
+        - Trades in DataFrame
+        """
         display(pd.DataFrame(self.trades_summary, index=['Strategy summary']).T )
         self.show_trades().show()
         display(self.trades)

{goldhand-15.5 → goldhand-15.7}/goldhand/stocks.py

@@ -13,12 +13,12 @@ class GoldHand:
     def __init__(self, ticker, ad_ticker=True, range='18y', interval='1d'):
         """
         GoldHand class to download and analyze stock data
-        params:
-        ticker: str, ticker symbol
-        ad_ticker: bool, add ticker column to the dataframe
-        range: str, time range to download data
-        interval: str, interval to download data
-        
+
+        Paramseters:
+        - ticker: str, ticker symbol of the stocks or crypto or ETF
+        - ad_ticker: bool, add ticker column to the DataFrame
+        - range: str, time range to download data for example 5y,1y, 1mo, 1d, 1h
+        - interval: str, interval to download data for example 1d, 1h, 5m
         """
         self.scraper = cloudscraper.create_scraper()
         self.ad_ticker = ad_ticker
@@ -31,8 +31,8 @@ class GoldHand:
 
     def get_olhc(self):
         """
-        Download historical stock data for the last year
-        # https://cryptocointracker.com/yahoo-finance/yahoo-finance-api
+        Download historical stock, crypto or ETF data from yahoo finance
+        API documentation location: https://cryptocointracker.com/yahoo-finance/yahoo-finance-api
         """
         #scraper = cloudscraper.create_scraper()
         response = self.scraper.get(f"https://query1.finance.yahoo.com/v8/finance/chart/{self.ticker}?interval={self.interval}&range={self.range}")
@@ -46,13 +46,13 @@ class GoldHand:
     
     def smma(self, data, window, colname):
         """
-        Calculate Smoothed Moving Average (SMMA)
-        params:
-        data: dataframe
-        window: int, window size
-        colname: str, name of the column to add to the dataframe
-        return:
-        data: dataframe with added column
+        Calculate Smoothed Simple Moving Average (SMMA)
+        Parameters:
+        - data: Pandas DataFrame
+        - window: int, window size
+        - colname: str, name of the column to add to the DataFrame
+        
+        Return: DataFrame with added column
         """
         hl2 = data['hl2'].values
         smma_values = [hl2[0]]
@@ -68,7 +68,7 @@ class GoldHand:
 
     def download_historical_data(self):
         """
-        Download historical stock data of 18 years
+        Download historical stock, crypto or ETF data 
         """
         # Download historical stock data for the last year
         self.df = self.get_olhc()
@@ -173,13 +173,12 @@ class GoldHand:
     def plotly_last_year(self, plot_title, plot_height=900, ndays=500, ad_local_min_max=True):
         """
         Plot last year interactive plot of a stock analyzing the local minimums and maximums
-        params:
-        plot_title: str, title of the plot
-        plot_height: int, height of the plot
-        ndays: int, number of days to plot
-        ad_local_min_max: bool, add local min max to the plot
-        return:
-        fig: plotly figure
+        Parameters:
+        - plot_title: str, title of the plot
+        - plot_height: int, height of the plot
+        - ndays: int, number of days to plot
+        - ad_local_min_max: bool, add local min max to the plot
+        Return: plotly figure
         """
         tdf = self.df.tail(ndays)
 
@@ -208,14 +207,13 @@ class GoldHand:
 
     def plot_goldhand_line(self, plot_title, plot_height=900, ndays=800,  ad_local_min_max=True):
         """
-        Plot last year interactive plot of a stock analyzing the local minimums and maximums using the goldhand line indicator
-        params:
-        plot_title: str, title of the plot
-        plot_height: int, height of the plot
-        ndays: int, number of days to plot
-        ad_local_min_max: bool, add local min max to the plot
-        return:
-        fig: plotly figure
+        Plot last year interactive plot of a stock analyzing the local minimums and maximums using the GoldHandLine indicator
+        Parameters:
+        - plot_title: str, title of the plot
+        - plot_height: int, height of the plot
+        - ndays: int, number of days to plot
+        - ad_local_min_max: bool, add local min max to the plot
+        Return: plotly figure
         """
         
         data = self.df.copy()

{goldhand-15.5 → goldhand-15.7}/goldhand/strategy_goldhand_line.py

@@ -10,13 +10,14 @@ from goldhand import *
 
 def goldhand_line_strategy(data, buy_at='gold', sell_at='grey'):
     """
-    This function implements the goldhand line strategy.
+    This function implements the GoldHandLine strategy.
+    
     Parameters:
-    data (pandas.DataFrame): The dataframe containing the data.
-    buy_at (str): The color of the line to buy at. Default is 'gold'.
-    sell_at (str): The color of the line to sell at. Default is 'grey'.
-    Returns:
-    res_df (pandas.DataFrame): The dataframe containing the results.
+    - data (pandas DataFrame) : The DataFrame containing the data.
+    - buy_at (str): The color of the line to buy at. Default is 'gold'.
+    - sell_at (str): The color of the line to sell at. Default is 'grey'.
+    
+    Returns: The trades of the GoldHandLine strategy. 
     """
 
     data['hl2'] = (data['high'] + data['low'])/2
@@ -112,15 +113,18 @@ def goldhand_line_strategy(data, buy_at='gold', sell_at='grey'):
 
 def show_indicator_goldhand_line_strategy(ticker, plot_title = '', buy_at='gold', sell_at='grey', ndays=0, plot_height=1000, add_strategy_summary = True):
     """
-    This function shows the goldhand line strategy on a plotly candlestick chart.   
+    This function shows the GoldHandLine strategy on a plotly chart including the price,  trades, strategy summary and GoldHandLine indicator.
+       
     Parameters:
-    ticker (str): The ticker of the stock or ETF.
-    plot_title (str): The title of the plot.
-    ndays (int): The number of days to show. If 0, all data will be shown.
-    plot_height (int): The height of the plot.
-    add_strategy_summary (bool): If True, the strategy summary will be added to the plot.
-    Returns:
-    fig (plotly.graph_objects.Figure): The plotly figure.
+    - ticker (str): The ticker of the stock or crypto or ETF.
+    - plot_title (str): The title of the plot.
+    - buy_at (str): The color of the line to buy at. Default is 'gold'.
+    - sell_at (str): The color of the line to sell at. Default is 'grey'.
+    - ndays (int): The number of days to show. If 0, all data will be shown.
+    - plot_height (int): The height of the plot.
+    - add_strategy_summary (bool): If True, the strategy summary will be added to the plot.
+    
+    Returns: The plot including the price,  trades, strategy summary and GoldHandLine indicator.
     """
 
     data = GoldHand(ticker).df
@@ -159,10 +163,10 @@ def show_indicator_goldhand_line_strategy(ticker, plot_title = '', buy_at='gold'
     # Create a 'group' column and increase the value only when there's a color change
     data['group'] = (data['color_change']).cumsum()
 
-    ##### data prepar end
+    ##### data preparation end
 
     ##### backtest
-    backtest = Backtest( data, goldhand_line_strategy, plot_title =plot_title, buy_at='gold', sell_at='grey')
+    backtest = Backtest( data, goldhand_line_strategy, plot_title =plot_title, buy_at= buy_at, sell_at=sell_at)
     trades =backtest.trades
 
     if ndays!=0:

{goldhand-15.5 → goldhand-15.7}/goldhand/strategy_rsi.py

@@ -9,6 +9,14 @@ from goldhand import *
 
 
 def rsi_strategy(data, buy_threshold = 30, sell_threshold = 70):
+    """
+    RSI strategy for backtesting with Backtest class
+    
+    Parameters:
+    - data: pandas DataFrame with columns: date, open, high, low, close, volume and rsi
+    - buy_threshold: int, default 30,  buy when RSI is below this value
+    - sell_threshold: int, default 70, sell when RSI is above this value
+    """
 
     in_trade = False  # Flag to track if already in a trade
     trade_id = 1
@@ -78,6 +86,17 @@ def rsi_strategy(data, buy_threshold = 30, sell_threshold = 70):
 
 
 def show_indicator_rsi_strategy(ticker, buy_threshold = 30, sell_threshold = 70, plot_title = '', ndays=0, plot_height=1000, add_strategy_summary = True):
+    """
+    Show RSI strategy result in one plot: candlestick chart, SMA lines, trades, RSI indicator, summary of the strategy on the left side of the plot
+    Parameters:
+    - ticker: str, ticker symbol
+    - buy_threshold: int, default 30,  buy when RSI is below this value
+    - sell_threshold: int, default 70, sell when RSI is above this value
+    - plot_title: str, default '', title of the plot
+    - ndays: int, default 0, number of days to show, if 0, show all data
+    - plot_height: int, default 1000, height of the plot
+    - add_strategy_summary: bool, default True, add strategy summary to the plot
+    """
 
     tdf = GoldHand(ticker).df
     backtest = Backtest( tdf, rsi_strategy, buy_threshold=buy_threshold, sell_threshold=sell_threshold)

{goldhand-15.5 → goldhand-15.7}/goldhand/tw.py

@@ -10,7 +10,7 @@ import json
 class Tw:
     def __init__(self):
         """ 
-        Get all stocks, cryptos and etfs from TradingView
+        Get all stock, crypto and ETF data from TradingView
         """
         self.stock = pd.DataFrame()
         self.crypto = pd.DataFrame()
@@ -22,8 +22,9 @@ class Tw:
 
     def get_all_stock(self):
         """ 
-        Get all stocks from TradingView
+        Get all stocks data from TradingView
         """
+        
         data_query = '{"filter":[{"left":"type","operation":"in_range","right":["stock","dr","fund"]},{"left":"subtype","operation":"in_range","right":["common","foreign-issuer","","etf","etf,odd","etf,otc","etf,cfd"]},{"left":"exchange","operation":"in_range","right":["AMEX","NASDAQ","NYSE"]},{"left":"is_primary","operation":"equal","right":true},{"left":"active_symbol","operation":"equal","right":true}],"options":{"lang":"en"},"markets":["america"],"symbols":{"query":{"types":[]},"tickers":[]},"columns":["logoid","name","close","change","change_abs","Recommend.All","volume","Value.Traded","market_cap_basic","price_earnings_ttm","earnings_per_share_basic_ttm","number_of_employees","sector","High.3M","Low.3M","Perf.3M","Perf.5Y","High.1M","Low.1M","High.6M","Low.6M","Perf.6M","beta_1_year","price_52_week_high","price_52_week_low","High.All","Low.All","BB.lower","BB.upper","change|1M","change_abs|1M","change|1W","change_abs|1W","change|240","country","EMA50","EMA100","EMA200","MACD.macd","MACD.signal","Mom","Perf.1M","RSI7","SMA50","SMA100","SMA200","Stoch.RSI.K","Stoch.RSI.D","Perf.W","Perf.Y","Perf.YTD","industry","Perf.All","description","type","subtype","update_mode","pricescale","minmov","fractional","minmove2","Mom[1]","RSI7[1]","Rec.Stoch.RSI","currency","fundamental_currency_code"],"sort":{"sortBy":"market_cap_basic","sortOrder":"desc"},"range":[0,8000]}'
         response = requests.post('https://scanner.tradingview.com/america/scan', data=data_query)
         data = response.json()
@@ -35,8 +36,9 @@ class Tw:
 
     def get_all_crypto(self):
         """
-        Get all cryptos from TradingView       
+        Get all crypto data from TradingView       
         """
+
         data_query = '{"columns":["base_currency","base_currency_desc","base_currency_logoid","update_mode","type","typespecs","exchange","crypto_total_rank","close","pricescale","minmov","fractional","minmove2","currency","24h_close_change|5","market_cap_calc","fundamental_currency_code","24h_vol_cmc","circulating_supply","crypto_common_categories","crypto_blockchain_ecosystems"],"ignore_unknown_fields":false,"options":{"lang":"en"},"range":[0,300],"sort":{"sortBy":"crypto_total_rank","sortOrder":"asc"},"markets":["coin"]}'
         response = requests.post('https://scanner.tradingview.com/coin/scan', data=data_query)
         data = response.json()
@@ -49,7 +51,7 @@ class Tw:
 
     def get_all_etf(self):
         """
-        Get all etfs from TradingView
+        Get all ETFs from TradingView
         """
         data_query = '{"columns":["name","description","logoid","update_mode","type","typespecs","close","pricescale","minmov","fractional","minmove2","currency","change","Value.Traded","relative_volume_10d_calc","aum","fundamental_currency_code","nav_total_return.5Y","expense_ratio","asset_class.tr","focus.tr","nav_discount_premium","category.tr","brand.tr","niche.tr"],"ignore_unknown_fields":false,"options":{"lang":"en"},"price_conversion":{"to_symbol":true},"range":[0,3000],"sort":{"sortBy":"aum","sortOrder":"desc"},"markets":["america"],"filter2":{"operator":"and","operands":[{"operation":{"operator":"or","operands":[{"operation":{"operator":"and","operands":[{"expression":{"left":"typespecs","operation":"has","right":["etn"]}}]}},{"operation":{"operator":"and","operands":[{"expression":{"left":"typespecs","operation":"has","right":["etf"]}}]}}]}}]}}'
         response = requests.post('https://scanner.tradingview.com/america/scan', data=data_query)
@@ -58,27 +60,42 @@ class Tw:
         self.etf = pd.DataFrame(list_elements)
         self.etf.columns = json.loads(data_query)['columns']
         self.etf = self.etf[self.etf['name'].str.contains('\\.')!=True]
+    
+    
+    def moneystring( self, money):
+        """
+        Convert money to string with unit
+        Parameters:
+        - money: money to convert
+        Return: string with unit
+        """
+
+        if money > 1_000_000_000_000:
+            money_str = f"{round(money / 1_000_000_000_000, 2)} Trillion" 
+        elif money>1_000_000_000:
+            money_str = f"{round(money / 1_000_000_000, 2)} Billion"
+        else:
+            money_str = f"{round(money / 1_000_000, 2)} Million"
+        return money_str
 
 
     def get_one_stock_info(self, ticker):
         """
         Get info about one stock
-        param: ticker
-        return: dict
+        Parameters: 
+        - ticker: ticker of the stock
+        Return: type dictioanry
         """
+
         ticker = ticker.upper()
         one_row = self.stock.loc[self.stock['name']==ticker,].iloc[0]
         tsec = self.stock.loc[self.stock['sector']==one_row['sector']].reset_index(drop=True)
         tind = self.stock.loc[self.stock['industry']== one_row['industry']].reset_index(drop=True)
-        if one_row['market_cap_basic'] <100_000_000:
-            market_cap = f"💲{round(one_row['market_cap_basic']/1_000_000, 2):,} Million"
-        else:
-            market_cap = f"💲{round(one_row['market_cap_basic']/1_000_000):,} Million"
         return({'ticker': one_row['name'],
                 'price': one_row['close'],
                 'market_cap': one_row['market_cap_basic'] ,
                 'n_emp': one_row['number_of_employees'],
-                'market_cap_text': market_cap,
+                'market_cap_text': self.moneystring(one_row['market_cap_basic']),
                 'name': one_row['description'],
                 'sector': one_row['sector'],
                 'industry': one_row['industry'],
@@ -91,85 +108,102 @@ class Tw:
     def get_top_n_stocks_by_sector(self,percent=10):
         """
         Get top n % stocks by sector
-        param: percent
-        return: pandas dataframe
+        Parameters: 
+        - percent: percent of stocks to return
+        Return: Pandas DataFrame
         """
-        return( 
-               (
+
+        return ( 
                 self.stock.groupby('sector')
                  .apply(lambda x: x.nlargest(int(len(x) * round((percent/100),2) ), 'market_cap_basic'))
                  .reset_index(drop=True)
-                )
+                
             )
-
+        
 
     def get_plotly_title(self, ticker):
         """
         Get plotly title for stock or crypto
-        param: ticker
-        return: str
+        Parameters: 
+        - ticker: ticker of the stock
+        Return: Summary of the stock or crypto to be used as plotly title
         """
 
         if '-USD' in ticker:
+            # crypto
             coin = self.crypto.loc[self.crypto['ticker']==ticker].iloc[0]
-            plotly_title = f"{coin['base_currency_desc']} ({coin['base_currency']})<br>💲{round(coin['market_cap_calc']/1_000_000):,} Million | {', '.join(coin['crypto_common_categories'])}"
+            plotly_title = f"{coin['base_currency_desc']} ({coin['base_currency']})<br>💲{self.moneystring(coin['market_cap_calc'])} | {', '.join(coin['crypto_common_categories'])}"
+        
+        elif ticker in self.etf['name'].tolist() :
+            # ETF
+            t = self.etf.loc[self.etf['name']==ticker].iloc[0]
+            plotly_title = f"{t['description']} ({t['name']}) | 💲{round(t['close'], 2)} <br>AUM:💲{self.moneystring(t['aum'])} | {t['focus.tr']}  | Expense ratio {t['expense_ratio']}"
         else:
+            # stock
             t = self.get_one_stock_info(ticker)
-            plotly_title = f"{t['name']} ({t['ticker']}) | 💲{round(t['price'], 2)} | {t['sector']} | {t['industry']} <br>{t['market_cap_text']} | 👨‍💼 {round(t['n_emp']):,} <br>Sector location: {t['sec_loc']} | Industry location: {t['ind_loc']}"
+            plotly_title = f"{t['name']} ({t['ticker']}) | 💲{round(t['price'], 2)} | {t['sector']} | {t['industry']} <br>💲{t['market_cap_text']} | 👨‍💼 {round(t['n_emp']):,} <br>Sector location: {t['sec_loc']} | Industry location: {t['ind_loc']}"
         return(plotly_title)
 
+
     def get_sec_plot(self, ticker):
         """
-        Get plotly figure for a ticker showing the sector location of the ticker
-        param: ticker
-        return: plotly figure
+        Get plotly figure for a ticker showing the sector location of the stock 
+        Parameters: 
+        - ticker: ticker of the stock
+        Return: plotly figure showing the sector location of the stock
         """
+
+        if ticker in self.etf['name'].tolist():
+            return f"{ticker} is ETF use get_etf_plot"
         row_df = self.stock.loc[self.stock['name']==ticker]
         row_df.rename(columns = {'description': 'Company'}, inplace=True)
         secdf = self.stock.loc[ (self.stock['sector'] ==row_df['sector'].iloc[0] ) ].reset_index(drop=True)
-        if row_df['market_cap_basic'].iloc[0] <100_000_000:
-            market_cap = f"💲{round(row_df['market_cap_basic'].iloc[0]/1_000_000, 2):,} Million"
-        else:
-            market_cap = f"💲{round(row_df['market_cap_basic'].iloc[0]/1_000_000):,} Million"
-
-
         secdf.rename(columns = {'description': 'Company'}, inplace=True)
-
-        #fig = px.bar(secdf, x='name', y='market_cap_basic', title = f"{row_df['Company'].iloc[0]} ({row_df['name'].iloc[0]})<br>{row_df['sector'].iloc[0]} | {row_df['industry'].iloc[0]}", labels={'market_cap_basic':'Market kapitalization'}, text='Company')
-
-        fig = px.bar(secdf, x='name', y='market_cap_basic', title = self.get_plotly_title(ticker), labels={'market_cap_basic':'Market kapitalization'}, text='Company')
-
-        fig.add_annotation( x=row_df['name'].iloc[0], y=row_df['market_cap_basic'].iloc[0], text= f"{market_cap}",  showarrow=True, align="center", bordercolor="#c7c7c7", font=dict(family="Courier New, monospace", size=16, color="#214e34" ), borderwidth=2, borderpad=4, bgcolor="#f4fdff", opacity=0.8, arrowhead=2, arrowsize=1, arrowwidth=1, ax=65,ay=-45)
+        fig = px.bar(secdf, x='name', y='market_cap_basic', title = f"{row_df['Company'].iloc[0]} ({row_df['name'].iloc[0]})<br>{row_df['sector'].iloc[0]} | {row_df['industry'].iloc[0]}", labels={'market_cap_basic':'Market kapitalization'}, text='Company')
+        fig = px.bar(secdf, x='name', y='market_cap_basic', title = self.get_plotly_title(ticker), labels={'market_cap_basic':'Market Capitalization'}, text='Company')
+        fig.add_annotation( x=row_df['name'].iloc[0], y=row_df['market_cap_basic'].iloc[0], text= f"{self.moneystring(row_df['market_cap_basic'].iloc[0])}",  showarrow=True, align="center", bordercolor="#c7c7c7", font=dict(family="Courier New, monospace", size=16, color="#214e34" ), borderwidth=2, borderpad=4, bgcolor="#f4fdff", opacity=0.8, arrowhead=2, arrowsize=1, arrowwidth=1, ax=65,ay=-45)
         fig.update_layout(showlegend=False, plot_bgcolor='white', height=600)
         return (fig)
 
+
     def get_ind_plot(self, ticker):
         """
-        Get plotly figure for a ticker showing the industry location of the ticker
-        param: ticker
-        return: plotly figure
+        Get plotly figure for a ticker showing the industry location of the stock
+        Parameters: 
+        - ticker: ticker of the stock
+        Return: plotly figure showing the industry location of the stock
         """
-        
+
+        if ticker in self.etf['name'].tolist():
+            return f"{ticker} is ETF use get_etf_plot"
         row_df = self.stock.loc[self.stock['name']==ticker]
         inddf = self.stock.loc[ (self.stock['industry'] ==row_df['industry'].iloc[0] ) ].reset_index(drop=True)
         row_df.rename(columns = {'description': 'Company'}, inplace=True)
-
-        if row_df['market_cap_basic'].iloc[0] <100_000_000:
-            market_cap = f"💲{round(row_df['market_cap_basic'].iloc[0]/1_000_000, 2):,} Million"
-        else:
-            market_cap = f"💲{round(row_df['market_cap_basic'].iloc[0]/1_000_000):,} Million"
-
         inddf.rename(columns = {'description': 'Company'}, inplace=True)
-
-        fig = px.bar(inddf, x='name', y='market_cap_basic', title = self.get_plotly_title(ticker), labels={'market_cap_basic':'Market kapitalization'}, text='Company')
+        fig = px.bar(inddf, x='name', y='market_cap_basic', title = self.get_plotly_title(ticker), labels={'market_cap_basic':'Market Capitalization'}, text='Company')
+        fig.add_annotation( x=row_df['name'].iloc[0], y=row_df['market_cap_basic'].iloc[0], text= f"{self.moneystring(row_df['market_cap_basic'].iloc[0])}",  showarrow=True, align="center", bordercolor="#c7c7c7", font=dict(family="Courier New, monospace", size=16, color="#214e34" ), borderwidth=2, borderpad=4, bgcolor="#f4fdff", opacity=0.8, arrowhead=2, arrowsize=1, arrowwidth=1, ax=65,ay=-45)
+        fig.update_layout(showlegend=False, plot_bgcolor='white', height=600)
+        return (fig)
 
 
-        fig.add_annotation( x=row_df['name'].iloc[0], y=row_df['market_cap_basic'].iloc[0], text= f"{market_cap}",  showarrow=True, align="center", bordercolor="#c7c7c7", font=dict(family="Courier New, monospace", size=16, color="#214e34" ), borderwidth=2, borderpad=4, bgcolor="#f4fdff", opacity=0.8, arrowhead=2, arrowsize=1, arrowwidth=1, ax=65,ay=-45)
+    def get_etf_plot(self, ticker):
+        """
+        Get plotly figure for a ETF showing the ETF location with the same focus
+        Parameters:
+        - ticker: ticker of the ETF
+        Return: plotly figure showing the ETF location with the same focus
+        
+        """
+        row_df = self.etf.loc[self.etf['name']==ticker]
+        focdf = self.etf.loc[ (self.etf['focus.tr'] ==row_df['focus.tr'].iloc[0] ) ].reset_index(drop=True)
+        fig = px.bar(focdf, x='name', y='aum', title = self.get_plotly_title(ticker), labels={'aum':'Assets Under Management'}, text='description')
+        fig.add_annotation( x=row_df['name'].iloc[0], y=row_df['aum'].iloc[0], text= f"{self.moneystring(row_df['aum'].iloc[0])}",  showarrow=True, align="center", bordercolor="#c7c7c7", font=dict(family="Courier New, monospace", size=16, color="#214e34" ), borderwidth=2, borderpad=4, bgcolor="#f4fdff", opacity=0.8, arrowhead=2, arrowsize=1, arrowwidth=1, ax=65,ay=-45)
         fig.update_layout(showlegend=False, plot_bgcolor='white', height=600)
         return (fig)
         
 
 
+
 #tw = Tw()
 #print(tw.stock.head(1).T)
 #print(tw.crypto.head(1).T)

{goldhand-15.5 → goldhand-15.7}/goldhand.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: goldhand
-Version: 15.5
+Version: 15.7
 Summary: A package working with financial data
 Home-page: https://github.com/misrori/goldhand
 Author: Mihaly

{goldhand-15.5 → goldhand-15.7}/setup.py

@@ -8,7 +8,7 @@ long_description = (this_directory / "README.md").read_text()
 
 setup(
     name="goldhand",
-    version="15.5",
+    version="15.7",
     author="Mihaly",
     author_email="ormraat.pte@gmail.com",
     description="A package working with financial data",
@@ -20,6 +20,5 @@ setup(
     # other arguments omitted
     long_description=long_description,
     long_description_content_type='text/markdown'
-
 )