add_first.py

"""
SCRIPT DE COMMANDES FIRSTS LEADERBOARD
======================================

Ce script permet d'ajouter facilement des "firsts" au classement.
Utilisez-le depuis VS Code ou la ligne de commande.
"""

import requests
import json
import sys
import argparse
from datetime import datetime

SERVER_URL = "http://localhost:8080"

def add_first(username, count=1):
    """Ajouter un first à un utilisateur"""
    try:
        # Méthode GET simple
        url = f"{SERVER_URL}/add-first-simple"
        params = {
            'username': username,
            'count': count
        }
        
        response = requests.get(url, params=params, timeout=5)
        
        if response.status_code == 200:
            data = response.json()
            print(f"[Succes] {data['message']}")
            return True
        else:
            print(f"[Erreur] Erreur {response.status_code}: {response.text}")
            return False
            
    except requests.exceptions.ConnectionError:
        print(f"[Erreur] Impossible de se connecter au serveur sur {SERVER_URL}")
        print("[Info] Assurez-vous que le serveur est démarré avec: python firsts_server.py")
        return False
    except Exception as e:
        print(f"[Erreur] {e}")
        return False

def main():
    global SERVER_URL
    
    parser = argparse.ArgumentParser(description='Ajouter des firsts au classement')
    parser.add_argument('username', help='Nom d\'utilisateur')
    parser.add_argument('-c', '--count', type=int, default=1, help='Nombre de firsts à ajouter (défaut: 1)')
    parser.add_argument('--server', default=SERVER_URL, help=f'URL du serveur (défaut: {SERVER_URL})')
    
    args = parser.parse_args()
    
    # Mettre à jour l'URL du serveur si spécifiée
    SERVER_URL = args.server
    
    print(f"[Info] Ajout de {args.count} first(s) pour {args.username}...")
    
    success = add_first(args.username, args.count)
    
    if success:
        print(f"[Succes] Terminé ! [{datetime.now().strftime('%H:%M:%S')}]")
    else:
        sys.exit(1)

if __name__ == "__main__":
    if len(sys.argv) == 1:
        # Mode interactif si aucun argument
        print("[Info] COMMANDES FIRSTS LEADERBOARD")
        print("=" * 40)
        print()
        
        while True:
            try:
                username = input("[Info] Nom d'utilisateur (ou 'quit' pour quitter): ").strip()
                
                if username.lower() in ['quit', 'exit', 'q']:
                    print("[Info] Au revoir !")
                    break
                
                if not username:
                    print("[Erreur] Veuillez entrer un nom d'utilisateur")
                    continue
    
                count = 1
                
                add_first(username, count)
                print()
                
            except KeyboardInterrupt:
                print("\n[Info] Au revoir !")
                break
            except Exception as e:
                print(f"[Erreur] {e}")
    else:
        main()