Update app.py

app.py

@@ -1,536 +1,163 @@
-from flask import Flask, render_template, request, jsonify, redirect, url_for
+from flask import Flask, request, jsonify, redirect, url_for
 import requests
 from bs4 import BeautifulSoup
 import random
 import string
 import json
 import threading
-import os
-import time
-from datetime import datetime
 
 app = Flask(__name__)
 
-# Configuration globale
-config = {
-    'base_url': "https://ivoire-startup-tracker-edithbrou.replit.app",
-    'accounts_file': "accounts_data.json",
-    'is_running': False,
-    'progress': {
-        'total': 0,
-        'current': 0,
-        'success': 0,
-        'failed': 0,
-        'last_username': '',
-        'last_status': '',
-        'start_time': None,
-        'end_time': None
-    }
+# Fichier JSON pour sauvegarder la progression
+JSON_FILE = "accounts.json"
+
+# Variable globale pour suivre la progression
+progress_data = {
+    "accounts": [],
+    "total": 0,
+    "created": 0,
+    "in_progress": False
 }
 
-# Fonction pour générer un nom d'utilisateur aléatoire (min 80 caractères)
-def generate_random_username(min_length=80, max_length=100):
-    """Génère un nom d'utilisateur aléatoire d'au moins 80 caractères"""
-    length = random.randint(min_length, max_length)
+# Fonctions de génération aléatoire
+
+def generate_random_username(length=80):
+    """Génère un nom d'utilisateur aléatoire avec une chaîne continue d'au moins 80 caractères."""
     return ''.join(random.choice(string.ascii_lowercase) for _ in range(length))
 
-# Fonction pour générer une adresse email aléatoire
 def generate_random_email():
-    """Génère une adresse email aléatoire"""
-    username = ''.join(random.choice(string.ascii_lowercase) for _ in range(12))
+    """Génère une adresse email aléatoire avec un nom court."""
+    username = ''.join(random.choice(string.ascii_lowercase) for _ in range(8))
     domains = ["gmail.com", "yahoo.com", "outlook.com", "example.com"]
     return f"{username}@{random.choice(domains)}"
 
-# Fonction pour générer un mot de passe aléatoire
-def generate_random_password(length=12):
-    """Génère un mot de passe aléatoire"""
-    chars = string.ascii_letters + string.digits + "!@#$%^&*"
+def generate_random_password(length=10):
+    """Génère un mot de passe aléatoire."""
+    chars = string.ascii_letters + string.digits + string.punctuation
     return ''.join(random.choice(chars) for _ in range(length))
 
-# Fonction pour créer un compte
+# URL du site à enregistrer
+BASE_URL = "https://ivoire-startup-tracker-edithbrou.replit.app"
+REGISTER_URL = f"{BASE_URL}/register"
+
 def create_account(is_startup_rep=False):
-    """Crée un compte sur le site web"""
-    register_url = f"{config['base_url']}/register"
-    
-    # Première requête pour récupérer le token CSRF
+    """Crée un compte sur le site et renvoie les informations du compte en cas de succès."""
     session = requests.Session()
-    try:
-        response = session.get(register_url)
-        
-        if response.status_code != 200:
-            return {'success': False, 'error': f"Erreur lors de l'accès à la page: {response.status_code}"}
-        
-        # Extraire le token CSRF
-        soup = BeautifulSoup(response.text, 'html.parser')
-        csrf_token = soup.find('input', {'id': 'csrf_token'}).get('value')
-        
-        if not csrf_token:
-            return {'success': False, 'error': "Impossible de trouver le token CSRF"}
-        
-        # Générer des informations de compte aléatoires
-        username = generate_random_username()
-        email = generate_random_email()
-        password = generate_random_password()
-        
-        # Préparer les données du formulaire
-        form_data = {
-            'csrf_token': csrf_token,
-            'username': username,
-            'email': email,
-            'password': password,
-            'confirm_password': password,
-            'submit': 'Register'
-        }
-        
-        # Ajouter l'option startup rep si nécessaire
-        if is_startup_rep:
-            form_data['is_startup_rep'] = 'y'
-        
-        # Envoyer le formulaire
-        headers = {
-            'Referer': register_url,
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
-        }
-        
-        response = session.post(register_url, data=form_data, headers=headers)
-        
-        result = {
-            'success': response.status_code == 200 or response.status_code == 302,
+    response = session.get(REGISTER_URL)
+    
+    if response.status_code != 200:
+        print(f"Erreur lors de l'accès à la page d'inscription: {response.status_code}")
+        return False
+    
+    soup = BeautifulSoup(response.text, 'html.parser')
+    csrf_input = soup.find('input', {'id': 'csrf_token'})
+    if not csrf_input:
+        print("Impossible de trouver le token CSRF")
+        return False
+    csrf_token = csrf_input.get('value')
+    
+    # Utilisation d'un nom d'utilisateur d'au moins 80 caractères
+    username = generate_random_username(80)
+    email = generate_random_email()
+    password = generate_random_password()
+    
+    form_data = {
+        'csrf_token': csrf_token,
+        'username': username,
+        'email': email,
+        'password': password,
+        'confirm_password': password,
+        'submit': 'Register'
+    }
+    if is_startup_rep:
+        form_data['is_startup_rep'] = 'y'
+    
+    headers = {
+        'Referer': REGISTER_URL,
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64)'
+    }
+    
+    response = session.post(REGISTER_URL, data=form_data, headers=headers)
+    if response.status_code in [200, 302]:
+        print(f"Compte créé avec succès: {username}")
+        account = {
             'username': username,
             'email': email,
             'password': password,
-            'is_startup_rep': is_startup_rep,
-            'created_at': datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-            'status_code': response.status_code
+            'is_startup_rep': is_startup_rep
         }
-        
-        return result
-    
-    except Exception as e:
-        return {'success': False, 'error': str(e)}
+        return account
+    else:
+        print(f"Échec de la création du compte: {response.status_code}")
+        return False
 
-# Fonction pour créer plusieurs comptes en arrière-plan
-def create_accounts_background(num_accounts, startup_ratio=0.3):
-    config['progress'] = {
-        'total': num_accounts,
-        'current': 0,
-        'success': 0,
-        'failed': 0,
-        'last_username': '',
-        'last_status': 'Démarrage...',
-        'start_time': datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-        'end_time': None
-    }
-    
-    # Charger les comptes existants
-    accounts = []
-    if os.path.exists(config['accounts_file']):
-        try:
-            with open(config['accounts_file'], 'r') as f:
-                accounts = json.load(f)
-        except:
-            accounts = []
+def create_multiple_accounts(num_accounts, startup_ratio=0.3):
+    """Crée plusieurs comptes en tâche de fond et met à jour le fichier JSON de progression."""
+    global progress_data
+    progress_data["in_progress"] = True
+    progress_data["total"] = num_accounts
+    progress_data["created"] = 0
+    progress_data["accounts"] = []
     
     for i in range(num_accounts):
-        if not config['is_running']:
-            break
-            
         is_startup = random.random() < startup_ratio
-        
-        config['progress']['current'] = i + 1
-        config['progress']['last_status'] = f"Création du compte {i+1}/{num_accounts}..."
-        
-        result = create_account(is_startup_rep=is_startup)
-        
-        if result.get('success', False):
-            config['progress']['success'] += 1
-            config['progress']['last_username'] = result['username']
-            config['progress']['last_status'] = f"Compte {i+1} créé avec succès"
-            accounts.append(result)
-        else:
-            config['progress']['failed'] += 1
-            config['progress']['last_status'] = f"Échec de la création du compte {i+1}: {result.get('error', 'Erreur inconnue')}"
-        
-        # Enregistrer les données régulièrement
-        with open(config['accounts_file'], 'w') as f:
-            json.dump(accounts, f, indent=2)
-            
-        # Petite pause pour éviter de surcharger le serveur
-        time.sleep(1)
-    
-    config['is_running'] = False
-    config['progress']['end_time'] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    config['progress']['last_status'] = "Terminé"
-    
-    # Enregistrement final
-    with open(config['accounts_file'], 'w') as f:
-        json.dump(accounts, f, indent=2)
+        print(f"\nCréation du compte {i+1}/{num_accounts}...")
+        account = create_account(is_startup_rep=is_startup)
+        if account:
+            progress_data["accounts"].append(account)
+            progress_data["created"] += 1
+            # Écriture dans le fichier JSON après chaque création
+            with open(JSON_FILE, "w") as f:
+                json.dump(progress_data, f, indent=2)
+    
+    progress_data["in_progress"] = False
+    # Mise à jour finale du fichier JSON
+    with open(JSON_FILE, "w") as f:
+        json.dump(progress_data, f, indent=2)
 
 # Routes Flask
-@app.route('/')
+
+@app.route("/")
 def index():
-    return render_template('index.html', config=config)
+    return '''
+    <h1>Gestion de création de comptes</h1>
+    <form action="/start" method="post">
+        <label for="num">Nombre de comptes à créer:</label>
+        <input type="number" id="num" name="num" value="5" min="1"><br><br>
+        <label for="startup_ratio">Ratio de représentants startup (0-1):</label>
+        <input type="text" id="startup_ratio" name="startup_ratio" value="0.3"><br><br>
+        <button type="submit">Start</button>
+    </form>
+    <br>
+    <a href="/progress">Voir la progression</a>
+    '''
 
-@app.route('/start', methods=['POST'])
+@app.route("/start", methods=["POST"])
 def start():
-    if config['is_running']:
-        return jsonify({"status": "error", "message": "Une génération est déjà en cours"})
-    
-    num_accounts = int(request.form.get('num_accounts', 10))
-    startup_ratio = float(request.form.get('startup_ratio', 0.3))
-    
-    config['is_running'] = True
+    try:
+        num = int(request.form.get("num", 5))
+    except ValueError:
+        num = 5
+    try:
+        startup_ratio = float(request.form.get("startup_ratio", 0.3))
+    except ValueError:
+        startup_ratio = 0.3
     
-    # Démarrer le processus en arrière-plan
-    thread = threading.Thread(target=create_accounts_background, args=(num_accounts, startup_ratio))
-    thread.daemon = True
+    # Démarrer la création en tâche de fond
+    thread = threading.Thread(target=create_multiple_accounts, args=(num, startup_ratio))
     thread.start()
     
-    return jsonify({"status": "success", "message": "Génération démarrée"})
-
-@app.route('/stop', methods=['POST'])
-def stop():
-    config['is_running'] = False
-    return jsonify({"status": "success", "message": "Arrêt demandé"})
+    return redirect(url_for("index"))
 
-@app.route('/progress')
+@app.route("/progress")
 def progress():
-    return jsonify(config['progress'])
-
-@app.route('/accounts')
-def view_accounts():
-    page = int(request.args.get('page', 1))
-    per_page = 20
-    
-    accounts = []
-    if os.path.exists(config['accounts_file']):
-        try:
-            with open(config['accounts_file'], 'r') as f:
-                accounts = json.load(f)
-        except:
-            accounts = []
-            
-    total_accounts = len(accounts)
-    total_pages = (total_accounts + per_page - 1) // per_page
-    
-    start_idx = (page - 1) * per_page
-    end_idx = start_idx + per_page
-    
-    current_accounts = accounts[start_idx:end_idx]
-    
-    return render_template(
-        'accounts.html',
-        accounts=current_accounts,
-        page=page,
-        total_pages=total_pages,
-        total_accounts=total_accounts
-    )
+    """Retourne le fichier JSON de progression ou les données globales en cas d'absence du fichier."""
+    try:
+        with open(JSON_FILE, "r") as f:
+            data = json.load(f)
+    except Exception as e:
+        data = progress_data
+    return jsonify(data)
 
-if __name__ == '__main__':
-    # Créer le dossier templates s'il n'existe pas
-    if not os.path.exists('templates'):
-        os.makedirs('templates')
-    
-    # Créer les templates HTML
-    with open('templates/index.html', 'w') as f:
-        f.write('''
-<!DOCTYPE html>
-<html lang="fr">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Générateur de comptes automatique</title>
-    <link href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css" rel="stylesheet">
-    <style>
-        body { background-color: #f5f5f5; }
-        .card { box-shadow: 0 4px 8px rgba(0,0,0,0.1); }
-        .progress { height: 25px; }
-        #status-display { height: 200px; overflow-y: auto; }
-    </style>
-</head>
-<body>
-    <div class="container mt-5">
-        <div class="row">
-            <div class="col-md-12 mb-4">
-                <div class="card">
-                    <div class="card-header bg-primary text-white">
-                        <h3 class="card-title mb-0">Générateur de comptes - Startup Côte d'Ivoire</h3>
-                    </div>
-                    <div class="card-body">
-                        <form id="generator-form">
-                            <div class="row mb-3">
-                                <div class="col-md-6">
-                                    <label for="num_accounts" class="form-label">Nombre de comptes à créer</label>
-                                    <input type="number" class="form-control" id="num_accounts" name="num_accounts" min="1" max="1000" value="10">
-                                </div>
-                                <div class="col-md-6">
-                                    <label for="startup_ratio" class="form-label">Ratio de représentants de startup (%)</label>
-                                    <input type="number" class="form-control" id="startup_ratio" name="startup_ratio" min="0" max="100" value="30">
-                                </div>
-                            </div>
-                            <div class="d-grid gap-2 d-md-flex justify-content-md-center">
-                                <button type="submit" id="start-btn" class="btn btn-success btn-lg">
-                                    <i class="fas fa-play"></i> Démarrer
-                                </button>
-                                <button type="button" id="stop-btn" class="btn btn-danger btn-lg">
-                                    <i class="fas fa-stop"></i> Arrêter
-                                </button>
-                                <a href="/accounts" class="btn btn-primary btn-lg">
-                                    <i class="fas fa-list"></i> Voir les comptes
-                                </a>
-                            </div>
-                        </form>
-                    </div>
-                </div>
-            </div>
-            
-            <div class="col-md-12">
-                <div class="card">
-                    <div class="card-header bg-info text-white">
-                        <h4 class="card-title mb-0">Progression</h4>
-                    </div>
-                    <div class="card-body">
-                        <div class="progress mb-3">
-                            <div id="progress-bar" class="progress-bar progress-bar-striped progress-bar-animated" role="progressbar" style="width: 0%">0%</div>
-                        </div>
-                        
-                        <div class="row text-center mb-3">
-                            <div class="col">
-                                <div class="card bg-light">
-                                    <div class="card-body">
-                                        <h5>Total</h5>
-                                        <h3 id="total-count">0</h3>
-                                    </div>
-                                </div>
-                            </div>
-                            <div class="col">
-                                <div class="card bg-success text-white">
-                                    <div class="card-body">
-                                        <h5>Réussis</h5>
-                                        <h3 id="success-count">0</h3>
-                                    </div>
-                                </div>
-                            </div>
-                            <div class="col">
-                                <div class="card bg-danger text-white">
-                                    <div class="card-body">
-                                        <h5>Échecs</h5>
-                                        <h3 id="failed-count">0</h3>
-                                    </div>
-                                </div>
-                            </div>
-                        </div>
-                        
-                        <div class="card mb-3">
-                            <div class="card-header">
-                                <h5>Dernier statut</h5>
-                            </div>
-                            <div class="card-body">
-                                <p id="last-status">-</p>
-                                <p>Dernier compte créé: <span id="last-username" class="text-primary fw-bold">-</span></p>
-                            </div>
-                        </div>
-                        
-                        <div class="row">
-                            <div class="col-md-6">
-                                <p>Heure de début: <span id="start-time">-</span></p>
-                            </div>
-                            <div class="col-md-6 text-end">
-                                <p>Heure de fin: <span id="end-time">-</span></p>
-                            </div>
-                        </div>
-                    </div>
-                </div>
-            </div>
-        </div>
-    </div>
-    
-    <script src="https://cdn.jsdelivr.net/npm/[email protected]/dist/js/bootstrap.bundle.min.js"></script>
-    <script src="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/js/all.min.js"></script>
-    <script>
-        document.addEventListener('DOMContentLoaded', function() {
-            // Formulaire de génération
-            document.getElementById('generator-form').addEventListener('submit', function(e) {
-                e.preventDefault();
-                
-                const numAccounts = document.getElementById('num_accounts').value;
-                const startupRatio = document.getElementById('startup_ratio').value / 100;
-                
-                fetch('/start', {
-                    method: 'POST',
-                    headers: {
-                        'Content-Type': 'application/x-www-form-urlencoded',
-                    },
-                    body: `num_accounts=${numAccounts}&startup_ratio=${startupRatio}`
-                })
-                .then(response => response.json())
-                .then(data => {
-                    if (data.status === 'success') {
-                        startProgressUpdates();
-                    } else {
-                        alert(data.message);
-                    }
-                });
-            });
-            
-            // Bouton d'arrêt
-            document.getElementById('stop-btn').addEventListener('click', function() {
-                fetch('/stop', {
-                    method: 'POST'
-                })
-                .then(response => response.json())
-                .then(data => {
-                    alert(data.message);
-                });
-            });
-            
-            // Mettre à jour la progression
-            function updateProgress() {
-                fetch('/progress')
-                .then(response => response.json())
-                .then(data => {
-                    const progressPercent = data.total > 0 ? Math.round((data.current / data.total) * 100) : 0;
-                    
-                    document.getElementById('progress-bar').style.width = `${progressPercent}%`;
-                    document.getElementById('progress-bar').textContent = `${progressPercent}%`;
-                    
-                    document.getElementById('total-count').textContent = data.total;
-                    document.getElementById('success-count').textContent = data.success;
-                    document.getElementById('failed-count').textContent = data.failed;
-                    
-                    document.getElementById('last-status').textContent = data.last_status;
-                    document.getElementById('last-username').textContent = data.last_username || '-';
-                    
-                    document.getElementById('start-time').textContent = data.start_time || '-';
-                    document.getElementById('end-time').textContent = data.end_time || '-';
-                });
-            }
-            
-            let progressInterval;
-            
-            function startProgressUpdates() {
-                // Arrêter l'intervalle précédent s'il existe
-                if (progressInterval) {
-                    clearInterval(progressInterval);
-                }
-                
-                // Mettre à jour immédiatement
-                updateProgress();
-                
-                // Puis toutes les 2 secondes
-                progressInterval = setInterval(updateProgress, 2000);
-            }
-            
-            // Vérifier si une génération est déjà en cours au chargement de la page
-            updateProgress();
-        });
-    </script>
-</body>
-</html>
-        ''')
-    
-    with open('templates/accounts.html', 'w') as f:
-        f.write('''
-<!DOCTYPE html>
-<html lang="fr">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Liste des comptes générés</title>
-    <link href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css" rel="stylesheet">
-    <style>
-        body { background-color: #f5f5f5; }
-        .card { box-shadow: 0 4px 8px rgba(0,0,0,0.1); }
-        .username-cell {
-            max-width: 200px;
-            overflow: hidden;
-            text-overflow: ellipsis;
-            white-space: nowrap;
-        }
-        .pagination { margin-bottom: 0; }
-    </style>
-</head>
-<body>
-    <div class="container mt-5">
-        <div class="row">
-            <div class="col-md-12 mb-4">
-                <div class="card">
-                    <div class="card-header bg-primary text-white d-flex justify-content-between align-items-center">
-                        <h3 class="mb-0">Liste des comptes générés</h3>
-                        <a href="/" class="btn btn-light">Retour au générateur</a>
-                    </div>
-                    <div class="card-body">
-                        <div class="alert alert-info">
-                            Total des comptes générés: {{ total_accounts }}
-                        </div>
-                        
-                        <div class="table-responsive">
-                            <table class="table table-striped table-hover">
-                                <thead>
-                                    <tr>
-                                        <th>#</th>
-                                        <th>Nom d'utilisateur</th>
-                                        <th>Email</th>
-                                        <th>Mot de passe</th>
-                                        <th>Startup Rep</th>
-                                        <th>Date de création</th>
-                                        <th>Statut</th>
-                                    </tr>
-                                </thead>
-                                <tbody>
-                                    {% for account in accounts %}
-                                    <tr>
-                                        <td>{{ (page - 1) * 20 + loop.index }}</td>
-                                        <td class="username-cell" title="{{ account.username }}">{{ account.username }}</td>
-                                        <td>{{ account.email }}</td>
-                                        <td>{{ account.password }}</td>
-                                        <td>{% if account.is_startup_rep %}Oui{% else %}Non{% endif %}</td>
-                                        <td>{{ account.created_at }}</td>
-                                        <td>
-                                            {% if account.success %}
-                                            <span class="badge bg-success">Succès</span>
-                                            {% else %}
-                                            <span class="badge bg-danger">Échec</span>
-                                            {% endif %}
-                                        </td>
-                                    </tr>
-                                    {% endfor %}
-                                </tbody>
-                            </table>
-                        </div>
-                        
-                        {% if total_pages > 1 %}
-                        <div class="d-flex justify-content-center mt-4">
-                            <nav aria-label="Page navigation">
-                                <ul class="pagination">
-                                    <li class="page-item {% if page == 1 %}disabled{% endif %}">
-                                        <a class="page-link" href="{{ url_for('view_accounts', page=page-1) if page > 1 else '#' }}">Précédent</a>
-                                    </li>
-                                    
-                                    {% for p in range(1, total_pages + 1) %}
-                                        {% if p == page %}
-                                        <li class="page-item active"><span class="page-link">{{ p }}</span></li>
-                                        {% else %}
-                                        <li class="page-item"><a class="page-link" href="{{ url_for('view_accounts', page=p) }}">{{ p }}</a></li>
-                                        {% endif %}
-                                    {% endfor %}
-                                    
-                                    <li class="page-item {% if page == total_pages %}disabled{% endif %}">
-                                        <a class="page-link" href="{{ url_for('view_accounts', page=page+1) if page < total_pages else '#' }}">Suivant</a>
-                                    </li>
-                                </ul>
-                            </nav>
-                        </div>
-                        {% endif %}
-                    </div>
-                </div>
-            </div>
-        </div>
-    </div>
-    
-    <script src="https://cdn.jsdelivr.net/npm/[email protected]/dist/js/bootstrap.bundle.min.js"></script>
-</body>
-</html>
-        ''')
-    
-    app.run(debug=True)
+if __name__ == "__main__":
+    app.run(debug=True)