Compare commits
13 Commits
Author | SHA1 | Date | |
---|---|---|---|
1b15946a55 | |||
af86d4aa8f | |||
f2b5b184ea | |||
50a9c1f827 | |||
584c222076 | |||
70a7df1f97 | |||
70b723b781 | |||
62ef15c928 | |||
fe3e8f9609 | |||
a95d1d41d5 | |||
b2789c8942 | |||
998c7dd500 | |||
4699f686a2 |
@ -4,13 +4,8 @@ RUN apk add --no-cache ca-certificates
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
COPY nvidia-stock-bot.py /app/
|
||||
|
||||
COPY requirements.txt /app/
|
||||
COPY /app/ /app/
|
||||
|
||||
RUN pip install --no-cache-dir -r requirements.txt
|
||||
|
||||
ENV DISCORD_WEBHOOK_URL="https://example.com/webhook" \
|
||||
REFRESH_TIME="30"
|
||||
|
||||
CMD ["python", "nvidia-stock-bot.py"]
|
||||
CMD ["python", "main.py"]
|
21
README.md
21
README.md
@ -27,11 +27,13 @@
|
||||
|
||||
## ✨ Features
|
||||
|
||||
- Discord `@everyone` notification on SKU change (possible imminent drop)
|
||||
- Discord `@everyone` notification when stock is detected, including model, price, and link
|
||||
- Selectable GPU models
|
||||
- Discord `@everyone` or specified role on SKU change (possible imminent drop)
|
||||
- Discord `@everyone` or specified role notification when stock is detected, including model, price, and link
|
||||
- Silent Discord notification when no stock is detected
|
||||
- Selectable notification language
|
||||
- Selectable notification server name in footer
|
||||
- Selectable check frequency
|
||||
- Selectable GPU model
|
||||
|
||||
## 🐳 Docker Installation without the repo (quick)
|
||||
|
||||
@ -52,22 +54,23 @@ services:
|
||||
container_name: nvidia-stock-bot
|
||||
restart: unless-stopped
|
||||
environment:
|
||||
# Minimal environment variables
|
||||
- PRODUCT_NAMES= # Exact GPU name (e.g. "RTX 5080, RTX 5090")
|
||||
- DISCORD_ROLES= # List of Discord roles ID (e.g. "<@&12345>, <@&6789>"), in the same order than PRODUCT_NAMES values. @everyone by default.
|
||||
- DISCORD_WEBHOOK_URL= # Your Discord webhook URL
|
||||
- API_URL_SKU= # API listing the product
|
||||
- API_URL_STOCK= # API providing stock data
|
||||
- PRODUCT_URL= # GPU purchase URL
|
||||
- API_URL_SKU= # API listing the product for your country
|
||||
- API_URL_STOCK= # API providing stock data for your country
|
||||
- PRODUCT_URL= # GPU purchase URL for your country
|
||||
- PYTHONUNBUFFERED=1 # Enables real-time log output
|
||||
command: python nvidia-stock-bot.py
|
||||
```
|
||||
|
||||
**Environment Variables:**
|
||||
|
||||
| Variable | Description | Possible Values | Default Value |
|
||||
|---------------------|-------------------------------------------------|------------------------------------------------------------------|-------------------------------------------------------------------------------------------------------------|
|
||||
| `DISCORD_WEBHOOK_URL` | Your Discord webhook URL | A valid URL | |
|
||||
| `PRODUCT_NAMES` | The exact GPU names you're searching for | `RTX 5080, RTX 5090` | |
|
||||
| `DISCORD_WEBHOOK_URL` | Your Discord webhook URL | A valid URL | |
|
||||
| `DISCORD_SERVER_NAME` | The name of your server, displayed in notification's footer | A text | Shared for free |
|
||||
| `DISCORD_NOTIFICATION_LANGUAGE` | Your language for notification's content | `bg`, `cs`, `da`, `de`, `el`, `en`, `es`, `et`, `fi`, `fr`, `ga`, `hr`, `hu`, `it`, `lt`, `lv`, `mt`, `nl`, `pl`, `pt`, `ro`, `sk`, `sl`, `sv` | `en` |
|
||||
| `DISCORD_ROLES` | List of Discord roles ID in the same order than `PRODUCT_NAMES` values, found in your discord server settings (with user profile developer mode enabled) | `<@&12345><@&6789>` | @everyone |
|
||||
| `REFRESH_TIME` | Script refresh interval in seconds | `60`, `30`, etc. | `30` |
|
||||
| `API_URL_SKU` | API listing the product | A URL. API url can change over time. For now, you can use the default one and change the `locale` parameter to yours (e.g. `locale=en-gb`) | `https://api.nvidia.partners/edge/product/search?page=1&limit=100&locale=fr-fr&Manufacturer=Nvidia` |
|
||||
|
143
app/env_config.py
Normal file
143
app/env_config.py
Normal file
@ -0,0 +1,143 @@
|
||||
import os
|
||||
import re
|
||||
import logging
|
||||
import json
|
||||
import sys
|
||||
|
||||
# Logger setup
|
||||
logging.basicConfig(
|
||||
level=logging.INFO,
|
||||
format="%(asctime)s [%(levelname)s] %(message)s",
|
||||
)
|
||||
logging.info("Script started")
|
||||
|
||||
try:
|
||||
# Env variables
|
||||
DISCORD_WEBHOOK_URL = os.environ['DISCORD_WEBHOOK_URL']
|
||||
DISCORD_SERVER_NAME = os.environ.get('DISCORD_SERVER_NAME', 'Shared for free')
|
||||
API_URL_SKU = os.environ.get('API_URL_SKU', 'https://api.nvidia.partners/edge/product/search?page=1&limit=100&locale=fr-fr&Manufacturer=Nvidia')
|
||||
API_URL_STOCK = os.environ.get('API_URL_STOCK', 'https://api.store.nvidia.com/partner/v1/feinventory?locale=fr-fr&skus=')
|
||||
REFRESH_TIME = int(os.environ.get('REFRESH_TIME', 30))
|
||||
TEST_MODE = os.environ.get('TEST_MODE', 'False').lower() == 'true'
|
||||
PRODUCT_URL = os.environ.get('PRODUCT_URL', 'https://marketplace.nvidia.com/fr-fr/consumer/graphics-cards/?locale=fr-fr&page=1&limit=12&manufacturer=NVIDIA')
|
||||
DISCORD_ROLES = os.environ.get('DISCORD_ROLES')
|
||||
PRODUCT_NAMES = os.environ['PRODUCT_NAMES']
|
||||
except KeyError as e:
|
||||
logging.error(f"Missing environment variable: {e}")
|
||||
sys.exit(1)
|
||||
except ValueError:
|
||||
logging.error("REFRESH_TIME must be a valid integer.")
|
||||
sys.exit(1)
|
||||
|
||||
if not PRODUCT_NAMES:
|
||||
logging.error("❌ PRODUCT_NAMES is required but not defined.")
|
||||
sys.exit(1)
|
||||
|
||||
PRODUCT_NAMES = [name.strip() for name in PRODUCT_NAMES.split(',')]
|
||||
|
||||
# Role mapping
|
||||
DISCORD_ROLE_MAP = {}
|
||||
if not DISCORD_ROLES or not DISCORD_ROLES.strip():
|
||||
logging.warning("⚠️ DISCORD_ROLES not defined or empty. Defaulting all roles to @everyone.")
|
||||
for name in PRODUCT_NAMES:
|
||||
DISCORD_ROLE_MAP[name] = '@everyone'
|
||||
else:
|
||||
roles = [r.strip() if r.strip() else '@everyone' for r in DISCORD_ROLES.split(',')]
|
||||
if len(roles) != len(PRODUCT_NAMES):
|
||||
logging.error("❌ The number of DISCORD_ROLES must match PRODUCT_NAMES.")
|
||||
sys.exit(1)
|
||||
for name, role in zip(PRODUCT_NAMES, roles):
|
||||
if role != '@everyone' and not re.match(r'^<@&\d{17,20}>$', role):
|
||||
logging.error(f"❌ Invalid DISCORD_ROLE format for {name}: {role}")
|
||||
sys.exit(1)
|
||||
DISCORD_ROLE_MAP[name] = role
|
||||
|
||||
if not DISCORD_WEBHOOK_URL:
|
||||
logging.error("❌ DISCORD_WEBHOOK_URL is required but not defined.")
|
||||
sys.exit(1)
|
||||
|
||||
# Masked webhook for display
|
||||
match = re.search(r'/(\d+)/(.*)', DISCORD_WEBHOOK_URL)
|
||||
if match:
|
||||
webhook_id = match.group(1)
|
||||
webhook_token = match.group(2)
|
||||
masked_webhook_id = webhook_id[:len(webhook_id) - 10] + '*' * 10
|
||||
masked_webhook_token = webhook_token[:len(webhook_token) - 120] + '*' * 10
|
||||
wh_masked_url = f"https://discord.com/api/webhooks/{masked_webhook_id}/{masked_webhook_token}"
|
||||
else:
|
||||
wh_masked_url = "[Invalid webhook URL]"
|
||||
|
||||
# Test mode
|
||||
if TEST_MODE:
|
||||
logging.warning("🚧 Test mode is active. No real alerts will be sent.")
|
||||
|
||||
# HTTP headers
|
||||
HEADERS = {
|
||||
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) "
|
||||
"AppleWebKit/537.36 (KHTML, like Gecko) "
|
||||
"Chrome/131.0.0.0 Safari/537.36",
|
||||
"Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,"
|
||||
"image/avif,image/webp,image/apng,*/*;q=0.8",
|
||||
"Accept-Encoding": "gzip, deflate, br, zstd",
|
||||
"Accept-Language": "en-US,en;q=0.5",
|
||||
"Cache-Control": "max-age=0",
|
||||
"Connection": "keep-alive",
|
||||
"Sec-Fetch-Dest": "document",
|
||||
"Sec-Fetch-Mode": "navigate",
|
||||
"Sec-Fetch-Site": "none",
|
||||
"Sec-Fetch-User": "?1",
|
||||
"Upgrade-Insecure-Requests": "1",
|
||||
"Sec-GPC": "1",
|
||||
}
|
||||
|
||||
# Localization loading
|
||||
language = os.environ.get("DISCORD_NOTIFICATION_LANGUAGE", "en").lower()
|
||||
try:
|
||||
with open("localization.json", "r", encoding="utf-8") as f:
|
||||
localization = json.load(f)
|
||||
except FileNotFoundError:
|
||||
logging.error("❌ localization.json file not found.")
|
||||
sys.exit(1)
|
||||
|
||||
required_keys = [
|
||||
"in_stock_title", "out_of_stock_title", "sku_change_title",
|
||||
"buy_now", "price", "time", "footer",
|
||||
"sku_description", "imminent_drop"
|
||||
]
|
||||
loc = localization.get(language, localization.get("en", {}))
|
||||
logging.info(f"Notification language: {language}")
|
||||
|
||||
if not loc:
|
||||
logging.warning(f"⚠️ Language '{language}' not found. Falling back to English.")
|
||||
loc = localization.get("en", {})
|
||||
language = "en"
|
||||
|
||||
missing_keys = [key for key in required_keys if key not in loc]
|
||||
fallback = localization.get("en", {})
|
||||
for key in missing_keys:
|
||||
if key in fallback:
|
||||
loc[key] = fallback[key]
|
||||
else:
|
||||
logging.error(f"❌ Missing required key '{key}' in both '{language}' and fallback 'en'.")
|
||||
sys.exit(1)
|
||||
|
||||
# Public constants
|
||||
in_stock_title = loc["in_stock_title"]
|
||||
out_of_stock_title = loc["out_of_stock_title"]
|
||||
sku_change_title = loc["sku_change_title"]
|
||||
buy_now = loc["buy_now"]
|
||||
price_label = loc["price"]
|
||||
time_label = loc["time"]
|
||||
footer = loc["footer"]
|
||||
sku_description = loc["sku_description"]
|
||||
imminent_drop = loc["imminent_drop"]
|
||||
|
||||
# Logging
|
||||
logging.info(f"GPU: {PRODUCT_NAMES}")
|
||||
logging.info(f"Discord Webhook URL: {wh_masked_url}")
|
||||
logging.info(f"Discord Role Mention: {DISCORD_ROLES}")
|
||||
logging.info(f"API URL SKU: {API_URL_SKU}")
|
||||
logging.info(f"API URL Stock: {API_URL_STOCK}")
|
||||
logging.info(f"Product URL: {PRODUCT_URL}")
|
||||
logging.info(f"Refresh time: {REFRESH_TIME} seconds")
|
||||
logging.info(f"Test Mode: {TEST_MODE}")
|
104
app/gpu_checker.py
Normal file
104
app/gpu_checker.py
Normal file
@ -0,0 +1,104 @@
|
||||
import requests
|
||||
import logging
|
||||
from env_config import API_URL_SKU, API_URL_STOCK, HEADERS, PRODUCT_NAMES, PRODUCT_URL
|
||||
from notifier import send_discord_notification, send_out_of_stock_notification, send_sku_change_notification
|
||||
|
||||
from requests.adapters import HTTPAdapter, Retry
|
||||
|
||||
session = requests.Session()
|
||||
retries = Retry(total=5, backoff_factor=1, status_forcelist=[500, 502, 503, 504])
|
||||
session.mount('https://', HTTPAdapter(max_retries=retries))
|
||||
|
||||
last_sku_dict = {}
|
||||
global_stock_status_dict = {}
|
||||
first_run_dict = {name: True for name in PRODUCT_NAMES}
|
||||
|
||||
def check_rtx_50_founders():
|
||||
global last_sku_dict, global_stock_status_dict, first_run_dict
|
||||
|
||||
try:
|
||||
response = session.get(API_URL_SKU, headers=HEADERS, timeout=10)
|
||||
logging.info(f"SKU API response: {response.status_code}")
|
||||
response.raise_for_status()
|
||||
data = response.json()
|
||||
except requests.exceptions.RequestException as e:
|
||||
logging.error(f"SKU API error: {e}")
|
||||
return
|
||||
|
||||
# All available products
|
||||
all_products = data['searchedProducts']['productDetails']
|
||||
|
||||
for product_name in PRODUCT_NAMES:
|
||||
product_details = None
|
||||
for p in all_products:
|
||||
if p.get("gpu", "").strip() == product_name:
|
||||
product_details = p
|
||||
break
|
||||
|
||||
if not product_details:
|
||||
logging.warning(f"⚠️ No product with GPU '{product_name}' found.")
|
||||
continue
|
||||
|
||||
product_sku = product_details['productSKU']
|
||||
product_upc = product_details.get('productUPC', "")
|
||||
if not isinstance(product_upc, list):
|
||||
product_upc = [product_upc]
|
||||
|
||||
# Check SKU change
|
||||
old_sku = last_sku_dict.get(product_name)
|
||||
if old_sku and old_sku != product_sku and not first_run_dict[product_name]:
|
||||
logging.warning(f"⚠️ SKU changed for {product_name}: {old_sku} → {product_sku}")
|
||||
send_sku_change_notification(product_name, old_sku, product_sku, PRODUCT_URL)
|
||||
|
||||
last_sku_dict[product_name] = product_sku
|
||||
first_run_dict[product_name] = False
|
||||
|
||||
# Stock check
|
||||
api_stock_url = API_URL_STOCK + product_sku
|
||||
logging.info(f"[{product_name}] Checking stock: {api_stock_url}")
|
||||
|
||||
try:
|
||||
response = session.get(api_stock_url, headers=HEADERS, timeout=10)
|
||||
logging.info(f"[{product_name}] Stock API response: {response.status_code}")
|
||||
response.raise_for_status()
|
||||
stock_data = response.json()
|
||||
except requests.exceptions.RequestException as e:
|
||||
logging.error(f"Stock API error: {e}")
|
||||
continue
|
||||
|
||||
products = stock_data.get("listMap", [])
|
||||
products_price = "Price not available"
|
||||
if isinstance(products, list) and len(products) > 0:
|
||||
for p in products:
|
||||
price = p.get("price", 'Price not available')
|
||||
if price != 'Price not available':
|
||||
products_price = price
|
||||
break
|
||||
else:
|
||||
logging.error(f"[{product_name}] Product list is empty or malformed.")
|
||||
|
||||
found_in_stock = set()
|
||||
for p in products:
|
||||
gpu_name = p.get("fe_sku", "").upper()
|
||||
is_active = p.get("is_active") == "true"
|
||||
if is_active and any(upc.upper() in gpu_name for upc in product_upc):
|
||||
found_in_stock.add(gpu_name)
|
||||
|
||||
for upc in product_upc:
|
||||
upc_upper = upc.upper()
|
||||
currently_in_stock = upc_upper in found_in_stock
|
||||
previously_in_stock = global_stock_status_dict.get((product_name, upc_upper), False)
|
||||
|
||||
if currently_in_stock and not previously_in_stock:
|
||||
send_discord_notification(product_name, PRODUCT_URL, products_price)
|
||||
global_stock_status_dict[(product_name, upc_upper)] = True
|
||||
logging.info(f"[{product_name}] {upc} is now in stock!")
|
||||
elif not currently_in_stock and previously_in_stock:
|
||||
send_out_of_stock_notification(product_name, PRODUCT_URL, products_price)
|
||||
global_stock_status_dict[(product_name, upc_upper)] = False
|
||||
logging.info(f"[{product_name}] {upc} is now out of stock.")
|
||||
elif currently_in_stock:
|
||||
logging.info(f"[{product_name}] {upc} still in stock.")
|
||||
else:
|
||||
logging.info(f"[{product_name}] {upc} still out of stock.")
|
||||
|
266
app/localization.json
Normal file
266
app/localization.json
Normal file
@ -0,0 +1,266 @@
|
||||
{
|
||||
"en": {
|
||||
"in_stock_title": "🚀 {gpu_name} IN STOCK!",
|
||||
"out_of_stock_title": "❌ {gpu_name} is out of stock",
|
||||
"sku_change_title": "🔄 SKU change detected for {gpu_name}",
|
||||
"buy_now": "**:point_right: [Buy now]({product_link})**",
|
||||
"price": "Price",
|
||||
"time": "Time",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Old SKU** : `{old_sku}`\n**New SKU** : `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Possible imminent drop!"
|
||||
},
|
||||
"bg": {
|
||||
"in_stock_title": "🚀 {gpu_name} ИМА НАЛИЧНОСТ!",
|
||||
"out_of_stock_title": "❌ {gpu_name} е изчерпан",
|
||||
"sku_change_title": "🔄 Засечена промяна на SKU за {gpu_name}",
|
||||
"buy_now": "**:point_right: [Купи сега]({product_link})**",
|
||||
"price": "Цена",
|
||||
"time": "Час",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Стар SKU**: `{old_sku}`\n**Нов SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Възможен наближаващ спад!"
|
||||
},
|
||||
"cs": {
|
||||
"in_stock_title": "🚀 {gpu_name} SKLADEM!",
|
||||
"out_of_stock_title": "❌ {gpu_name} je vyprodán",
|
||||
"sku_change_title": "🔄 Změna SKU nalezena pro {gpu_name}",
|
||||
"buy_now": "**:point_right: [Koupit nyní]({product_link})**",
|
||||
"price": "Cena",
|
||||
"time": "Čas",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Starý SKU**: `{old_sku}`\n**Nový SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Možný blížící se pokles!"
|
||||
},
|
||||
"da": {
|
||||
"in_stock_title": "🚀 {gpu_name} PÅ LAGER!",
|
||||
"out_of_stock_title": "❌ {gpu_name} er udsolgt",
|
||||
"sku_change_title": "🔄 SKU-ændring fundet for {gpu_name}",
|
||||
"buy_now": "**:point_right: [Køb nu]({product_link})**",
|
||||
"price": "Pris",
|
||||
"time": "Tid",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Gammelt SKU**: `{old_sku}`\n**Nyt SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Muligt snarligt drop!"
|
||||
},
|
||||
"de": {
|
||||
"in_stock_title": "🚀 {gpu_name} AUF LAGER!",
|
||||
"out_of_stock_title": "❌ {gpu_name} ist ausverkauft",
|
||||
"sku_change_title": "🔄 SKU-Änderung erkannt für {gpu_name}",
|
||||
"buy_now": "**:point_right: [Jetzt kaufen]({product_link})**",
|
||||
"price": "Preis",
|
||||
"time": "Zeit",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Alte SKU**: `{old_sku}`\n**Neue SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Möglicher bevorstehender Drop!"
|
||||
},
|
||||
"el": {
|
||||
"in_stock_title": "🚀 {gpu_name} ΔΙΑΘΕΣΙΜΟ!",
|
||||
"out_of_stock_title": "❌ {gpu_name} εξαντλήθηκε",
|
||||
"sku_change_title": "🔄 Ανίχνευση αλλαγής SKU για {gpu_name}",
|
||||
"buy_now": "**:point_right: [Αγόρασε τώρα]({product_link})**",
|
||||
"price": "Τιμή",
|
||||
"time": "Ώρα",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Παλαιό SKU**: `{old_sku}`\n**Νέο SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Πιθανή επικείμενη πτώση!"
|
||||
},
|
||||
"es": {
|
||||
"in_stock_title": "🚀 {gpu_name} ¡EN STOCK!",
|
||||
"out_of_stock_title": "❌ {gpu_name} está agotado",
|
||||
"sku_change_title": "🔄 Cambio de SKU detectado para {gpu_name}",
|
||||
"buy_now": "**:point_right: [Comprar ahora]({product_link})**",
|
||||
"price": "Precio",
|
||||
"time": "Hora",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**SKU antiguo**: `{old_sku}`\n**SKU nuevo**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ ¡Posible drop inminente!"
|
||||
},
|
||||
"et": {
|
||||
"in_stock_title": "🚀 {gpu_name} LAOS!",
|
||||
"out_of_stock_title": "❌ {gpu_name} on välja müüdud",
|
||||
"sku_change_title": "🔄 Tuvastatud SKU muudatus {gpu_name}",
|
||||
"buy_now": "**:point_right: [Osta kohe]({product_link})**",
|
||||
"price": "Hind",
|
||||
"time": "Aeg",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Vana SKU**: `{old_sku}`\n**Uus SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Võimalik varsti tulemas drop!"
|
||||
},
|
||||
"fi": {
|
||||
"in_stock_title": "🚀 {gpu_name} VARASTOSSA!",
|
||||
"out_of_stock_title": "❌ {gpu_name} on loppunut",
|
||||
"sku_change_title": "🔄 SKU-muutos havaittu {gpu_name}",
|
||||
"buy_now": "**:point_right: [Osta nyt]({product_link})**",
|
||||
"price": "Hinta",
|
||||
"time": "Aika",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Vanha SKU**: `{old_sku}`\n**Uusi SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Mahdollinen lähestyvä drop!"
|
||||
},
|
||||
"fr": {
|
||||
"in_stock_title": "🚀 {gpu_name} EN STOCK !",
|
||||
"out_of_stock_title": "❌ {gpu_name} est en rupture",
|
||||
"sku_change_title": "🔄 Changement de SKU détecté pour {gpu_name}",
|
||||
"buy_now": "**:point_right: [Acheter maintenant]({product_link})**",
|
||||
"price": "Prix",
|
||||
"time": "Heure",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Ancien SKU** : `{old_sku}`\n**Nouveau SKU** : `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Éventuel drop imminent!"
|
||||
},
|
||||
"hr": {
|
||||
"in_stock_title": "🚀 {gpu_name} NA ZALIHI!",
|
||||
"out_of_stock_title": "❌ {gpu_name} je rasprodan",
|
||||
"sku_change_title": "🔄 Promjena SKU za {gpu_name}",
|
||||
"buy_now": "**:point_right: [Kupi sada]({product_link})**",
|
||||
"price": "Cijena",
|
||||
"time": "Vrijeme",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Stari SKU**: `{old_sku}`\n**Novi SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Mogući nadolazeći drop!"
|
||||
},
|
||||
"hu": {
|
||||
"in_stock_title": "🚀 {gpu_name} RAKTÁRON!",
|
||||
"out_of_stock_title": "❌ {gpu_name} elfogyott",
|
||||
"sku_change_title": "🔄 SKU változás észlelve: {gpu_name}",
|
||||
"buy_now": "**:point_right: [Vásárlás most]({product_link})**",
|
||||
"price": "Ár",
|
||||
"time": "Idő",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Régi SKU**: `{old_sku}`\n**Új SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Lehetséges közelgő drop!"
|
||||
},
|
||||
"ga": {
|
||||
"in_stock_title": "🚀 {gpu_name} I STOC!",
|
||||
"out_of_stock_title": "❌ níl {gpu_name} ar fáil",
|
||||
"sku_change_title": "🔄 SKU athraithe aimsithe do {gpu_name}",
|
||||
"buy_now": "**:point_right: [Ceannaigh anois]({product_link})**",
|
||||
"price": "Praghas",
|
||||
"time": "Am",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Sean SKU**: `{old_sku}`\n**SKU Nua**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ I bhfad is gaire drop féideartha!"
|
||||
},
|
||||
"it": {
|
||||
"in_stock_title": "🚀 {gpu_name} DISPONIBILE!",
|
||||
"out_of_stock_title": "❌ {gpu_name} esaurito",
|
||||
"sku_change_title": "🔄 Modifica SKU rilevata per {gpu_name}",
|
||||
"buy_now": "**:point_right: [Acquista ora]({product_link})**",
|
||||
"price": "Prezzo",
|
||||
"time": "Ora",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Vecchio SKU**: `{old_sku}`\n**Nuovo SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Possibile drop imminente!"
|
||||
},
|
||||
"lv": {
|
||||
"in_stock_title": "🚀 {gpu_name} KRĀJUMĀ!",
|
||||
"out_of_stock_title": "❌ {gpu_name} nav pieejams",
|
||||
"sku_change_title": "🔄 SKU izmaiņas: {gpu_name}",
|
||||
"buy_now": "**:point_right: [Pirkt tagad]({product_link})**",
|
||||
"price": "Cena",
|
||||
"time": "Laiks",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Vecais SKU**: `{old_sku}`\n**Jaunais SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Iespējams gaidāms drop!"
|
||||
},
|
||||
"lt": {
|
||||
"in_stock_title": "🚀 {gpu_name} SANDĖLYJE!",
|
||||
"out_of_stock_title": "❌ {gpu_name} išparduotas",
|
||||
"sku_change_title": "🔄 SKU pakeitimas aptiktas: {gpu_name}",
|
||||
"buy_now": "**:point_right: [Pirkti dabar]({product_link})**",
|
||||
"price": "Kaina",
|
||||
"time": "Laikas",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Senas SKU**: `{old_sku}`\n**Naujas SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Rugsėjo drop'as gali artėti!"
|
||||
},
|
||||
"mt": {
|
||||
"in_stock_title": "🚀 {gpu_name} F'ĠESTA!",
|
||||
"out_of_stock_title": "❌ {gpu_name} mhux disponibbli",
|
||||
"sku_change_title": "🔄 Bidla SKU skoperta għal {gpu_name}",
|
||||
"buy_now": "**:point_right: [Ixtri issa]({product_link})**",
|
||||
"price": "Prezz",
|
||||
"time": "Ħin",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**SKU Antik**: `{old_sku}`\n**SKU Ġdid**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Possibbli drop imminenti!"
|
||||
},
|
||||
"nl": {
|
||||
"in_stock_title": "🚀 {gpu_name} OP VOORRAAD!",
|
||||
"out_of_stock_title": "❌ {gpu_name} is uitverkocht",
|
||||
"sku_change_title": "🔄 SKU-wijziging gedetecteerd voor {gpu_name}",
|
||||
"buy_now": "**:point_right: [Nu kopen]({product_link})**",
|
||||
"price": "Prijs",
|
||||
"time": "Tijd",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Oude SKU**: `{old_sku}`\n**Nieuwe SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Mogelijke aanstaande drop!"
|
||||
},
|
||||
"pl": {
|
||||
"in_stock_title": "🚀 {gpu_name} DOSTĘPNE!",
|
||||
"out_of_stock_title": "❌ {gpu_name} jest niedostępny",
|
||||
"sku_change_title": "🔄 Wykryto zmianę SKU dla {gpu_name}",
|
||||
"buy_now": "**:point_right: [Kup teraz]({product_link})**",
|
||||
"price": "Cena",
|
||||
"time": "Czas",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Stare SKU**: `{old_sku}`\n**Nowe SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Możliwy nadchodzący drop!"
|
||||
},
|
||||
"pt": {
|
||||
"in_stock_title": "🚀 {gpu_name} EM STOCK!",
|
||||
"out_of_stock_title": "❌ {gpu_name} esgotado",
|
||||
"sku_change_title": "🔄 Alteração de SKU detectada para {gpu_name}",
|
||||
"buy_now": "**:point_right: [Comprar agora]({product_link})**",
|
||||
"price": "Preço",
|
||||
"time": "Hora",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**SKU antigo**: `{old_sku}`\n**SKU novo**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Possível drop iminente!"
|
||||
},
|
||||
"ro": {
|
||||
"in_stock_title": "🚀 {gpu_name} ÎN STOC!",
|
||||
"out_of_stock_title": "❌ {gpu_name} este epuizat",
|
||||
"sku_change_title": "🔄 Schimbare SKU detectată pentru {gpu_name}",
|
||||
"buy_now": "**:point_right: [Cumpără acum]({product_link})**",
|
||||
"price": "Preț",
|
||||
"time": "Timp",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**SKU vechi**: `{old_sku}`\n**SKU nou**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Posibil drop iminent!"
|
||||
},
|
||||
"sk": {
|
||||
"in_stock_title": "🚀 {gpu_name} NA SKLADE!",
|
||||
"out_of_stock_title": "❌ {gpu_name} je vypredaný",
|
||||
"sku_change_title": "🔄 Zmena SKU zistená pre {gpu_name}",
|
||||
"buy_now": "**:point_right: [Kúp teraz]({product_link})**",
|
||||
"price": "Cena",
|
||||
"time": "Čas",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Starý SKU**: `{old_sku}`\n**Nový SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Možný nadchádzajúci drop!"
|
||||
},
|
||||
"sl": {
|
||||
"in_stock_title": "🚀 {gpu_name} NA ZALOGI!",
|
||||
"out_of_stock_title": "❌ {gpu_name} je razprodan",
|
||||
"sku_change_title": "🔄 Sprememba SKU zaznana za {gpu_name}",
|
||||
"buy_now": "**:point_right: [Kupi zdaj]({product_link})**",
|
||||
"price": "Cena",
|
||||
"time": "Čas",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Stari SKU**: `{old_sku}`\n**Novi SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Možen prihajajoči drop!"
|
||||
},
|
||||
"sv": {
|
||||
"in_stock_title": "🚀 {gpu_name} I LAGER!",
|
||||
"out_of_stock_title": "❌ {gpu_name} är slut",
|
||||
"sku_change_title": "🔄 SKU-ändring upptäckt för {gpu_name}",
|
||||
"buy_now": "**:point_right: [Köp nu]({product_link})**",
|
||||
"price": "Pris",
|
||||
"time": "Tid",
|
||||
"footer": "NviBot • {DISCORD_SERVER_NAME}",
|
||||
"sku_description": "**Gammalt SKU**: `{old_sku}`\n**Nytt SKU**: `{new_sku}`",
|
||||
"imminent_drop": "{DISCORD_ROLE} ⚠️ Möjlig förestående drop!"
|
||||
}
|
||||
}
|
27
app/main.py
Normal file
27
app/main.py
Normal file
@ -0,0 +1,27 @@
|
||||
import time
|
||||
import logging
|
||||
import signal
|
||||
import sys
|
||||
from gpu_checker import check_rtx_50_founders
|
||||
from env_config import REFRESH_TIME
|
||||
|
||||
# Signal handler function
|
||||
def handle_exit(signum, frame):
|
||||
logging.info(f"🛑 Received signal {signum}. Exiting gracefully...")
|
||||
sys.exit(0)
|
||||
|
||||
# Register signal handlers
|
||||
signal.signal(signal.SIGINT, handle_exit) # Ctrl+C
|
||||
signal.signal(signal.SIGTERM, handle_exit) # docker stop / kill -15
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
while True:
|
||||
start = time.time()
|
||||
check_rtx_50_founders()
|
||||
elapsed = time.time() - start
|
||||
time.sleep(max(0, REFRESH_TIME - elapsed))
|
||||
except KeyboardInterrupt:
|
||||
logging.info("🛑 Script interrupted by user (KeyboardInterrupt). Exiting gracefully.")
|
||||
sys.exit(0)
|
||||
|
112
app/notifier.py
Normal file
112
app/notifier.py
Normal file
@ -0,0 +1,112 @@
|
||||
import time
|
||||
import logging
|
||||
import requests
|
||||
from env_config import (
|
||||
DISCORD_WEBHOOK_URL, DISCORD_SERVER_NAME, DISCORD_ROLE_MAP, TEST_MODE,
|
||||
in_stock_title, out_of_stock_title, sku_change_title,
|
||||
buy_now, price_label, time_label, footer, sku_description, imminent_drop
|
||||
)
|
||||
|
||||
AVATAR = "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/ds_wh_pp.jpg"
|
||||
THUMBNAIL = "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/RTX5000.jpg"
|
||||
|
||||
# In stock
|
||||
def send_discord_notification(gpu_name, product_link, products_price):
|
||||
timestamp = int(time.time())
|
||||
if TEST_MODE:
|
||||
logging.info(f"[TEST MODE] Notification: {gpu_name} available!")
|
||||
return
|
||||
|
||||
embed = {
|
||||
"title": in_stock_title.format(gpu_name=gpu_name),
|
||||
"color": 3066993,
|
||||
"thumbnail": {"url": THUMBNAIL},
|
||||
"author": {"name": "Nvidia Founder Editions"},
|
||||
"fields": [
|
||||
{"name": price_label, "value": f"`{products_price}€`", "inline": True},
|
||||
{"name": time_label, "value": f"<t:{timestamp}:d> <t:{timestamp}:T>", "inline": True}
|
||||
],
|
||||
"description": buy_now.format(product_link=product_link),
|
||||
"footer": {"text": footer.format(DISCORD_SERVER_NAME=DISCORD_SERVER_NAME), "icon_url": AVATAR}
|
||||
}
|
||||
|
||||
payload = {
|
||||
"content": DISCORD_ROLE_MAP.get(gpu_name, "@everyone"),
|
||||
"username": "NviBot",
|
||||
"avatar_url": AVATAR,
|
||||
"embeds": [embed]
|
||||
}
|
||||
|
||||
try:
|
||||
response = requests.post(DISCORD_WEBHOOK_URL, json=payload)
|
||||
if response.status_code == 204:
|
||||
logging.info("✅ Notification sent to Discord.")
|
||||
else:
|
||||
logging.error(f"❌ Webhook error: {response.status_code} - {response.text}")
|
||||
except Exception as e:
|
||||
logging.error(f"🚨 Error sending webhook: {e}")
|
||||
|
||||
# Out of stock
|
||||
def send_out_of_stock_notification(gpu_name, product_link, products_price):
|
||||
timestamp = int(time.time())
|
||||
if TEST_MODE:
|
||||
logging.info(f"[TEST MODE] Out of stock: {gpu_name}")
|
||||
return
|
||||
|
||||
embed = {
|
||||
"title": out_of_stock_title.format(gpu_name=gpu_name),
|
||||
"color": 15158332,
|
||||
"thumbnail": {"url": THUMBNAIL},
|
||||
"url": product_link,
|
||||
"author": {"name": "Nvidia Founder Editions"},
|
||||
"footer": {"text": footer.format(DISCORD_SERVER_NAME=DISCORD_SERVER_NAME), "icon_url": AVATAR},
|
||||
"fields": [{"name": time_label, "value": f"<t:{timestamp}:d> <t:{timestamp}:T>", "inline": True}]
|
||||
}
|
||||
|
||||
payload = {
|
||||
"username": "NviBot",
|
||||
"avatar_url": AVATAR,
|
||||
"embeds": [embed]
|
||||
}
|
||||
|
||||
try:
|
||||
response = requests.post(DISCORD_WEBHOOK_URL, json=payload)
|
||||
if response.status_code == 204:
|
||||
logging.info("✅ Out-of-stock notification sent.")
|
||||
else:
|
||||
logging.error(f"❌ Webhook error: {response.status_code} - {response.text}")
|
||||
except Exception as e:
|
||||
logging.error(f"🚨 Error sending webhook: {e}")
|
||||
|
||||
# SKU change
|
||||
def send_sku_change_notification(gpu_name, old_sku, new_sku, product_link):
|
||||
timestamp = int(time.time())
|
||||
if TEST_MODE:
|
||||
logging.info(f"[TEST MODE] SKU change: {old_sku} → {new_sku}")
|
||||
return
|
||||
|
||||
embed = {
|
||||
"title": sku_change_title.format(gpu_name=gpu_name),
|
||||
"url": product_link,
|
||||
"description": sku_description.format(old_sku=old_sku, new_sku=new_sku),
|
||||
"color": 16776960,
|
||||
"footer": {"text": footer.format(DISCORD_SERVER_NAME=DISCORD_SERVER_NAME), "icon_url": AVATAR},
|
||||
"fields": [{"name": time_label, "value": f"<t:{timestamp}:d> <t:{timestamp}:T>", "inline": True}]
|
||||
}
|
||||
|
||||
payload = {
|
||||
"content": imminent_drop.format(DISCORD_ROLE=DISCORD_ROLE_MAP.get(gpu_name, '@everyone')),
|
||||
"username": "NviBot",
|
||||
"avatar_url": AVATAR,
|
||||
"embeds": [embed]
|
||||
}
|
||||
|
||||
try:
|
||||
response = requests.post(DISCORD_WEBHOOK_URL, json=payload)
|
||||
if response.status_code == 204:
|
||||
logging.info("✅ SKU change notification sent.")
|
||||
else:
|
||||
logging.error(f"❌ Webhook error: {response.status_code} - {response.text}")
|
||||
except Exception as e:
|
||||
logging.error(f"🚨 Error sending webhook: {e}")
|
||||
|
11
docker/.env
11
docker/.env
@ -1,6 +1,5 @@
|
||||
DS_HOOK= # Your Discord webhook URL
|
||||
FREQ= # Refresh frequency in seconds, default is 30
|
||||
API_URL_SKU= # API listing the product, default is https://api.nvidia.partners/edge/product/search?page=1&limit=100&locale=fr-fr&Manufacturer=Nvidia
|
||||
API_URL_STOCK= # API used to check stock without specifying the SKU, default is https://api.store.nvidia.com/partner/v1/feinventory?locale=fr-fr&skus=
|
||||
PRODUCT_URL= # Purchase URL of the GPU
|
||||
PRODUCT_NAME= # The exact name of the GPU you're looking for, e.g., "RTX 5080"
|
||||
PRODUCT_NAMES= # Exact GPU name (e.g. "RTX 5080, RTX 5090")
|
||||
DISCORD_WEBHOOK_URL= # Your Discord webhook URL
|
||||
API_URL_SKU= # API listing the product for your country
|
||||
API_URL_STOCK= # API providing stock data for your country
|
||||
PRODUCT_URL= # GPU purchase URL for your country
|
@ -6,11 +6,10 @@ services:
|
||||
env_file:
|
||||
- .env
|
||||
environment:
|
||||
- DISCORD_WEBHOOK_URL=${DS_HOOK}
|
||||
- REFRESH_TIME=${FREQ}
|
||||
- API_URL_SKU=${API_URL_SKU}
|
||||
- API_URL_STOCK=${API_URL_STOCK}
|
||||
- PRODUCT_URL=${PRODUCT_URL}
|
||||
- PRODUCT_NAME=${PRODUCT_NAME}
|
||||
- PYTHONUNBUFFERED=1 # Allow to display log in real-time
|
||||
command: python nvidia-stock-bot.py # Run the script
|
||||
# Minimal environment variables
|
||||
- PRODUCT_NAMES=${PRODUCT_NAMES} # Exact GPU name (e.g. "RTX 5080, RTX 5090")
|
||||
- DISCORD_WEBHOOK_URL=${DISCORD_WEBHOOK_URL} # Your Discord webhook URL
|
||||
- API_URL_SKU=${API_URL_SKU} # API listing the product for your country
|
||||
- API_URL_STOCK=${API_URL_STOCK} # API providing stock data for your country
|
||||
- PRODUCT_URL=${PRODUCT_URL} # GPU purchase URL for your country
|
||||
- PYTHONUNBUFFERED=1 # Enables real-time log output
|
@ -1,361 +0,0 @@
|
||||
import requests
|
||||
import logging
|
||||
import time
|
||||
import os
|
||||
import re
|
||||
from requests.adapters import HTTPAdapter, Retry
|
||||
|
||||
# Logger configuration
|
||||
logging.basicConfig(
|
||||
level=logging.INFO,
|
||||
format="%(asctime)s [%(levelname)s] %(message)s",
|
||||
)
|
||||
logging.info("Script started")
|
||||
|
||||
# Retrieve environment variables
|
||||
try:
|
||||
DISCORD_WEBHOOK_URL = os.environ.get('DISCORD_WEBHOOK_URL')
|
||||
API_URL_SKU = os.environ.get('API_URL_SKU', 'https://api.nvidia.partners/edge/product/search?page=1&limit=100&locale=fr-fr&Manufacturer=Nvidia')
|
||||
API_URL_STOCK = os.environ.get('API_URL_STOCK', 'https://api.store.nvidia.com/partner/v1/feinventory?locale=fr-fr&skus=')
|
||||
REFRESH_TIME = int(os.environ.get('REFRESH_TIME'))
|
||||
TEST_MODE = os.environ.get('TEST_MODE', 'False').lower() == 'true'
|
||||
PRODUCT_URL = os.environ.get('PRODUCT_URL', 'https://marketplace.nvidia.com/fr-fr/consumer/graphics-cards/?locale=fr-fr&page=1&limit=12&manufacturer=NVIDIA')
|
||||
DISCORD_ROLES = os.environ.get('DISCORD_ROLES')
|
||||
PRODUCT_NAMES = os.environ.get('PRODUCT_NAMES')
|
||||
|
||||
if not PRODUCT_NAMES:
|
||||
logging.error("❌ PRODUCT_NAMES is required but not defined.")
|
||||
exit(1)
|
||||
|
||||
PRODUCT_NAMES = [name.strip() for name in PRODUCT_NAMES.split(',')]
|
||||
DISCORD_ROLE_MAP = {}
|
||||
|
||||
if not DISCORD_ROLES or not DISCORD_ROLES.strip():
|
||||
logging.warning("⚠️ DISCORD_ROLES not defined or empty. Defaulting all roles to @everyone.")
|
||||
for name in PRODUCT_NAMES:
|
||||
DISCORD_ROLE_MAP[name] = '@everyone'
|
||||
else:
|
||||
roles = [r.strip() if r.strip() else '@everyone' for r in DISCORD_ROLES.split(',')]
|
||||
|
||||
if len(roles) != len(PRODUCT_NAMES):
|
||||
logging.error("❌ The number of DISCORD_ROLES must match PRODUCT_NAMES.")
|
||||
exit(1)
|
||||
|
||||
for name, role in zip(PRODUCT_NAMES, roles):
|
||||
if role != '@everyone' and not re.match(r'^<@&\d{17,20}>$', role):
|
||||
logging.error(f"❌ Invalid DISCORD_ROLE format for {name}: {role}")
|
||||
exit(1)
|
||||
DISCORD_ROLE_MAP[name] = role
|
||||
|
||||
|
||||
if not DISCORD_WEBHOOK_URL:
|
||||
logging.error("❌ DISCORD_WEBHOOK_URL is required but not defined.")
|
||||
exit(1)
|
||||
|
||||
# Regex to extract ID and token
|
||||
match = re.search(r'/(\d+)/(.*)', DISCORD_WEBHOOK_URL)
|
||||
if match:
|
||||
webhook_id = match.group(1)
|
||||
webhook_token = match.group(2)
|
||||
|
||||
# Mask last characters of the ID
|
||||
masked_webhook_id = webhook_id[:len(webhook_id) - 10] + '*' * 10
|
||||
|
||||
# Mask last characters of the token
|
||||
masked_webhook_token = webhook_token[:len(webhook_token) - 120] + '*' * 10
|
||||
|
||||
# Rebuild masked URL
|
||||
wh_masked_url = f"https://discord.com/api/webhooks/{masked_webhook_id}/{masked_webhook_token}"
|
||||
|
||||
# Error logging
|
||||
except KeyError as e:
|
||||
logging.error(f"Missing environment variable: {e}")
|
||||
exit(1)
|
||||
except ValueError:
|
||||
logging.error("REFRESH_TIME must be a valid integer.")
|
||||
exit(1)
|
||||
|
||||
# Display URLs and configurations
|
||||
logging.info(f"GPU: {PRODUCT_NAMES}")
|
||||
logging.info(f"Discord Webhook URL: {wh_masked_url}")
|
||||
logging.info(f"Discord Role Mention: {DISCORD_ROLES}")
|
||||
logging.info(f"API URL SKU: {API_URL_SKU}")
|
||||
logging.info(f"API URL Stock: {API_URL_STOCK}")
|
||||
logging.info(f"Product URL: {PRODUCT_URL}")
|
||||
logging.info(f"Refresh time: {REFRESH_TIME} seconds")
|
||||
logging.info(f"Test Mode: {TEST_MODE}")
|
||||
|
||||
# HTTP headers
|
||||
HEADERS = {
|
||||
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) "
|
||||
"AppleWebKit/537.36 (KHTML, like Gecko) "
|
||||
"Chrome/131.0.0.0 Safari/537.36",
|
||||
"Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,"
|
||||
"image/avif,image/webp,image/apng,*/*;q=0.8",
|
||||
"Accept-Encoding": "gzip, deflate, br, zstd",
|
||||
"Accept-Language": "en-US,en;q=0.5",
|
||||
"Cache-Control": "max-age=0",
|
||||
"Connection": "keep-alive",
|
||||
"Sec-Fetch-Dest": "document",
|
||||
"Sec-Fetch-Mode": "navigate",
|
||||
"Sec-Fetch-Site": "none",
|
||||
"Sec-Fetch-User": "?1",
|
||||
"Upgrade-Insecure-Requests": "1",
|
||||
"Sec-GPC": "1",
|
||||
}
|
||||
|
||||
# Session with retries
|
||||
session = requests.Session()
|
||||
retries = Retry(total=5, backoff_factor=1, status_forcelist=[500, 502, 503, 504])
|
||||
session.mount('https://', HTTPAdapter(max_retries=retries))
|
||||
|
||||
last_sku_dict = {}
|
||||
global_stock_status_dict = {}
|
||||
first_run_dict = {name: True for name in PRODUCT_NAMES}
|
||||
|
||||
# Discord notifications
|
||||
def send_discord_notification(gpu_name: str, product_link: str, products_price: str):
|
||||
|
||||
# Get current UNIX timestamp
|
||||
timestamp_unix = int(time.time())
|
||||
|
||||
if TEST_MODE:
|
||||
logging.info(f"[TEST MODE] Discord Notification: {gpu_name} available!")
|
||||
return
|
||||
|
||||
embed = {
|
||||
"title": f"🚀 {gpu_name} IN STOCK!",
|
||||
"color": 3066993,
|
||||
"thumbnail": {
|
||||
"url": "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/RTX5000.jpg"
|
||||
},
|
||||
"author": {
|
||||
"name": "Nvidia Founder Editions"
|
||||
},
|
||||
|
||||
"fields": [
|
||||
{
|
||||
"name": "Price",
|
||||
"value": f"`{products_price}€`",
|
||||
"inline": True
|
||||
},
|
||||
|
||||
{
|
||||
"name": "Time",
|
||||
"value": f"<t:{timestamp_unix}:d> <t:{timestamp_unix}:T>",
|
||||
"inline": True
|
||||
},
|
||||
],
|
||||
"description": f"**:point_right: [Buy now]({product_link})**",
|
||||
"footer": {
|
||||
"text": "NviBot • JV Hardware 2.0",
|
||||
"icon_url": "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/ds_wh_pp.jpg"
|
||||
}
|
||||
}
|
||||
|
||||
payload = {
|
||||
"content": f"{DISCORD_ROLE_MAP.get(gpu_name, '@everyone')}",
|
||||
"username": "NviBot",
|
||||
"avatar_url": "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/ds_wh_pp.jpg",
|
||||
"embeds": [embed]
|
||||
}
|
||||
|
||||
try:
|
||||
response = requests.post(DISCORD_WEBHOOK_URL, json=payload)
|
||||
if response.status_code == 204:
|
||||
logging.info("✅ Notification sent to Discord.")
|
||||
else:
|
||||
logging.error(f"❌ Webhook error: {response.status_code} - {response.text}")
|
||||
except Exception as e:
|
||||
logging.error(f"🚨 Error sending webhook: {e}")
|
||||
|
||||
def send_out_of_stock_notification(gpu_name: str, product_link: str, products_price: str):
|
||||
|
||||
# Get current UNIX timestamp
|
||||
timestamp_unix = int(time.time())
|
||||
|
||||
if TEST_MODE:
|
||||
logging.info(f"[TEST MODE] Discord Notification: {gpu_name} out of stock!")
|
||||
return
|
||||
|
||||
embed = {
|
||||
"title": f"❌ {gpu_name} is out of stock",
|
||||
"color": 15158332, # Red for out of stock
|
||||
"thumbnail": {
|
||||
"url": "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/RTX5000.jpg"
|
||||
},
|
||||
"url": f"{product_link}",
|
||||
"author": {
|
||||
"name": "Nvidia Founder Editions"
|
||||
},
|
||||
|
||||
"footer": {
|
||||
"text": "NviBot • JV Hardware 2.0",
|
||||
"icon_url": "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/ds_wh_pp.jpg"
|
||||
},
|
||||
|
||||
"fields": [
|
||||
{
|
||||
"name": "Time",
|
||||
"value": f"<t:{timestamp_unix}:d> <t:{timestamp_unix}:T>",
|
||||
"inline": True
|
||||
}
|
||||
]
|
||||
}
|
||||
payload = {"username": "NviBot", "avatar_url": "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/ds_wh_pp.jpg", "embeds": [embed]}
|
||||
try:
|
||||
response = requests.post(DISCORD_WEBHOOK_URL, json=payload)
|
||||
if response.status_code == 204:
|
||||
logging.info("✅ 'Out of stock' notification sent to Discord.")
|
||||
else:
|
||||
logging.error(f"❌ Webhook error: {response.status_code} - {response.text}")
|
||||
except Exception as e:
|
||||
logging.error(f"🚨 Error sending webhook: {e}")
|
||||
|
||||
def send_sku_change_notification(gpu_name: str, old_sku: str, new_sku: str, product_link: str):
|
||||
|
||||
# Get current UNIX timestamp
|
||||
timestamp_unix = int(time.time())
|
||||
|
||||
if TEST_MODE:
|
||||
logging.info(f"[TEST MODE] SKU change detected: {old_sku} → {new_sku}")
|
||||
return
|
||||
|
||||
embed = {
|
||||
"title": f"🔄 {gpu_name} SKU change detected",
|
||||
"url": f"{product_link}",
|
||||
"description": f"**Old SKU** : `{old_sku}`\n**New SKU** : `{new_sku}`",
|
||||
"color": 16776960, # Yellow
|
||||
|
||||
"footer": {
|
||||
"text": "NviBot • JV Hardware 2.0",
|
||||
"icon_url": "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/ds_wh_pp.jpg"
|
||||
},
|
||||
|
||||
"fields": [
|
||||
{
|
||||
"name": "Time",
|
||||
"value": f"<t:{timestamp_unix}:d> <t:{timestamp_unix}:T>",
|
||||
"inline": True
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
payload = {
|
||||
"content": f"{DISCORD_ROLE_MAP.get(gpu_name, '@everyone')} ⚠️ Possible imminent drop!",
|
||||
"username": "NviBot",
|
||||
"avatar_url": "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/ds_wh_pp.jpg",
|
||||
"embeds": [embed]
|
||||
}
|
||||
|
||||
try:
|
||||
response = requests.post(DISCORD_WEBHOOK_URL, json=payload)
|
||||
if response.status_code == 204:
|
||||
logging.info("✅ SKU change notification sent to Discord.")
|
||||
else:
|
||||
logging.error(f"❌ Webhook error: {response.status_code} - {response.text}")
|
||||
except Exception as e:
|
||||
logging.error(f"🚨 Error sending webhook: {e}")
|
||||
|
||||
# Stock search
|
||||
def check_rtx_50_founders():
|
||||
global last_sku_dict, global_stock_status_dict, first_run_dict
|
||||
|
||||
try:
|
||||
response = session.get(API_URL_SKU, headers=HEADERS, timeout=10)
|
||||
logging.info(f"SKU API response: {response.status_code}")
|
||||
response.raise_for_status()
|
||||
data = response.json()
|
||||
except requests.exceptions.RequestException as e:
|
||||
logging.error(f"SKU API error: {e}")
|
||||
return
|
||||
|
||||
# All available products
|
||||
all_products = data['searchedProducts']['productDetails']
|
||||
|
||||
for product_name in PRODUCT_NAMES:
|
||||
product_details = None
|
||||
for p in all_products:
|
||||
if p.get("gpu", "").strip() == product_name:
|
||||
product_details = p
|
||||
break
|
||||
|
||||
if not product_details:
|
||||
logging.warning(f"⚠️ No product with GPU '{product_name}' found.")
|
||||
continue
|
||||
|
||||
product_sku = product_details['productSKU']
|
||||
product_upc = product_details.get('productUPC', "")
|
||||
if not isinstance(product_upc, list):
|
||||
product_upc = [product_upc]
|
||||
|
||||
# Check SKU change
|
||||
old_sku = last_sku_dict.get(product_name)
|
||||
if old_sku and old_sku != product_sku and not first_run_dict[product_name]:
|
||||
logging.warning(f"⚠️ SKU changed for {product_name}: {old_sku} → {product_sku}")
|
||||
send_sku_change_notification(product_name, old_sku, product_sku, PRODUCT_URL)
|
||||
|
||||
last_sku_dict[product_name] = product_sku
|
||||
first_run_dict[product_name] = False
|
||||
|
||||
# Stock check
|
||||
api_stock_url = API_URL_STOCK + product_sku
|
||||
logging.info(f"[{product_name}] Checking stock: {api_stock_url}")
|
||||
|
||||
try:
|
||||
response = session.get(api_stock_url, headers=HEADERS, timeout=10)
|
||||
logging.info(f"[{product_name}] Stock API response: {response.status_code}")
|
||||
response.raise_for_status()
|
||||
stock_data = response.json()
|
||||
except requests.exceptions.RequestException as e:
|
||||
logging.error(f"Stock API error: {e}")
|
||||
continue
|
||||
|
||||
products = stock_data.get("listMap", [])
|
||||
products_price = "Price not available"
|
||||
if isinstance(products, list) and len(products) > 0:
|
||||
for p in products:
|
||||
price = p.get("price", 'Price not available')
|
||||
if price != 'Price not available':
|
||||
products_price = price
|
||||
break
|
||||
else:
|
||||
logging.error(f"[{product_name}] Product list is empty or malformed.")
|
||||
|
||||
found_in_stock = set()
|
||||
for p in products:
|
||||
gpu_name = p.get("fe_sku", "").upper()
|
||||
is_active = p.get("is_active") == "true"
|
||||
if is_active and any(upc.upper() in gpu_name for upc in product_upc):
|
||||
found_in_stock.add(gpu_name)
|
||||
|
||||
for upc in product_upc:
|
||||
upc_upper = upc.upper()
|
||||
currently_in_stock = upc_upper in found_in_stock
|
||||
previously_in_stock = global_stock_status_dict.get((product_name, upc_upper), False)
|
||||
|
||||
if currently_in_stock and not previously_in_stock:
|
||||
send_discord_notification(product_name, PRODUCT_URL, products_price)
|
||||
global_stock_status_dict[(product_name, upc_upper)] = True
|
||||
logging.info(f"[{product_name}] {upc} is now in stock!")
|
||||
elif not currently_in_stock and previously_in_stock:
|
||||
send_out_of_stock_notification(product_name, PRODUCT_URL, products_price)
|
||||
global_stock_status_dict[(product_name, upc_upper)] = False
|
||||
logging.info(f"[{product_name}] {upc} is now out of stock.")
|
||||
elif currently_in_stock:
|
||||
logging.info(f"[{product_name}] {upc} still in stock.")
|
||||
else:
|
||||
logging.info(f"[{product_name}] {upc} still out of stock.")
|
||||
# Loop
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
while True:
|
||||
start = time.time()
|
||||
check_rtx_50_founders()
|
||||
elapsed = time.time() - start
|
||||
time.sleep(max(0, REFRESH_TIME - elapsed))
|
||||
|
||||
# Gracefully shut down
|
||||
except KeyboardInterrupt:
|
||||
logging.info("🛑 Script interrupted by user. Exiting gracefully.")
|
||||
exit(0)
|
Reference in New Issue
Block a user