first commit

Dockerfile

@@ -0,0 +1,22 @@
+# Utiliser une image de base légère de Python (ici Python 3.9)
+FROM python:3.9-slim
+
+# Définir le répertoire de travail à l'intérieur du conteneur
+WORKDIR /app
+
+# Copier le script Python dans le répertoire de travail
+COPY nvidia-stock-bot.py /app/
+
+# Copier un éventuel fichier requirements.txt pour installer des dépendances
+# Si des dépendances supplémentaires sont nécessaires, ajoutez-les dans requirements.txt
+COPY requirements.txt /app/
+
+# Installer les dépendances Python à partir de requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Définir les variables d'environnement par défaut (modifiable lors du lancement du conteneur)
+ENV DISCORD_WEBHOOK_URL="https://example.com/webhook" \
+    REFRESH_TIME="60"
+
+# Exposer un point de commande pour exécuter le script
+CMD ["python", "bot_nvidia.py"]

README.md

@@ -0,0 +1,41 @@
+# Nvidia Stock Bot
+Par KevOut & Djeex
+
+Ce robot :
+- Appelle régulièrement l'api des stocks français de nvidia (par défaut toutes les 60s)
+- Vérifie si RTX 5090, RTX 5080; RTX 5070ti et RTX 5070 sont en stock
+- Si du stock est trouvé  envoie une notification discord via le webhook paramétré
+
+## Installation
+
+### Pré-requis
+- Git
+- Docker
+
+### Cloner et paramétrer
+Clonez le repo :
+```sh
+git clone https://git.djeex.fr/Djeex/nvidia-stock-bot.git
+```
+
+Rendez vous dans le dossier `nvidia-stock-bot` et compilez l'image docker :
+```sh
+docker build -t nvidia-stock-bot .
+```
+
+Rendez-vous dans le dossier `nvidia-stock-bot/docker` et éditez le fichier `.env` avec :
+- l'url de votre webhook discord
+- la fréquence de consultation des stock (par défaut 60s, attention à ne pas trop descendre sous peine de blocage de votre adresse IP par nVidia)
+
+### Lancer l'image
+
+Rendez-vous dans le dossier `nvidia-stock-bot/docker` et lancez le conteneur :
+```sh
+docker compose up -d
+```
+
+### Voir les logs pour vérifier le bon fonctionnement
+
+```sh
+docker logs -f nvidia-stock-bot
+```

docker/.env

@@ -0,0 +1,2 @@
+DS_HOOK="votre url du webhook discord"
+FREQ=60 #frequence de rafraichissement en secondes

docker/compose.yaml

@@ -0,0 +1,13 @@
+version: "3.8"
+services:
+  nvidia-stock-bot:
+    image: docker.io/library/nvidia-stock-bot
+    container_name: nvidia-stock-bot
+    restart: always # Le conteneur redémarrera automatiquement en cas d'échec
+    env_file:
+      - .env
+    environment:
+      - DISCORD_WEBHOOK_URL=${DS_HOOK}
+      - REFRESH_TIME=${FREQ}
+      - PYTHONUNBUFFERED=1 # Permet d'afficher les logs en temps réel
+    command: python bot_nvidia.py # Lance le script Python

nvidia-stock-bot.py

@@ -0,0 +1,140 @@
+import requests
+import logging
+import time
+import os
+
+# Configuration du logger
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s [%(levelname)s] %(message)s",
+)
+
+logging.info("Démarrage du script")
+
+# Récupération des variables d'environnement
+try:
+    DISCORD_WEBHOOK_URL = os.environ['DISCORD_WEBHOOK_URL']
+    REFRESH_TIME = int(os.environ['REFRESH_TIME'])  # Convertir en entier
+except KeyError as e:
+    logging.error(f"Variable d'environnement manquante : {e}")
+    exit(1)  # Quitter le script proprement en cas d'erreur
+except ValueError:
+    logging.error("REFRESH_TIME doit être un entier valide.")
+    exit(1)
+
+# Afficher les valeurs des variables d'environnement
+print(f"url du webhook Discord: {DISCORD_WEBHOOK_URL}")
+print(f"Temps d'actualisation (en secondes) : {REFRESH_TIME}")
+
+# L’URL de l’API (exemple)
+API_URL = "https://api.nvidia.partners/edge/product/search?page=1&limit=100&locale=fr-fr&Manufacturer=Nvidia"
+
+# GPUs à surveiller
+GPU_TARGETS = ["RTX 5070 Ti", "RTX 5070", "RTX 5080", "RTX 5090"]
+
+# Entêtes HTTP pour la requête
+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",
+}
+
+# Dictionnaire stockant l'état de stock pour chaque GPU.
+stock_status = {gpu.upper(): False for gpu in GPU_TARGETS}
+
+session = requests.Session()
+
+
+def send_discord_notification(message: str):
+    """Envoie un message sur un salon Discord via un webhook."""
+    payload = {
+        "content": message
+    }
+    try:
+        response = requests.post(DISCORD_WEBHOOK_URL, json=payload)
+        if response.status_code == 204:
+            logging.info("Notification envoyée sur Discord.")
+        else:
+            logging.error(f"Échec de l'envoi du webhook : {response.status_code} - {response.text}")
+    except Exception as e:
+        logging.error(f"Erreur lors de l'appel au webhook Discord : {e}")
+
+
+def check_rtx_50_founders():
+    """Vérifie l'état de stock des GPU Founders Edition et notifie Discord si un GPU repasse en stock."""
+    try:
+        session.get("https://api.nvidia.partners/edge/product/search?page=1&limit=1",
+                    headers=HEADERS,
+                    timeout=10)
+
+        response = session.get(API_URL, headers=HEADERS, timeout=10)
+        logging.info(f"Réponse de l'API : {response.status_code}")
+        response.raise_for_status()
+        data = response.json()
+
+    except requests.exceptions.ReadTimeout:
+        logging.info("Time-out lors de l'appel API.")
+        return
+    except Exception as e:
+        logging.info(f"Erreur lors de l'appel API : {e}")
+        return
+
+    products = data.get("searchedProducts", {}).get("productDetails", [])
+    found_in_stock = set()
+
+    for p in products:
+        gpu_name = p.get("gpu", "").upper()
+        is_founder = p.get("isFounderEdition") is True
+        is_nvidia = (p.get("manufacturer") == "NVIDIA")
+        is_buy_now = (p.get("prdStatus") == "buy_now" or p.get("productAvailable") is True)
+
+        if is_founder and is_nvidia and is_buy_now:
+            if any(target.upper() in gpu_name for target in GPU_TARGETS):
+                found_in_stock.add(gpu_name)
+
+    for gpu in GPU_TARGETS:
+        gpu_upper = gpu.upper()
+        currently_in_stock = (gpu_upper in found_in_stock)
+        previously_in_stock = stock_status[gpu_upper]
+
+        if currently_in_stock and not previously_in_stock:
+            for product in products:
+                product_name = product.get("gpu", "").upper()
+                if product_name == gpu_upper:
+                    real_gpu_name = product.get("gpu", "Inconnu")
+                    product_link = product.get("internalLink", "(pas de lien trouvé)")
+                    msg = (
+                        f"@everyone **Disponibilité repérée !**\n"
+                        f"Carte RTX Founders Edition détectée : `{real_gpu_name}`\n"
+                        f"Lien : {product_link}"
+                    )
+                    send_discord_notification(msg)
+
+            stock_status[gpu_upper] = True
+            print(f"{gpu} est maintenant en stock!")
+
+        elif (not currently_in_stock) and previously_in_stock:
+            logging.info(f"{gpu} n'est plus en stock.")
+            stock_status[gpu_upper] = False
+            print(f"{gpu} est hors stock !")
+
+        elif not currently_in_stock:
+            print(f"{gpu} est actuellement hors stock.")
+
+
+if __name__ == "__main__":
+    while True:
+        check_rtx_50_founders()
+        time.sleep(REFRESH_TIME)

requirements.txt

@@ -0,0 +1 @@
+requests