Skip to content

Commit

Permalink
format email content enrichment
Browse files Browse the repository at this point in the history
  • Loading branch information
JackZhao516 committed Jan 6, 2025
1 parent 6ead240 commit 3cba86b
Showing 1 changed file with 9 additions and 4 deletions.
13 changes: 9 additions & 4 deletions smrti_quant_alerts/alerts/stock_alerts/stock_screener_alert.py
Original file line number Diff line number Diff line change
Expand Up @@ -211,18 +211,19 @@ def _get_newly_added_stocks_str(self) -> str:
StockAlertDBUtils.reset_stocks(screener_name)
StockAlertDBUtils.add_stocks(stocks, screener_name)
if not new_stocks:
content += f" ·{screener_name} no newly added stocks\n"
content += f" ~{screener_name} no newly added stocks\n"
else:
content += f" ·{screener_name} newly added stocks:\n"
content += f" ~{screener_name} newly added stocks:\n"
for sector, sector_stocks in sector_to_stocks.items():
sector_stocks_set = set(sector_stocks)
chosen_stocks = [stock.ticker for stock in new_stocks if stock in sector_stocks_set]
if chosen_stocks:
if sector == "":
sector = "Others"
content += f" ·{sector}: {chosen_stocks}\n"
content += "\n\n"

return content + "\n"
return content + "\n\n"

def _get_screener_stocks_in_top_performer(self) -> str:
"""
Expand All @@ -236,11 +237,15 @@ def _get_screener_stocks_in_top_performer(self) -> str:
content += f"~Top {self._price_top_percent}% price performer in {timeframe}:\n"
for screener_name, screener_stocks in self._screener_name_to_stocks.items():
screener_stocks = set(screener_stocks)
content += f" ·{screener_name} top performer: \n"
content += f" ~{screener_name} top performer: \n"
for sector, top_performers in value.items():
chosen_stocks = [stock[0] for stock in top_performers if stock[0] in screener_stocks]
if chosen_stocks:
if sector == "":
sector = "Others"
content += f" ·{sector}: {chosen_stocks}\n"
content += "\n\n"
content += "\n\n"

return content + "\n"

Expand Down

0 comments on commit 3cba86b

Please sign in to comment.