feat<main.py>: enable multiple iterations and summary stats/graph of results

scripts/Internet Speed Checker/main.py

@@ -1,20 +1,17 @@
-# Internet Speed Checker
-
 import speedtest
-import time
-import os
+import matplotlib.pyplot as plt
+import pandas as pd
 
 
-def main():
+def run_speed_test():
 
-    # Clear the screen
-    os.system('cls' if os.name == 'nt' else 'clear')
+    download_speeds = []
+    upload_speeds = []
+    ping_values = []
 
-    # Print the title
+    # Run the speedtest while the user wants to continue running iterations
+    while True:
         print('Internet Speed Checker')
-    # Print the current time
-    print(time.strftime('%H:%M:%S'))
-    print("Connecting to server...")
 
         # Get the speed
         s = speedtest.Speedtest()
@@ -23,15 +20,57 @@ def main():
         s.upload()
         results_dict = s.results.dict()
 
+        # Append speeds and ping
+        download_speeds.append(results_dict['download'] / 1024 / 1024)
+        upload_speeds.append(results_dict['upload'] / 1024 / 1024)
+        ping_values.append(results_dict['ping'])
+
         # Print the results
-    print(f"Download: {results_dict['download'] / 1024 / 1024:.2f} Mbit/s")
-    print(f"Upload: {results_dict['upload'] / 1024 / 1024:.2f} Mbit/s")
-    print(f"Ping: {results_dict['ping']:.2f} ms")
+        print(f"Iteration {len(download_speeds)}:")
+        print(f"  Download Speed: {download_speeds[-1]:.2f} Mbit/s")
+        print(f"  Upload Speed: {upload_speeds[-1]:.2f} Mbit/s")
+        print(f"  Ping: {ping_values[-1]:.2f} ms")
         print(f"  ISP: {results_dict['client']['isp']}")
 
-    # Wait 5 seconds
-    time.sleep(5)
+        # Ask the user whether they want to run another iteration
+        user_input = input("Do you want to run another iteration? (yes/no): ").lower()
+        if user_input != 'yes':
+            break
+
+    avg_download_speed = sum(download_speeds) / len(download_speeds)
+    avg_upload_speed = sum(upload_speeds) / len(upload_speeds)
+    avg_ping = sum(ping_values) / len(ping_values)
+
+    max_download_iteration = download_speeds.index(max(download_speeds)) + 1
+    max_upload_iteration = upload_speeds.index(max(upload_speeds)) + 1
+    min_ping_iteration = ping_values.index(min(ping_values)) + 1
+
+    # Storing results
+    df = pd.DataFrame({
+        'Iteration': [i for i in range(1, len(download_speeds) + 1)],
+        'Download Speed': download_speeds,
+        'Upload Speed': upload_speeds,
+        'Ping': ping_values
+    })
+
+    print("\nSummary:")
+    print(f"  Average Download Speed: {avg_download_speed:.2f} Mbit/s")
+    print(f"  Average Upload Speed: {avg_upload_speed:.2f} Mbit/s")
+    print(f"  Average Ping: {avg_ping:.2f} ms")
+    print(f"  Iteration with Highest Download Speed: Iteration {max_download_iteration}")
+    print(f"  Iteration with Highest Upload Speed: Iteration {max_upload_iteration}")
+    print(f"  Iteration with Lowest Ping: Iteration {min_ping_iteration}")
+
+    # Plot the results with integer x-axis labels
+    plt.plot(df['Iteration'], df['Download Speed'], label='Download Speed')
+    plt.plot(df['Iteration'], df['Upload Speed'], label='Upload Speed')
+    plt.plot(df['Iteration'], df['Ping'], label='Ping')
+    plt.xlabel('Iteration Number')
+    plt.ylabel('Speed/Ping (Mbit/s)')
+    plt.title('Internet Speed Test Results')
+    plt.legend()
+    plt.show()
 
 
 if __name__ == "__main__":
-    main()
+    run_speed_test()

scripts/Internet Speed Checker/requirements.txt

@@ -1 +1,2 @@
 speedtest-cli==2.1.3
+matplotlib