11 Commits
v3.2 ... v3.3.1

10 changed files with 441 additions and 451 deletions

View File

@ -4,15 +4,8 @@ RUN apk add --no-cache ca-certificates
WORKDIR /app
COPY app/nvidia-stock-bot.py /app/
COPY app/requirements.txt /app/
COPY app/localization.json /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"]

View File

@ -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)
@ -54,13 +56,12 @@ services:
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
- DISCORD_NOTIFICATION_CURRENCY= # Set your country currency
- 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:**
@ -69,6 +70,7 @@ services:
|---------------------|-------------------------------------------------|------------------------------------------------------------------|-------------------------------------------------------------------------------------------------------------|
| `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_NOTIFICATION_CURRENCY` | Your country currency | A text, e.g. `$`, `€`, `£`... | `€` |
| `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 |
@ -151,6 +153,7 @@ git clone https://git.djeex.fr/Djeex/nvidia-stock-bot.git
```sh
export DISCORD_WEBHOOK_URL="https://your_discord_url"
export DISCORD_NOTIFICATION_CURRENCY=
export PRODUCT_NAMES=RTX 5080, RTX 5090
export DISCORD_ROLES=<@&12345>, <@&6789>
export REFRESH_TIME="60"

144
app/env_config.py Normal file
View File

@ -0,0 +1,144 @@
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')
DISCORD_NOTIFICATION_CURRENCY = os.environ.get('DISCORD_NOTIFICATION_CURRENCY') or ''
API_URL_SKU = os.environ.get('API_URL_SKU') or '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') or 'https://api.store.nvidia.com/partner/v1/feinventory?locale=fr-fr&skus='
REFRESH_TIME = int(os.environ.get('REFRESH_TIME') or 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)
if not DISCORD_WEBHOOK_URL:
logging.error("❌ DISCORD_WEBHOOK_URL 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
# 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"Currency used for notifications: {DISCORD_NOTIFICATION_CURRENCY}")
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
View 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.")

View File

@ -8,7 +8,7 @@
"time": "Time",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Old SKU** : `{old_sku}`\n**New SKU** : `{new_sku}`",
"imminent_drop": "⚠️ Possible imminent drop!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Possible imminent drop!"
},
"bg": {
"in_stock_title": "🚀 {gpu_name} ИМА НАЛИЧНОСТ!",
@ -19,7 +19,7 @@
"time": "Час",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Стар SKU**: `{old_sku}`\n**Нов SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Възможен наближаващ спад!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Възможен наближаващ спад!"
},
"cs": {
"in_stock_title": "🚀 {gpu_name} SKLADEM!",
@ -30,7 +30,7 @@
"time": "Čas",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Starý SKU**: `{old_sku}`\n**Nový SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Možný blížící se pokles!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Možný blížící se pokles!"
},
"da": {
"in_stock_title": "🚀 {gpu_name} PÅ LAGER!",
@ -41,7 +41,7 @@
"time": "Tid",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Gammelt SKU**: `{old_sku}`\n**Nyt SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Muligt snarligt drop!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Muligt snarligt drop!"
},
"de": {
"in_stock_title": "🚀 {gpu_name} AUF LAGER!",
@ -52,7 +52,7 @@
"time": "Zeit",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Alte SKU**: `{old_sku}`\n**Neue SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Möglicher bevorstehender Drop!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Möglicher bevorstehender Drop!"
},
"el": {
"in_stock_title": "🚀 {gpu_name} ΔΙΑΘΕΣΙΜΟ!",
@ -63,7 +63,7 @@
"time": "Ώρα",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Παλαιό SKU**: `{old_sku}`\n**Νέο SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Πιθανή επικείμενη πτώση!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Πιθανή επικείμενη πτώση!"
},
"es": {
"in_stock_title": "🚀 {gpu_name} ¡EN STOCK!",
@ -74,7 +74,7 @@
"time": "Hora",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**SKU antiguo**: `{old_sku}`\n**SKU nuevo**: `{new_sku}`",
"imminent_drop": "⚠️ ¡Posible drop inminente!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ ¡Posible drop inminente!"
},
"et": {
"in_stock_title": "🚀 {gpu_name} LAOS!",
@ -85,7 +85,7 @@
"time": "Aeg",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Vana SKU**: `{old_sku}`\n**Uus SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Võimalik varsti tulemas drop!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Võimalik varsti tulemas drop!"
},
"fi": {
"in_stock_title": "🚀 {gpu_name} VARASTOSSA!",
@ -96,7 +96,7 @@
"time": "Aika",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Vanha SKU**: `{old_sku}`\n**Uusi SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Mahdollinen lähestyvä drop!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Mahdollinen lähestyvä drop!"
},
"fr": {
"in_stock_title": "🚀 {gpu_name} EN STOCK !",
@ -107,7 +107,7 @@
"time": "Heure",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Ancien SKU** : `{old_sku}`\n**Nouveau SKU** : `{new_sku}`",
"imminent_drop": "⚠️ Éventuel drop imminent!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Éventuel drop imminent!"
},
"hr": {
"in_stock_title": "🚀 {gpu_name} NA ZALIHI!",
@ -118,7 +118,7 @@
"time": "Vrijeme",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Stari SKU**: `{old_sku}`\n**Novi SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Mogući nadolazeći drop!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Mogući nadolazeći drop!"
},
"hu": {
"in_stock_title": "🚀 {gpu_name} RAKTÁRON!",
@ -129,7 +129,7 @@
"time": "Idő",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Régi SKU**: `{old_sku}`\n**Új SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Lehetséges közelgő drop!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Lehetséges közelgő drop!"
},
"ga": {
"in_stock_title": "🚀 {gpu_name} I STOC!",
@ -140,7 +140,7 @@
"time": "Am",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Sean SKU**: `{old_sku}`\n**SKU Nua**: `{new_sku}`",
"imminent_drop": "⚠️ I bhfad is gaire drop féideartha!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ I bhfad is gaire drop féideartha!"
},
"it": {
"in_stock_title": "🚀 {gpu_name} DISPONIBILE!",
@ -151,7 +151,7 @@
"time": "Ora",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Vecchio SKU**: `{old_sku}`\n**Nuovo SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Possibile drop imminente!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Possibile drop imminente!"
},
"lv": {
"in_stock_title": "🚀 {gpu_name} KRĀJUMĀ!",
@ -162,7 +162,7 @@
"time": "Laiks",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Vecais SKU**: `{old_sku}`\n**Jaunais SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Iespējams gaidāms drop!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Iespējams gaidāms drop!"
},
"lt": {
"in_stock_title": "🚀 {gpu_name} SANDĖLYJE!",
@ -173,7 +173,7 @@
"time": "Laikas",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Senas SKU**: `{old_sku}`\n**Naujas SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Rugsėjo drop'as gali artėti!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Rugsėjo drop'as gali artėti!"
},
"mt": {
"in_stock_title": "🚀 {gpu_name} F'ĠESTA!",
@ -184,7 +184,7 @@
"time": "Ħin",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**SKU Antik**: `{old_sku}`\n**SKU Ġdid**: `{new_sku}`",
"imminent_drop": "⚠️ Possibbli drop imminenti!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Possibbli drop imminenti!"
},
"nl": {
"in_stock_title": "🚀 {gpu_name} OP VOORRAAD!",
@ -195,7 +195,7 @@
"time": "Tijd",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Oude SKU**: `{old_sku}`\n**Nieuwe SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Mogelijke aanstaande drop!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Mogelijke aanstaande drop!"
},
"pl": {
"in_stock_title": "🚀 {gpu_name} DOSTĘPNE!",
@ -206,7 +206,7 @@
"time": "Czas",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Stare SKU**: `{old_sku}`\n**Nowe SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Możliwy nadchodzący drop!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Możliwy nadchodzący drop!"
},
"pt": {
"in_stock_title": "🚀 {gpu_name} EM STOCK!",
@ -217,7 +217,7 @@
"time": "Hora",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**SKU antigo**: `{old_sku}`\n**SKU novo**: `{new_sku}`",
"imminent_drop": "⚠️ Possível drop iminente!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Possível drop iminente!"
},
"ro": {
"in_stock_title": "🚀 {gpu_name} ÎN STOC!",
@ -228,7 +228,7 @@
"time": "Timp",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**SKU vechi**: `{old_sku}`\n**SKU nou**: `{new_sku}`",
"imminent_drop": "⚠️ Posibil drop iminent!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Posibil drop iminent!"
},
"sk": {
"in_stock_title": "🚀 {gpu_name} NA SKLADE!",
@ -239,7 +239,7 @@
"time": "Čas",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Starý SKU**: `{old_sku}`\n**Nový SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Možný nadchádzajúci drop!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Možný nadchádzajúci drop!"
},
"sl": {
"in_stock_title": "🚀 {gpu_name} NA ZALOGI!",
@ -250,7 +250,7 @@
"time": "Čas",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Stari SKU**: `{old_sku}`\n**Novi SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Možen prihajajoči drop!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Možen prihajajoči drop!"
},
"sv": {
"in_stock_title": "🚀 {gpu_name} I LAGER!",
@ -261,6 +261,6 @@
"time": "Tid",
"footer": "NviBot • {DISCORD_SERVER_NAME}",
"sku_description": "**Gammalt SKU**: `{old_sku}`\n**Nytt SKU**: `{new_sku}`",
"imminent_drop": "⚠️ Möjlig förestående drop!"
"imminent_drop": "{DISCORD_ROLE} ⚠️ Möjlig förestående drop!"
}
}

27
app/main.py Normal file
View 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)

113
app/notifier.py Normal file
View File

@ -0,0 +1,113 @@
import time
import logging
import requests
from env_config import (
DISCORD_WEBHOOK_URL, DISCORD_SERVER_NAME, DISCORD_ROLE_MAP, TEST_MODE, DISCORD_NOTIFICATION_CURRENCY,
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())
currency = DISCORD_NOTIFICATION_CURRENCY
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"`{currency}{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}")

View File

@ -1,393 +0,0 @@
import requests
import logging
import time
import os
import re
import json
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')
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'))
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)
# Localization
try:
with open("localization.json", "r", encoding="utf-8") as f:
localization = json.load(f)
except FileNotFoundError:
logging.error("❌ localization.json file not found.")
exit(1)
language = os.environ.get("DISCORD_NOTIFICATION_LANGUAGE", "en").lower()
loc = localization.get(language, localization["en"])
if language not in localization:
logging.warning(f"⚠️ Language '{language}' not found. Falling back to English.")
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"]
required_keys = ["in_stock_title", "out_of_stock_title", "sku_change_title", "buy_now", "price", "time", "footer", "sku_description", "imminent_drop"]
for key in required_keys:
if key not in loc:
logging.error(f"❌ Missing localization key: '{key}' in language '{language}'")
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": in_stock_title.format(gpu_name=gpu_name),
"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_label,
"value": f"`{products_price}€`",
"inline": True
},
{
"name": time_label,
"value": f"<t:{timestamp_unix}:d> <t:{timestamp_unix}:T>",
"inline": True
},
],
"description": buy_now.format(product_link=product_link),
"footer": {
"text": footer.format(DISCORD_SERVER_NAME=DISCORD_SERVER_NAME),
"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": out_of_stock_title.format(gpu_name=gpu_name),
"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": footer.format(DISCORD_SERVER_NAME=DISCORD_SERVER_NAME),
"icon_url": "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/ds_wh_pp.jpg"
},
"fields": [
{
"name": time_label,
"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": sku_change_title.format(gpu_name=gpu_name),
"url": f"{product_link}",
"description": sku_description.format(old_sku=old_sku, new_sku=new_sku),
"color": 16776960, # Yellow
"footer": {
"text": footer.format(DISCORD_SERVER_NAME=DISCORD_SERVER_NAME),
"icon_url": "https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/ds_wh_pp.jpg"
},
"fields": [
{
"name": time_label,
"value": f"<t:{timestamp_unix}:d> <t:{timestamp_unix}:T>",
"inline": True
}
]
}
payload = {
"content": 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)

View File

@ -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

View File

@ -6,11 +6,11 @@ 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= # Exact GPU name (e.g. "RTX 5080, RTX 5090")
- DISCORD_WEBHOOK_URL= # Your Discord webhook URL
- DISCORD_NOTIFICATION_CURRENCY= # Set your country currency
- 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