wuliaozhiji
125 lines
3.9 KiB
Python
125 lines
3.9 KiB
Python
import os
|
|
import re
|
|
import sys
|
|
import time
|
|
import traceback
|
|
|
|
import requests
|
|
from selenium.webdriver import Chrome
|
|
from selenium.webdriver.chrome.options import Options
|
|
from selenium.webdriver.common.by import By
|
|
|
|
# Config
|
|
# Price increase ratio threshold (ignore everything higher than this ratio)
|
|
INCR_MAX = 0.2
|
|
# Telegram chat ID that receives update messages (could be a channel in @channel_id format)
|
|
# TG_RECEIVER = 1770239825
|
|
TG_RECEIVER = '@toronto_bestbuy_gpu'
|
|
# Telegram bot token
|
|
TG_TOKEN = os.environ['TG_TOKEN']
|
|
|
|
# Constants
|
|
CSS = By.CSS_SELECTOR
|
|
MODELS = [
|
|
['3060 ti', 400, 132],
|
|
['3070 ti', 600, 167],
|
|
['3080 ti', 1200, 233],
|
|
['3050', 250, 73],
|
|
['3060', 330, 98],
|
|
['3070', 500, 154],
|
|
['3080', 700, 204],
|
|
['3090', 1500, 236],
|
|
]
|
|
USD_TO_CAD = 1.27
|
|
AVAIL_TABLE: dict[str, bool] = {}
|
|
TITLE_SHORTEN = re.compile('(rtx|nvidia|geforce|edition|gddr[56]x*|video|card)', flags=re.IGNORECASE)
|
|
|
|
|
|
def shorten_title(title: str):
|
|
short_title = TITLE_SHORTEN.sub('', title)
|
|
while ' ' in short_title:
|
|
short_title = short_title.replace(' ', ' ')
|
|
return short_title.strip()
|
|
|
|
|
|
def send_message(msg: str):
|
|
r = requests.get(f'https://api.telegram.org/bot{TG_TOKEN}/sendMessage',
|
|
params={'chat_id': TG_RECEIVER, 'parse_mode': 'Markdown', 'text': msg})
|
|
|
|
if r.status_code != 200:
|
|
print('Request not OK:', r.status_code, r.text)
|
|
|
|
|
|
def parse_page(browser: Chrome):
|
|
# Parse page
|
|
for item in browser.find_elements(By.CLASS_NAME, 'x-productListItem'):
|
|
title = item.find_element(CSS, 'div[data-automation="productItemName"]').get_attribute('innerHTML')
|
|
title = shorten_title(title)
|
|
|
|
# Check availability
|
|
avail = item.find_elements(CSS, 'div[data-automation="store-availability-messages"] span[data-automation="store-availability-checkmark"]')
|
|
|
|
# Not available, check if it was previously available
|
|
if len(avail) == 0:
|
|
if title in AVAIL_TABLE:
|
|
send_message(f'Sold out: `{title}`')
|
|
del AVAIL_TABLE[title]
|
|
continue
|
|
|
|
# Already sent availability message
|
|
if title in AVAIL_TABLE:
|
|
continue
|
|
|
|
# Get price
|
|
price = item.find_element(CSS, 'div[data-automation="product-pricing"] > span > div').get_attribute('innerHTML')
|
|
price = round(float(price.replace(',', '')[1:]))
|
|
price_usd = price / USD_TO_CAD
|
|
|
|
# Find model
|
|
lower = title.lower()
|
|
model = [c for c in MODELS if c[0] in lower]
|
|
if not model:
|
|
continue
|
|
model = model[0]
|
|
|
|
# Calculate price increase
|
|
price_incr = (price_usd - model[1]) / model[1]
|
|
value = model[2] / price_usd * 100
|
|
|
|
# Check incr threshold
|
|
if price_incr > INCR_MAX:
|
|
print(' '.join(title.split()[:2]), f'is in stock but price increase {price_incr * 100:.0f}% '
|
|
f'is larger than defined threshold.')
|
|
# continue
|
|
|
|
# Find link
|
|
link = item.find_element(CSS, 'a').get_attribute('href')
|
|
|
|
# Available and meets threshold criteria, notify user
|
|
AVAIL_TABLE[title] = True
|
|
send_message(f'{model[0].upper()} Became Available!\n'
|
|
f'\n'
|
|
f'${price:.0f} | {price_incr * 100:.0f}% Incr | Value: {value:.0f}\n'
|
|
f'- [{title}]({link})')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
web_options = Options()
|
|
# web_options.headless = True
|
|
|
|
browser = Chrome(options=web_options)
|
|
browser.get('https://www.bestbuy.ca/en-ca/collection/graphics-cards-with-nvidia-chipset/349221')
|
|
|
|
# parse_page(browser)
|
|
# browser.close()
|
|
|
|
# Refresh indefinitely
|
|
while True:
|
|
try:
|
|
parse_page(browser)
|
|
except Exception as e:
|
|
traceback.print_exc()
|
|
time.sleep(5)
|
|
browser.refresh()
|
|
time.sleep(2)
|