def process_rows(symbols):
client = NotionClient(token_v2=NOTION_V2_TOKEN)
page = client.get_block(NOTION_PAGE_LINK)
database = page.children[DATABASE_POSITION].collection
updating_indicator = page.children[UPDATING_POSITION]
success = True
failed_to_load = []
for i, symbol in enumerate(symbols):
stock = yf.Ticker(symbol)
updating_indicator.title = "🟡 _Stocks are currently updating_"
try:
info = stock.info
hist = stock.history(period="1y")
name = info['shortName']
descr = info['longBusinessSummary']
price = info['open']
category = info['category']
min52Week = min(hist['Close'])
max52Week = max(hist['Close'])
latestPrice = hist['Close'].values.tolist()[-1]
ratio = (latestPrice - min52Week) / (max52Week - min52Week)
sector = info['sector']
dividend_rate = info[
'trailingAnnualDividendYield'] if 'trailingAnnualDividendYield' in info else 0.0
pe_ratio = info['forwardPE'] if 'forwardPE' in info else -1
market_cap = info['marketCap'] if 'marketCap' in info else 0
price_to_sales_ratio = info[
'priceToSalesTrailing12Months'] if 'priceToSalesTrailing12Months' in info else 0.0
eps = info['forwardEps'] if 'forwardEps' in info else 0.0
row = database.add_row(
symbol=symbol,
name=name,
price=price,
year_low=min52Week,
year_high=max52Week,
ratio=round(ratio or -1, 2),
dividend=dividend_rate
and round(dividend_rate * 1000, 2) / 1000,
sector=sector or "",
description=descr[0:150] + "...",
pe=round(pe_ratio or -1, 2),
market_cap='$' + str(round(market_cap / 1000000000 or -1, 1)) +
' B',
price_to_sales=round(price_to_sales_ratio or -1, 2),
eps=eps)
except:
success = False
failed_to_load.append(symbol)
if success:
updating_indicator.title = "✅ Stocks have been updated as of `%s`" % get_datetime(
)
else:
updating_indicator.title = "🚨 Error loading %s" % ', '.join(
failed_to_load)
def main():
client = NotionClient(token_v2=NOTION_V2_TOKEN)
page = client.get_block(NOTION_PAGE_LINK)
refresh_button = page.children[REFRESH_BUTTON_POSITION]
# symbols = get_stock_symbols()
# enter_symbols(symbols)
populate_stock_data()
def callback():
client = NotionClient(token_v2=NOTION_V2_TOKEN)
page = client.get_block(NOTION_PAGE_LINK)
refresh_button = page.children[REFRESH_BUTTON_POSITION]
updating_indicator = page.children[UPDATING_POSITION]
if refresh_button.checked == True:
refresh_button.checked = False
main()
def get_stock_symbols():
client = NotionClient(token_v2=NOTION_V2_TOKEN)
page = client.get_block(NOTION_PAGE_LINK)
refresh_button = page.children[STOCK_SYMBOLS_LIST]
clear_table()
raw_symbols = refresh_button.title
symbols = [x.strip() for x in raw_symbols.split(",")]
return symbols
def clear_table():
client = NotionClient(token_v2=NOTION_V2_TOKEN)
page = client.get_block(NOTION_PAGE_LINK)
database = page.children[DATABASE_POSITION].collection
updating_indicator = page.children[UPDATING_POSITION]
updating_indicator.title = "🗑 Clearing Stocks table"
rows = database.get_rows()
for row in rows:
row.remove()
updating_indicator.title = "🌱 Stocks table is ready to go"
def populate_stock_data():
client = NotionClient(token_v2=NOTION_V2_TOKEN)
page = client.get_block(NOTION_PAGE_LINK)
database = page.children[DATABASE_POSITION].collection
symbols = get_stock_symbols()
PARTITIONS = min(len(symbols), 10)
executor = ThreadPoolExecutor(max_workers=PARTITIONS)
for symbols_sub in [
symbols[int(i * len(symbols) /
PARTITIONS):int(i * len(symbols) / PARTITIONS +
len(symbols) / PARTITIONS)]
for i in range(PARTITIONS)
]:
executor.submit(process_rows, (symbols_sub))
def enter_symbols(symbols):
client = NotionClient(token_v2=NOTION_V2_TOKEN)
page = client.get_block(NOTION_PAGE_LINK)
database = page.children[DATABASE_POSITION].collection
updating_indicator = page.children[UPDATING_POSITION]
updating_indicator.title = "🔥 Updating Symbols -- progress: %s/%s" % (
str(0), str(len(symbols)))
for i, symbol in enumerate(symbols):
row = database.add_row()
row.symbol = symbol
updating_indicator.title = "🔥 Updating Symbols -- progress: %s/%s" % (
str(i + 1), str(len(symbols)))
updating_indicator.title = "👀 Symbols updated!"
def main():
client = NotionClient(token_v2=NOTION_V2_TOKEN)
page = client.get_block(NOTION_PAGE_LINK)
r = requests.get(url=API_URL)
data = r.json()
articles = data['articles'] or []
for child in page.children:
child.remove()
callout = page.children.add_new(
CalloutBlock,
title="News articles have been auto updated as of %s" % get_datetime())
for article in articles:
add_article(page, article)
def notion_update(data):
client = NotionClient(token_v2=NOTION_V2_TOKEN)
block = client.get_block(NOTION_PAGE_LINK)
cv = block.collection
rows = cv.get_rows()
cv.clear_properties()
for row in rows:
row.remove()
title = data[0][0]
cv.set_property('title', title, slug=title, property_type='title')
for i, prop in enumerate(data[0][1:]):
prop_type = get_prop_type(data[1][1:][i]) if data[1] and data[1][1:] else 'text'
cv.set_property(prop, prop, slug=prop, property_type=prop_type)
for row in data[1:]:
new_row = cv.add_row()
for i, value in enumerate(row):
prop = data[0][i]
new_row.set_property(prop, value)