1st commit

nvidia-stock-bot.py

@@ -0,0 +1,122 @@
+from selenium import webdriver
+from selenium.webdriver.common.by import By
+from selenium.webdriver.chrome.options import Options
+import time
+import traceback
+from dotenv import load_dotenv
+import os
+import requests
+
+# Charger les variables d'environnement depuis le fichier .env
+load_dotenv()
+
+URL_NVIDIA = os.getenv(
+    "URL",
+    "https://marketplace.nvidia.com/fr-fr/consumer/graphics-cards/?locale=fr-fr&page=1&limit=12&gpu=RTX%205090&manufacturer=NVIDIA&manufacturer_filter=NVIDIA~1"
+)
+
+N = int(os.getenv("ITER_CYCLES", 1))
+WAIT_TIME = int(os.getenv("WAIT_TIME", 3))
+RUN_INTERVAL = int(os.getenv("RUN_INTERVAL", 30))
+PRINT_BODY = os.getenv("PRINT_BODY", "FALSE").lower() == "true"
+ERROR_ALERT = os.getenv("ERROR_ALERT", "FALSE").lower() == "true"
+DISCORD_WEBHOOK_URL = os.getenv("DISCORD_WEBHOOK_URL", "")
+
+# Variable pour stocker l'état du stock (évite les notifications répétées)
+previous_stock_state = False  # False = hors stock, True = en stock
+
+def send_discord_message(message: str):
+    if not DISCORD_WEBHOOK_URL:
+        print("Aucun webhook Discord défini (DISCORD_WEBHOOK_URL).")
+        return
+    
+    data = {"content": message}
+    try:
+        response = requests.post(DISCORD_WEBHOOK_URL, json=data)
+        if response.status_code >= 400:
+            print(f"Erreur lors de l'envoi du message Discord : {response.text}")
+        else:
+            print("Message Discord envoyé avec succès.")
+    except Exception as e:
+        print(f"Exception lors de l'envoi du message Discord : {e}")
+
+def check_stock_button(driver):
+    """
+    Vérifie si le bouton indiquant une rupture de stock est présent.
+    Retourne True si stock disponible, False sinon.
+    """
+    try:
+        out_of_stock_buttons = driver.find_elements(By.XPATH, "//button[contains(@class, 'stock-grey-out')]")
+        return not bool(out_of_stock_buttons)
+    except Exception as e:
+        print(f"Erreur lors de la vérification du stock : {e}")
+        return False
+
+def checkStockNvidia():
+    global previous_stock_state
+    
+    chrome_options = Options()
+    chrome_options.add_argument("--headless")
+    chrome_options.add_argument("--no-sandbox")
+    chrome_options.add_argument("--disable-gpu")
+    chrome_options.add_argument("--disable-notifications")
+    chrome_options.add_argument("--user-agent=Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) "
+                               "AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36")
+    
+    driver = webdriver.Chrome(options=chrome_options)
+    
+    for i in range(N):
+        try:
+            driver.get(URL_NVIDIA)
+            time.sleep(WAIT_TIME)  # Attente fixe après chargement de la page
+            
+            if PRINT_BODY:
+                print("===== BODY START =====")
+                print(driver.page_source)
+                print("===== BODY END =====")
+            
+            is_available = check_stock_button(driver)
+            
+            if is_available and not previous_stock_state:
+                message = {
+                    "title": f"🚀 RTX 5090 FE en stock !",
+                    "description": f":point_right: **[Achetez ici](https://marketplace.nvidia.com/fr-fr/consumer/graphics-cards/?locale=fr-fr&page=1&limit=12&gpu=RTX%205090,RTX%205080&manufacturer=NVIDIA)**",
+                    "color": 3066993,  # Couleur verte
+                    "timestamp": time.strftime("%Y-%m-%dT%H:%M:%S", time.gmtime()),
+                    #"thumbnail": {
+                    #    "url": "https://www.nvidia.com/content/dam/en-zz/Solutions/geforce/graphic-cards/50-series/rtx-5090/geforce-rtx-5090-learn-more-og-1200x630.jpg"
+                    #}
+                }
+
+                payload = {
+                    "content": "@everyone",
+                    "username": "Nvidia Bot",
+                    "embeds": [message]
+                }
+                
+                send_discord_message(message)
+                previous_stock_state = True  # Met à jour l'état du stock
+            elif not is_available:
+                previous_stock_state = False  # Remet à zéro pour permettre une future notification
+                
+        except Exception as e:
+            print("ERREUR DURANT UNE VÉRIFICATION :")
+            traceback.print_exc()
+            if ERROR_ALERT:
+                send_discord_message(f"Erreur dans le script :\n```\n{traceback.format_exc()}\n```")
+    
+    driver.quit()
+
+if __name__ == "__main__":
+    print(f"Script démarré. Vérification toutes les {RUN_INTERVAL} secondes...")
+    while True:
+        try:
+            checkStockNvidia()
+        except Exception as e:
+            print("ERREUR GLOBALE : ")
+            traceback.print_exc()
+            if ERROR_ALERT:
+                send_discord_message(f"Erreur globale dans le script :\n```\n{traceback.format_exc()}\n```")
+        
+        print(f"Prochaine vérification dans {RUN_INTERVAL} secondes...")
+        time.sleep(RUN_INTERVAL)