feat: enable cache and other buttons added in settings

Author: codeperfectplus

src/models.py

@@ -25,6 +25,13 @@ class DashboardSettings(db.Model):
     speedtest_cooldown = db.Column(db.Integer, default=1)
     number_of_speedtests = db.Column(db.Integer, default=1)
     timezone = db.Column(db.String(50), default="Asia/Kolkata")
+    is_cpu_info_enabled = db.Column(db.Boolean, default=True)
+    is_memory_info_enabled = db.Column(db.Boolean, default=True)
+    is_disk_info_enabled = db.Column(db.Boolean, default=True)
+    is_network_info_enabled = db.Column(db.Boolean, default=True)
+    is_process_info_enabled = db.Column(db.Boolean, default=True)
+    is_speedtest_enabled = db.Column(db.Boolean, default=True)
+    enable_cache = db.Column(db.Boolean, default=True)
 
     def __repr__(self):
         return f"<DashboardSettings {self.speedtest_cooldown}, {self.timezone}, {self.number_of_speedtests}>"
@@ -76,4 +83,10 @@ class User(db.Model, UserMixin):
         hashed_password = generate_password_hash('adminpassword')
         admin_user = User(username='admin', password=hashed_password, user_level='admin')
         db.session.add(admin_user)
-        db.session.commit()
+        db.session.commit()
+
+# ibject for all templates
+@app.context_processor
+def inject_settings():
+    settings = DashboardSettings.query.first()
+    return dict(settings=settings)

src/routes/cpu_info.py

@@ -3,11 +3,16 @@
 
 from src.config import app
 from src.utils import get_cpu_core_count, get_cpu_frequency, cpu_usage_percent, get_cpu_temp, get_cached_value
+from src.models import DashboardSettings
 
 cpu_info_bp = blueprints.Blueprint("cpu_usage", __name__)
 
 @app.route("/cpu_usage")
 def cpu_usage():
+    settings = DashboardSettings.query.first()
+    if not settings.is_cpu_info_enabled:
+        return render_template("error/404.html")
+    
     current_temp, high_temp, critical_temp = get_cpu_temp()
     cpu_core = get_cached_value("cpu_core", get_cpu_core_count)
     system_info = {

src/routes/disk_info.py

@@ -4,11 +4,15 @@
 
 from src.config import app
 from src.utils import get_cached_value, get_disk_free, get_disk_total, get_disk_used, get_disk_usage_percent
+from src.models import DashboardSettings
 
 disk_info_bp = blueprints.Blueprint("disk_usage", __name__)
 
 @app.route("/disk_usage")
 def disk_usage():
+    settings = DashboardSettings.query.first()
+    if not settings.is_disk_info_enabled:
+        return render_template("error/404.html")
     disk_total = get_cached_value("disk_total", get_disk_total)
     system_info = {
         "disk_usage": get_disk_usage_percent(),

src/routes/memory_info.py

@@ -3,13 +3,16 @@
 
 from src.config import app
 from src.utils import get_cached_value, get_memory_percent, get_memory_available, get_memory_used, get_swap_memory_info
+from src.models import DashboardSettings
 
 memory_info_bp = blueprints.Blueprint("memory_usage", __name__)
 
 
 @app.route("/memory_usage")
 def memory_usage():
-    
+    settings = DashboardSettings.query.first()
+    if not settings.is_memory_info_enabled:
+        return render_template("error/404.html")
     memory_available = get_cached_value("memory_available", get_memory_available) 
     system_info = {
         "memory_percent": get_memory_percent(),

src/routes/network_info.py

@@ -3,10 +3,15 @@
 
 from src.config import app
 from src.utils import get_established_connections
+from src.models import DashboardSettings
+
 network_info_bp = blueprints.Blueprint("network_stats", __name__)
 
 @app.route("/network_stats")
 def network_stats():
+    settings = DashboardSettings.query.first()
+    if not settings.is_network_info_enabled:
+        return render_template("error/404.html")
     net_io = psutil.net_io_counters()
     ipv4_ip, ipv6_ip = get_established_connections()
     system_info = {

src/routes/process.py

@@ -3,12 +3,16 @@
 from flask_login import login_required, current_user
 from src.config import app
 from src.utils import get_top_processes
+from src.models import DashboardSettings
 
 process_bp = blueprints.Blueprint("process", __name__)
 
 @app.route("/process", methods=["GET", "POST"])
 @login_required
 def process():
+    settings = DashboardSettings.query.first()
+    if not settings.is_process_info_enabled:
+        return render_template("error/404.html")
     if current_user.user_level != "admin":
         flash("You do not have permission to view this page.", "danger")
         return redirect(url_for("dashboard"))

src/routes/settings.py

@@ -1,20 +1,29 @@
 from flask import render_template, request, flash, blueprints
-
 from src.config import app, db
 from src.models import DashboardSettings
 
 settings_bp = blueprints.Blueprint("settings", __name__)
 
 @app.route("/settings", methods=["GET", "POST"])
 def settings():
-    # Fetch the settings from the database and update them
+    # Fetch the settings from the database
     settings = DashboardSettings.query.first()
     if settings:
         if request.method == "POST":
+            # Update settings from the form inputs
             settings.speedtest_cooldown = int(request.form["speedtest_cooldown"])
             settings.number_of_speedtests = int(request.form["number_of_speedtests"])
             settings.timezone = request.form["timezone"]
+            settings.is_cpu_info_enabled = "is_cpu_info_enabled" in request.form
+            settings.is_memory_info_enabled = "is_memory_info_enabled" in request.form
+            settings.is_disk_info_enabled = "is_disk_info_enabled" in request.form
+            settings.is_network_info_enabled = "is_network_info_enabled" in request.form
+            settings.is_process_info_enabled = "is_process_info_enabled" in request.form
+            settings.is_speedtest_enabled = "is_speedtest_enabled" in request.form
+            settings.enable_cache = "enable_cache" in request.form
+
+            # Commit the changes to the database
             db.session.commit()
             flash("Settings updated successfully!", "success")
+        
         return render_template("settings.html", settings=settings)
-

src/scripts/locustfile.py

@@ -1,3 +1,5 @@
+import os
+
 from locust import HttpUser, TaskSet, task, between
 from locust.exception import StopUser
 
@@ -44,5 +46,4 @@ class WebsiteUser(HttpUser):
     wait_time = between(1, 5)  # Time to wait between tasks
 
 if __name__ == "__main__":
-    import os
-    # os.system("locust -f locustfile.py --host=http://localhost:5000")
+    os.system("locust -f locustfile.py --host=http://localhost:5000")

src/static/css/404.css

@@ -0,0 +1,35 @@
+.container {
+    text-align: center;
+    background-color: #ffffff;
+    padding: 30px 50px;
+    border-radius: 8px;
+    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);
+}
+
+h1 {
+    font-size: 100px;
+    margin: 0;
+    color: #dc3545;
+}
+
+p {
+    font-size: 20px;
+    margin: 20px 0;
+    color: #6c757d;
+}
+
+.btn {
+    display: inline-block;
+    padding: 10px 20px;
+    font-size: 16px;
+    color: #ffffff;
+    background-color: #007bff;
+    border: none;
+    border-radius: 4px;
+    text-decoration: none;
+    transition: background-color 0.3s ease;
+}
+
+.btn:hover {
+    background-color: #0056b3;
+}

src/static/css/settings.css

@@ -1,44 +1,93 @@
-.container {
+.settings-container {
     max-width: 600px;
-    margin: auto;
-    background: white;
+    margin: 0 auto;
     padding: 20px;
+    background-color: #f8f9fa;
     border-radius: 8px;
-    box-shadow: 0 0 10px rgba(0, 0, 0, 0.1);
-    margin-bottom: 10px;
+    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);
 }
 
-h1 {
+.settings-title {
+    font-size: 28px;
+    color: #343a40;
     margin-bottom: 20px;
-    font-size: 24px;
+    text-align: center;
+}
+
+.settings-form {
+    display: flex;
+    flex-direction: column;
+    gap: 15px;
 }
 
 .form-group {
-    margin-bottom: 15px;
+    display: flex;
+    flex-direction: row;
+    align-items: center; /* Align items vertically in the center */
+    justify-content: space-between; /* Space out label and input */
+    margin-bottom: 10px;
 }
 
 .form-group label {
-    display: block;
-    margin-bottom: 5px;
-    font-weight: bold;
+    font-weight: 600;
+    color: #495057;
+    margin-right: 10px; /* Add some space between the label and checkbox */
 }
 
-.form-group input,
+.form-group input[type="checkbox"] {
+    width: 20px;
+    height: 20px;
+    cursor: pointer;
+    margin: 0; /* Remove any default margin */
+}
+
+.form-group input[type="number"],
 .form-group select {
-    width: 100%;
     padding: 8px;
-    box-sizing: border-box;
+    border: 1px solid #ced4da;
+    border-radius: 4px;
+    background-color: #ffffff;
+    font-size: 16px;
+    width: 100%;
+}
+
+@media (max-width: 576px) {
+    .form-group {
+        flex-direction: column;
+        align-items: flex-start;
+    }
+
+    .form-group label {
+        margin-bottom: 5px;
+    }
 }
 
-.btn {
+.btn-primary {
+    padding: 10px 20px;
     background-color: #007bff;
-    color: white;
-    padding: 10px 15px;
+    color: #ffffff;
     border: none;
-    border-radius: 5px;
+    border-radius: 4px;
     cursor: pointer;
+    font-size: 16px;
+    text-align: center;
+    transition: background-color 0.3s ease;
 }
 
-.btn:hover {
+.btn-primary:hover {
     background-color: #0056b3;
 }
+
+@media (max-width: 576px) {
+    .settings-container {
+        padding: 15px;
+    }
+
+    .settings-title {
+        font-size: 24px;
+    }
+
+    .btn-primary {
+        width: 100%;
+    }
+}