auto sku
This commit is contained in:
parent
f9bc3dfe32
commit
86665b7137
12
README.md
12
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é
|
||||
|
||||
<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= #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
|
||||
```
|
||||
|
@ -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
|
@ -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]
|
||||
|
Loading…
x
Reference in New Issue
Block a user