auto sku

README.md

@@ -5,13 +5,11 @@ Par KevOut & Djeex
 
 
 Ce robot :
-- Appelle régulièrement l'api des stocks français de nvidia FE (par défaut toutes les 60s)
+- Appelle l'API de Nvidia listant le produit (par défaut toutes les 60s)
-- Vérifie si RTX 5090, RTX 5080, RTX 5070ti et RTX 5070 sont en stock
+- Récupère le SKU du produit concerné
+- Appelle le stock lié à ce SKU
 - Si du stock est trouvé, envoie une notification discord via le webhook paramétré
 
-<img src="https://git.djeex.fr/Djeex/nvidia-stock-bot/raw/branch/main/assets/img/nvbot.png" align="center">
-
-
 Trois modes d'installation :
 - [Avec le dépot Git et Docker](https://git.djeex.fr/Djeex/nvidia-stock-bot/#installation-avec-le-d%C3%A9pot)
 - [Sans le dépot Git et avec notre image docker fournie](https://git.djeex.fr/Djeex/nvidia-stock-bot/#installation-sans-le-d%C3%A9pot-avec-docker-compose)
@@ -73,8 +71,8 @@ services:
     environment:
       - DISCORD_WEBHOOK_URL= # URL de votre webhook Discord
       - REFRESH_TIME= # Durée de rafraichissement du script en secondes
-      - GPU_TARGETS= #SKU
+      - API_URL_SKU= # API listant le produit par exemple : https://api.nvidia.partners/edge/product/search?page=1&limit=100&locale=fr-fr&Manufacturer=Nvidia&gpu=RTX%205090
-      - API_URL= #URL de l'API
+      - API_URL_STOCK= # API appelant le stock sans préciser la valeur du sku, par exemple : https://api.store.nvidia.com/partner/v1/feinventory?locale=fr-fr&skus=
       - PYTHONUNBUFFERED=1 # Permet d'afficher les logs en temps réel
     command: python nvidia-stock-bot.py # Lance le script Python au démarrage du conteneur
 ```

docker/compose.yaml

@@ -9,7 +9,7 @@ services:
     environment:
       - DISCORD_WEBHOOK_URL=${DS_HOOK}
       - REFRESH_TIME=${FREQ}
-      - GPU_TARGETS=${GPU} #SKU
+      - API_URL_SKU=${API_URL_SKU}
-      - API_URL=${URL} #URL de l'API
+      - API_URL_STOCK=${API_URL_STOCK}
       - PYTHONUNBUFFERED=1 # Permet d'afficher les logs en temps réel
     command: python nvidia-stock-bot.py # Lance le script Python

nvidia-stock-bot.py

@@ -14,9 +14,8 @@ logging.info("Démarrage du script")
 # Récupération des variables d'environnement
 try:
     DISCORD_WEBHOOK_URL = os.environ['DISCORD_WEBHOOK_URL']
-    API_URL = os.environ['API_URL']
+    API_URL_SKU = os.environ['API_URL_SKU']
-    GPU_TARGETS = os.environ['GPU_TARGETS'].split(",")  # Séparer en liste
+    API_URL_STOCK = os.environ['API_URL_STOCK']
-    GPU_TARGETS = [gpu.strip() for gpu in GPU_TARGETS]  # Nettoyer les espaces
     REFRESH_TIME = int(os.environ['REFRESH_TIME'])  # Convertir en entier
 except KeyError as e:
     logging.error(f"Variable d'environnement manquante : {e}")
@@ -27,15 +26,11 @@ except ValueError:
 
 # Afficher les valeurs des variables d'environnement
 print(f"url du webhook Discord: {DISCORD_WEBHOOK_URL}")
-print(f"url de l'API: {API_URL}")
+print(f"url de l'API qui liste les SKU: {API_URL_SKU}")
-print(f"GPU recherché: {GPU_TARGETS}")
+print(f"url de l'API de stock: {API_URL_STOCK}")
+#print(f"GPU recherché: {GPU_TARGETS}")
 print(f"Temps d'actualisation (en secondes) : {REFRESH_TIME}")
 
-# L’URL de l’API (exemple)
-#API_URL = "https://api.store.nvidia.com/partner/v1/feinventory?locale=fr-fr&skus=5090LDLCFE"
-
-# GPUs à surveiller
-#GPU_TARGETS = ["5090LDLCFE_FR"]
 
 # Entêtes HTTP pour la requête
 HEADERS = {
@@ -56,9 +51,6 @@ HEADERS = {
     "Sec-GPC": "1",
 }
 
-# Dictionnaire stockant l'état de stock
-stock_status = {gpu.upper(): False for gpu in GPU_TARGETS}
-
 session = requests.Session()
 
 def send_discord_notification(gpu_name: str, product_link: str):
@@ -89,7 +81,31 @@ def send_discord_notification(gpu_name: str, product_link: str):
         logging.error(f"🚨 Erreur lors de l'envoi du webhook : {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."""
+    """Liste les SKU à rechercher"""
+    try:
+        response = session.get(API_URL_SKU, 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.RequestException as e:
+        logging.error(f"Erreur lors de l'appel API : {e}")
+        return
+
+    product_details = data['searchedProducts']['productDetails'][0]
+    product_sku = product_details['productSKU']
+    product_upc = product_details.get('productUPC', "")
+
+    if not isinstance(product_upc, list):
+        product_upc = [product_upc]
+    
+    # Dictionnaire stockant l'état de stock
+    stock_status = {gpu.upper(): False for gpu in product_upc}
+    
+    """Cherche les stock du sku concerné"""
+    API_URL = API_URL_STOCK + product_sku
+    
+    print(f"url de l'API de stock: {API_URL}")
+    
     try:
         response = session.get(API_URL, headers=HEADERS, timeout=10)
         logging.info(f"Réponse de l'API : {response.status_code}")
@@ -107,10 +123,10 @@ def check_rtx_50_founders():
         is_active = p.get("is_active") == "true"
 
         if is_active:
-            if any(target.upper() in gpu_name for target in GPU_TARGETS):
+            if any(target.upper() in gpu_name for target in product_upc):
                 found_in_stock.add(gpu_name)
 
-    for gpu in GPU_TARGETS:
+    for gpu in product_upc:
         gpu_upper = gpu.upper()
         currently_in_stock = (gpu_upper in found_in_stock)
         previously_in_stock = stock_status[gpu_upper]