From 86665b7137a5107eb744b68c353cea7e590a4c76 Mon Sep 17 00:00:00 2001 From: Djeex Date: Wed, 19 Feb 2025 17:19:49 +0000 Subject: [PATCH] auto sku --- README.md | 12 +++++------- docker/compose.yaml | 4 ++-- nvidia-stock-bot.py | 48 ++++++++++++++++++++++++++++++--------------- 3 files changed, 39 insertions(+), 25 deletions(-) diff --git a/README.md b/README.md index 75334d5..69d01d9 100644 --- a/README.md +++ b/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) -- Vérifie si RTX 5090, RTX 5080, RTX 5070ti et RTX 5070 sont en stock +- Appelle l'API de Nvidia listant le produit (par défaut toutes les 60s) +- 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é - - - 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= #URL de l'API + - 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_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 ``` diff --git a/docker/compose.yaml b/docker/compose.yaml index 0809028..596708b 100644 --- a/docker/compose.yaml +++ b/docker/compose.yaml @@ -9,7 +9,7 @@ services: environment: - DISCORD_WEBHOOK_URL=${DS_HOOK} - REFRESH_TIME=${FREQ} - - GPU_TARGETS=${GPU} #SKU - - API_URL=${URL} #URL de l'API + - API_URL_SKU=${API_URL_SKU} + - 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 \ No newline at end of file diff --git a/nvidia-stock-bot.py b/nvidia-stock-bot.py index e5bb208..85677e7 100644 --- a/nvidia-stock-bot.py +++ b/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'] - GPU_TARGETS = os.environ['GPU_TARGETS'].split(",") # Séparer en liste - GPU_TARGETS = [gpu.strip() for gpu in GPU_TARGETS] # Nettoyer les espaces + API_URL_SKU = os.environ['API_URL_SKU'] + API_URL_STOCK = os.environ['API_URL_STOCK'] 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"GPU recherché: {GPU_TARGETS}") +print(f"url de l'API qui liste les SKU: {API_URL_SKU}") +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]