Add files via upload

BDIXTester.py

@@ -0,0 +1,214 @@
+import requests
+import concurrent.futures
+import os
+import sys
+from datetime import datetime
+import time
+import traceback
+from rich.console import Console
+from rich.progress import Progress, SpinnerColumn, TextColumn
+from rich.panel import Panel
+from rich.text import Text
+
+class BDIXTester:
+    def __init__(self):
+        self.console = Console()
+        self.websites = []
+        self.working_sites = []
+        self.timeout = 5  # Default timeout
+        self.bdix_url = "https://raw.githubusercontent.com/shihabalter/BDIX-TESTER/refs/heads/main/bdix.txt"  # URL for bdix.txt
+
+    def set_timeout(self, timeout):
+        """Set the timeout value for requests."""
+        try:
+            self.timeout = float(timeout)
+            self.console.print(f"[green]Timeout set to {self.timeout} seconds[/green]")
+        except ValueError:
+            self.console.print("[red]Invalid timeout value. Using default (5 seconds)[/red]")
+            self.timeout = 5
+
+    def download_bdix_file(self):
+        """Download the bdix.txt file from the repository."""
+        try:
+            self.console.print("[yellow]Downloading bdix.txt file...[/yellow]")
+            response = requests.get(self.bdix_url, timeout=10)
+            response.raise_for_status()
+
+            # Get the path for bdix.txt
+            exe_dir = os.path.dirname(sys.executable) if getattr(sys, 'frozen', False) else os.path.dirname(__file__)
+            bdix_path = os.path.join(exe_dir, 'bdix.txt')
+
+            # Save the file
+            with open(bdix_path, 'w', encoding='utf-8') as file:
+                file.write(response.text)
+
+            self.console.print("[green]Successfully downloaded bdix.txt[/green]")
+            return True
+        except Exception as e:
+            self.console.print(f"[red]Error downloading bdix.txt: {str(e)}[/red]")
+            return False
+
+    def load_websites(self):
+        """Load websites from the local bdix.txt file or download if not present."""
+        try:
+            # Get the path for bdix.txt
+            exe_dir = os.path.dirname(sys.executable) if getattr(sys, 'frozen', False) else os.path.dirname(__file__)
+            bdix_path = os.path.join(exe_dir, 'bdix.txt')
+
+            # If file doesn't exist or is empty, try to download it
+            if not os.path.exists(bdix_path) or os.path.getsize(bdix_path) == 0:
+                if not self.download_bdix_file():
+                    return False
+
+            # Read the file
+            with open(bdix_path, 'r', encoding='utf-8') as file:
+                self.websites = [line.strip() for line in file if line.strip()]
+
+            if self.websites:
+                self.console.print(f"[green]Successfully loaded {len(self.websites)} websites[/green]")
+                return True
+
+            self.console.print("[red]Error: bdix.txt is empty.[/red]")
+            return False
+
+        except Exception as e:
+            self.console.print(f"[red]Error loading websites: {str(e)}[/red]")
+            return False
+
+    def test_website(self, url):
+        """Test a single website's connectivity."""
+        if not url.startswith(('http://', 'https://')):
+            url = f'http://{url}'
+        try:
+            response = requests.get(url, timeout=self.timeout)
+            if response.status_code == 200:
+                return url, True
+            return url, False
+        except:
+            return url, False
+
+    def save_results(self):
+        """Save working websites to a file."""
+        try:
+            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+            filename = f"working_sites_{timestamp}.txt"
+
+            # Save in the same directory as the executable
+            if getattr(sys, 'frozen', False):
+                filename = os.path.join(os.path.dirname(sys.executable), filename)
+
+            with open(filename, 'w', encoding='utf-8') as file:
+                for site in self.working_sites:
+                    file.write(f"{site}\n")
+
+            self.console.print(f"[green]Results saved to {filename}[/green]")
+        except Exception as e:
+            self.console.print(f"[red]Error saving results: {str(e)}[/red]")
+
+    def run_tests(self):
+        """Run the connectivity tests."""
+        try:
+            self.console.clear()
+            self.console.print(Panel.fit(
+                Text("BDIX TESTER", justify="center", style="bold white on blue"),
+                subtitle="Testing BDIX Connectivity"
+            ))
+
+            if not self.websites:
+                if self.load_websites():
+                    self.console.print("[green]Successfully reloaded websites.[/green]")
+                else:
+                    self.console.print("[red]No websites loaded. Please check your internet connection.[/red]")
+                    return
+
+            self.working_sites = []
+            total_sites = len(self.websites)
+
+            with Progress(
+                SpinnerColumn(),
+                TextColumn("[progress.description]{task.description}"),
+                console=self.console
+            ) as progress:
+                task = progress.add_task("[cyan]Testing websites...", total=total_sites)
+
+                with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
+                    future_to_url = {executor.submit(self.test_website, url): url for url in self.websites}
+
+                    for future in concurrent.futures.as_completed(future_to_url):
+                        url, is_working = future.result()
+                        progress.advance(task)
+
+                        if is_working:
+                            self.working_sites.append(url)
+                            self.console.print(f"[green]✓ {url} is working[/green]")
+                        else:
+                            self.console.print(f"[red]✗ {url} is not working[/red]")
+
+            self.console.print("\n[bold]Test Results:[/bold]")
+            self.console.print(f"Total sites tested: {total_sites}")
+            self.console.print(f"Working sites: {len(self.working_sites)}")
+            self.console.print(f"Failed sites: {total_sites - len(self.working_sites)}")
+
+        except Exception as e:
+            self.console.print(f"[red]Error during testing: {str(e)}[/red]")
+            self.console.print(traceback.format_exc())
+
+def main():
+    try:
+        # Add a small delay to ensure console is ready
+        time.sleep(1)
+
+        # Create console and show welcome message
+        console = Console()
+        console.print(Panel.fit(
+            Text("Welcome to BDIX TESTER", justify="center", style="bold white on blue"),
+            subtitle="Starting up..."
+        ))
+
+        tester = BDIXTester()
+        if not tester.load_websites():
+            console.print("[yellow]Attempting to download bdix.txt...[/yellow]")
+            if not tester.download_bdix_file() or not tester.load_websites():
+                console.print("[red]Failed to initialize. Please check your internet connection.[/red]")
+                input("Press Enter to exit...")
+                return
+
+        while True:
+            print("\n1. Start Testing")
+            print("2. Save Working Sites")
+            print("3. Set Timeout")
+            print("4. Reload BDIX List")
+            print("5. Exit")
+
+            try:
+                choice = input("\nEnter your choice (1-5): ")
+            except EOFError:
+                choice = '5'
+
+            if choice == '1':
+                tester.run_tests()
+            elif choice == '2':
+                if tester.working_sites:
+                    tester.save_results()
+                else:
+                    print("No results to save. Please run the tests first.")
+            elif choice == '3':
+                timeout = input("Enter timeout in seconds (default is 5): ")
+                tester.set_timeout(timeout)
+            elif choice == '4':
+                if tester.download_bdix_file() and tester.load_websites():
+                    print("Successfully reloaded BDIX list.")
+                else:
+                    print("Failed to reload BDIX list.")
+            elif choice == '5':
+                print("Thank you for using BDIX Tester!")
+                break
+            else:
+                print("Invalid choice. Please try again.")
+    except Exception as e:
+        print(f"Error: {str(e)}")
+        print(traceback.format_exc())
+        input("Press Enter to exit...")
+
+if __name__ == "__main__":
+    main()