141 lines
4.9 KiB
Python
141 lines
4.9 KiB
Python
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") != "out_of_stock")
|
||
|
||
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 **En stock !**\n"
|
||
f"Carte : `{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)
|