Compare commits
9 Commits
Author | SHA1 | Date | |
---|---|---|---|
1b15946a55 | |||
af86d4aa8f | |||
f2b5b184ea | |||
50a9c1f827 | |||
584c222076 | |||
70a7df1f97 | |||
70b723b781 | |||
62ef15c928 | |||
fe3e8f9609 |
11
Dockerfile
11
Dockerfile
@ -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"]
|
10
README.md
10
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)
|
||||
|
||||
@ -54,13 +56,11 @@ 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
|
||||
- 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:**
|
||||
|
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.")
|
||||
|
@ -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
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}")
|
||||
|
@ -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)
|
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
|
Reference in New Issue
Block a user