Completed the framework

Dockerfile

@@ -0,0 +1,28 @@
+# Use an official Python runtime as a parent image
+FROM python:3.8-slim
+
+# Set environment variables
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
+
+# Set the working directory
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    xvfb \
+    x11-xserver-utils \
+    chromium-driver
+
+# Install Python dependencies
+COPY requirements.txt /app/
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the application code
+COPY . /app/
+
+# Expose port 5000 for Flask
+EXPOSE 5000
+
+# Command to run the application
+CMD ["python", "app.py"]

README.md

@@ -1 +0,0 @@
-# GHM

app.py

@@ -0,0 +1,47 @@
+from flask import Flask, render_template, request, jsonify
+from form_filler import FormFiller
+import threading
+
+app = Flask(__name__)
+form_filler = FormFiller()
+
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+@app.route('/start', methods=['POST'])
+def start_form_filling():
+    url = request.json['url']
+    iterations = int(request.json['iterations'])
+    
+    def run_form_filler():
+        form_filler.fill_form_in_parallel(url, iterations)
+    
+    form_filler.total_iterations = iterations
+    form_filler.iterations_left = iterations
+    form_filler.responses_sent = 0
+    form_filler.errors = 0
+    form_filler.environment_status = []
+    
+    thread = threading.Thread(target=run_form_filler)
+    thread.start()
+    
+    return jsonify({"message": "Form filling started"})
+
+@app.route('/stop', methods=['POST'])
+def stop_form_filling():
+    form_filler.stop()
+    return jsonify({"message": "Form filling stopped"})
+
+@app.route('/status', methods=['GET'])
+def get_status():
+    return jsonify({
+        "total_iterations": form_filler.total_iterations,
+        "responses_sent": form_filler.responses_sent,
+        "errors": form_filler.errors,
+        "iterations_left": form_filler.iterations_left,
+        "environment_status": form_filler.environment_status
+    })
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=5000)

form_filler.py

@@ -0,0 +1,128 @@
+import logging
+import random
+from concurrent.futures import ThreadPoolExecutor
+import time
+import os
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.common.by import By
+from xvfbwrapper import Xvfb
+from threading import Lock, Event
+
+class FormFiller:
+    def __init__(self):
+        self.total_iterations = 0
+        self.responses_sent = 0
+        self.errors = 0
+        self.iterations_left = 0
+        self.stop_flag = Event()
+        self.lock = Lock()
+        self.environment_status = []
+
+    def fill_form(self, url, num_iterations, update_callback=None, environment_id=1):
+        self.stop_flag.clear()
+        display = None
+        driver = None
+
+        try:
+            # Start Xvfb display
+            display = Xvfb(width=1280, height=720)
+            display.start()
+
+            chrome_options = Options()
+            chrome_options.add_argument("--headless")
+            chrome_options.add_argument("--no-sandbox")
+            chrome_options.add_argument("--disable-dev-shm-usage")
+            chrome_options.add_argument("--disable-gpu")
+
+            driver = webdriver.Chrome(options=chrome_options)
+            driver.get(url)
+
+            for _ in range(num_iterations):
+                if self.stop_flag.is_set():
+                    break
+
+                try:
+                    # Handling radio buttons
+                    radio_buttons = driver.find_elements(By.CSS_SELECTOR, 'div[role="radiogroup"]')
+                    for radio_group in radio_buttons:
+                        options = radio_group.find_elements(By.CSS_SELECTOR, 'div[role="radio"]')
+                        random.choice(options).click()
+
+                    # Handling checkboxes
+                    checkboxes = driver.find_elements(By.CSS_SELECTOR, 'div[role="checkbox"]')
+                    for checkbox in checkboxes:
+                        if random.choice([True, False]):
+                            checkbox.click()
+
+                    # Handling multiple choice grids
+                    grids = driver.find_elements(By.CSS_SELECTOR, 'div[role="grid"]')
+                    for grid in grids:
+                        rows = grid.find_elements(By.CSS_SELECTOR, 'div[role="row"]')
+                        for row in rows:
+                            options = row.find_elements(By.CSS_SELECTOR, 'div[role="radio"]')
+                            random.choice(options).click()
+
+                    # Submit the form
+                    submit = driver.find_element(By.CSS_SELECTOR, 'div[role="button"][aria-label="Submit"]')
+                    submit.click()
+
+                    logging.info(f"Form submitted successfully by Environment {environment_id}")
+
+                    with self.lock:
+                        self.responses_sent += 1
+                        self.iterations_left = self.total_iterations - self.responses_sent
+                        self.environment_status.append(
+                            f"Environment {environment_id}: Total Responses Sent: {self.responses_sent}, Errors: {self.errors}, Iterations Left: {self.iterations_left}"
+                        )
+
+                    # Wait and reload the form
+                    time.sleep(5)
+                    driver.get(url)
+
+                except Exception as e:
+                    with self.lock:
+                        self.errors += 1
+                        self.iterations_left = self.total_iterations - self.responses_sent
+                        self.environment_status.append(
+                            f"Environment {environment_id}: Error occurred: {e}"
+                        )
+                    logging.error(f"Error occurred in Environment {environment_id}: {e}")
+
+        finally:
+            if driver:
+                driver.quit()
+            if display:
+                if 'DISPLAY' in os.environ:
+                    display.stop()
+
+    def stop(self):
+        self.stop_flag.set()
+
+    def fill_form_in_parallel(self, url, total_iterations, update_callback=None):
+        self.total_iterations = total_iterations
+        self.iterations_left = total_iterations
+        self.responses_sent = 0
+        self.errors = 0
+        self.environment_status = []
+
+        num_envs = 10  # Total number of environments to run concurrently
+        iterations_per_env = total_iterations // num_envs  # Each environment should process its share of iterations
+
+        with ThreadPoolExecutor(max_workers=num_envs) as executor:
+            futures = []
+            for i in range(num_envs):
+                futures.append(executor.submit(self.fill_form, url, iterations_per_env, update_callback, i + 1))
+
+            for future in futures:
+                future.result()
+
+        # Handle remaining iterations
+        remaining_iterations = total_iterations % num_envs
+        if remaining_iterations > 0:
+            with ThreadPoolExecutor(max_workers=remaining_iterations) as executor:
+                futures = []
+                for i in range(remaining_iterations):
+                    futures.append(executor.submit(self.fill_form, url, 1, update_callback, i + 1))
+                for future in futures:
+                    future.result()

requirements.txt

@@ -0,0 +1,4 @@
+Flask
+selenium
+flask
+xvfbwrapper

static/script.js

@@ -0,0 +1,50 @@
+document.addEventListener('DOMContentLoaded', () => {
+    const form = document.getElementById('form-filler-form');
+    const startButton = document.getElementById('start-button');
+    const stopButton = document.getElementById('stop-button');
+    const statusDiv = document.getElementById('status');
+
+    let intervalId;
+
+    form.addEventListener('submit', async (e) => {
+        e.preventDefault();
+        const url = document.getElementById('url').value;
+        const iterations = document.getElementById('iterations').value;
+
+        startButton.disabled = true;
+        stopButton.disabled = false;
+
+        await fetch('/start', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+            },
+            body: JSON.stringify({ url, iterations }),
+        });
+
+        intervalId = setInterval(updateStatus, 1000);
+    });
+
+    stopButton.addEventListener('click', async () => {
+        await fetch('/stop', { method: 'POST' });
+        startButton.disabled = false;
+        stopButton.disabled = true;
+        clearInterval(intervalId);
+    });
+
+    async function updateStatus() {
+        const response = await fetch('/status');
+        const data = await response.json();
+        
+        statusDiv.innerHTML = `
+            <p>Total Iterations: ${data.total_iterations}</p>
+            <p>Responses Sent: ${data.responses_sent}</p>
+            <p>Errors: ${data.errors}</p>
+            <p>Iterations Left: ${data.iterations_left}</p>
+            <h3>Environment Status:</h3>
+            <ul>
+                ${data.environment_status.map(status => `<li>${status}</li>`).join('')}
+            </ul>
+        `;
+    }
+});

templates/index.html

@@ -0,0 +1,41 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Form Filler</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            max-width: 800px;
+            margin: 0 auto;
+            padding: 20px;
+        }
+        form {
+            margin-bottom: 20px;
+        }
+        input, button {
+            margin: 10px 0;
+            padding: 5px;
+        }
+        #status {
+            border: 1px solid #ccc;
+            padding: 10px;
+            margin-top: 20px;
+        }
+    </style>
+</head>
+<body>
+    <h1>Form Filler</h1>
+    <form id="form-filler-form">
+        <label for="url">Form URL:</label><br>
+        <input type="url" id="url" name="url" required><br>
+        <label for="iterations">Number of Iterations:</label><br>
+        <input type="number" id="iterations" name="iterations" required><br>
+        <button type="submit" id="start-button">Start</button>
+    </form>
+    <button id="stop-button" disabled>Stop</button>
+    <div id="status"></div>
+    <script src="/static/script.js"></script>
+</body>
+</html>