feat: Update file paths for pre-defined users and cache in models and utils

Author: codeperfectplus

README.md

@@ -108,6 +108,7 @@ A Docker image has not been created for this project because it requires access
 - Threshold notifications
 - Customizable dashboards
 - Plugin support to make SystemGuard even more powerful.
+- make server logs
 
 ## Acknowledgments
 

app.py

@@ -1,7 +1,7 @@
 import datetime
 from src.config import app
 from src import routes
-from src.utils import render_template_from_file, get_flask_memory_usage, cpu_usage_percent, get_memory_percent
+from src.utils import render_template_from_file, get_flask_memory_usage, cpu_usage_percent, get_memory_percent, ROOT_DIR
 from src.models import UserProfile
 from src.scripts.email_me import send_smpt_email
 
@@ -30,7 +30,8 @@ def server_up_email():
                 "app_memory_usage": get_flask_memory_usage(),
 
             }
-            html_body = render_template_from_file("src/templates/email_templates/server_up.html", **context)
+            server_up_template = os.path.join(ROOT_DIR, "src/templates/email_templates/server_up.html")
+            html_body = render_template_from_file(server_up_template, **context)
             # send_smpt_email(admin_emails, subject, html_body, is_html=True)
             print("Server up email sent to", admin_emails)
 

src/routes/auth.py

@@ -1,3 +1,4 @@
+import os
 import datetime
 from flask import render_template, redirect, url_for, request, blueprints, flash
 from flask_login import LoginManager, login_user, login_required, logout_user, current_user
@@ -6,7 +7,7 @@
 from src.scripts.email_me import send_smpt_email
 from src.config import app, db
 from src.models import UserProfile, UserCardSettings, FeatureToggleSettings, UserDashboardSettings
-from src.utils import render_template_from_file
+from src.utils import render_template_from_file, ROOT_DIR
 from src.routes.helper import get_email_addresses
 
 auth_bp = blueprints.Blueprint('auth', __name__)
@@ -37,14 +38,20 @@ def login():
                     admin_email_address.remove(receiver_email)
                 if admin_email_address:
                     context = {"username": current_user.username, "login_time": datetime.datetime.now()}
-                    login_body = render_template_from_file("src/templates/email_templates/admin_login_alert.html", **context)
-                    send_smpt_email(admin_email_address, 'Login Alert', login_body, is_html=True)
+
+                    login_alert_template = os.path.join(ROOT_DIR, "src/templates/email_templates/admin_login_alert.html")
+                    html_body = render_template_from_file(login_alert_template, **context)
+
+                    send_smpt_email(admin_email_address, 'Login Alert', html_body, is_html=True)
 
             # log in alert to user
             if receiver_email:
                 context = {"username": current_user.username, "login_time": datetime.datetime.now()}
-                login_body = render_template_from_file("src/templates/email_templates/login.html", **context)
-                send_smpt_email(receiver_email, 'Login Alert', login_body, is_html=True)
+
+                login_message_template = os.path.join(ROOT_DIR, "src/templates/email_templates/login.html") 
+                html_body = render_template_from_file(login_message_template, **context)
+                
+                send_smpt_email(receiver_email, 'Login Alert', html_body, is_html=True)
             return redirect(url_for('dashboard'))
         flash('Invalid username or password', 'danger')
     return render_template('login.html')
@@ -54,8 +61,9 @@ def logout():
     receiver_email = current_user.email
     if receiver_email:
         context = {"username": current_user.username}
-        logout_body = render_template_from_file("src/templates/email_templates/logout.html", **context)
-        send_smpt_email(receiver_email, 'Logout Alert', logout_body, is_html=True)
+        logout_message_template = os.path.join(ROOT_DIR, "src/templates/email_templates/logout.html")
+        html_body = render_template_from_file(logout_message_template, **context)
+        send_smpt_email(receiver_email, 'Logout Alert', html_body, is_html=True)
     logout_user()
     return redirect(url_for('login'))
 
@@ -97,7 +105,8 @@ def signup():
                 "signup_time": datetime.datetime.now(),
                 "user_level": new_user.user_level
             }
-            html_body = render_template_from_file("src/templates/email_templates/new_user_alert.html", **context)
+            new_user_alert_template = os.path.join(ROOT_DIR, "src/templates/email_templates/new_user_alert.html")
+            html_body = render_template_from_file(new_user_alert_template, **context)
             send_smpt_email(admin_email_address, subject, html_body, is_html=True)
 
         # Send email to the new user
@@ -106,7 +115,8 @@ def signup():
             "username": new_user.username,
             "email": new_user.email,
         }
-        html_body = render_template_from_file("src/templates/email_templates/welcome.html", **context)
+        welcome_template = os.path.join(ROOT_DIR, "src/templates/email_templates/welcome.html")
+        html_body = render_template_from_file(welcome_template, **context)
         send_smpt_email(email, subject, html_body, is_html=True)
 
         db.session.add(new_user)

src/routes/process.py

@@ -2,7 +2,7 @@
 from flask import request, render_template, redirect, url_for, flash, session, blueprints
 from flask_login import login_required, current_user
 from src.config import app
-from src.utils import get_top_processes, render_template_from_file
+from src.utils import get_top_processes, render_template_from_file, ROOT_DIR
 from src.models import  FeatureToggleSettings
 from src.scripts.email_me import send_smpt_email
 
@@ -35,7 +35,8 @@ def process():
                 receiver_email = current_user.email
                 subject = f"Process '{process_name}' (PID {pid_to_kill}) killed successfully."
                 context = {"process_name": process_name, "pid_to_kill": pid_to_kill, "username": current_user.username}
-                html_body = render_template_from_file("src/templates/email_templates/process_killed.html", **context)
+                process_killed_template = os.path.join(ROOT_DIR, "src/templates/email_templates/process_killed.html")
+                html_body = render_template_from_file(process_killed_template, **context)
                 send_smpt_email(receiver_email, subject, html_body, is_html=True)
             except Exception as e:
                 flash(f"Failed to kill process '{process_name}' (PID {pid_to_kill}). Error: {e}", "danger")

src/routes/settings.py

@@ -1,10 +1,11 @@
+import os
 import datetime
 from flask import render_template, request, flash, blueprints, redirect, url_for
 
 from src.config import app, db
 from src.models import UserCardSettings, UserDashboardSettings, UserProfile, ApplicationGeneralSettings, FeatureToggleSettings
 from flask_login import login_required, current_user
-from src.utils import render_template_from_file
+from src.utils import render_template_from_file, ROOT_DIR
 from src.scripts.email_me import send_smpt_email
 
 settings_bp = blueprints.Blueprint("settings", __name__)
@@ -37,8 +38,8 @@ def general_settings():
                 "notifications_enabled": general_settings.enable_alerts,
                 "current_user": current_user.username
             }
-            html_body = render_template_from_file("src/templates/email_templates/notification_alert.html", **context)
-            print("Notification enabled:", general_settings.enable_alerts)
+            notficiation_alert_template = os.path.join(ROOT_DIR, "src/templates/email_templates/notification_alert.html")
+            html_body = render_template_from_file(notficiation_alert_template, **context)
             send_smpt_email(admin_emails, subject, html_body, is_html=True, bypass_alerts=True)
         db.session.commit()
         flash('General settings updated successfully!', 'success')

src/routes/speedtest.py

@@ -1,9 +1,10 @@
+import os
 import datetime
 from flask import render_template, Blueprint
 from flask_login import login_required, current_user
 from src.config import app, db
 from src.models import UserDashboardSettings, NetworkSpeedTestResult
-from src.utils import run_speedtest, render_template_from_file
+from src.utils import run_speedtest, render_template_from_file, ROOT_DIR
 from src.scripts.email_me import send_smpt_email
 
 speedtest_bp = Blueprint("speedtest", __name__)
@@ -43,8 +44,11 @@ def speedtest():
             receiver_email = current_user.email
             subject = "Speedtest Result"
             context = {"speedtest_result": current_speedtest_result}
-            body = render_template_from_file("src/templates/email_templates/speedtest_result.html", **context)
-            send_smpt_email(receiver_email, subject, body, is_html=True)
+            speedtest_result_template = os.path.join(
+                ROOT_DIR, "src/templates/email_templates/speedtest_result.html"
+            )
+            html_body = render_template_from_file(speedtest_result_template, **context)
+            send_smpt_email(receiver_email, subject, html_body, is_html=True)
 
             return render_template(
                 "speedtest_result.html",

src/routes/user.py

@@ -1,11 +1,12 @@
+import os
 import datetime
 from flask import render_template, redirect, url_for, request, blueprints, flash, blueprints
 from flask_login import login_required, current_user
 from werkzeug.security import generate_password_hash
 
 from src.config import app, db
 from src.models import UserProfile, UserDashboardSettings, UserCardSettings, FeatureToggleSettings
-from src.utils import render_template_from_file
+from src.utils import render_template_from_file, ROOT_DIR
 from src.scripts.email_me import send_smpt_email
 from src.routes.helper import get_email_addresses
 
@@ -48,7 +49,8 @@ def add_user():
                 "signup_time": datetime.datetime.now(),
                 "user_level": new_user.user_level
             }
-            html_body = render_template_from_file("src/templates/email_templates/new_user_create.html", **context)
+            new_user_alert_template =  os.path.join(ROOT_DIR, "src/templates/email_templates/new_user_create.html")
+            html_body = render_template_from_file(new_user_alert_template, **context)
             send_smpt_email(admin_email_address, subject, html_body, is_html=True)
 
         # Send welcome email to new user
@@ -57,7 +59,8 @@ def add_user():
             "username": new_user.username,
             "email": new_user.email,
         }
-        html_body = render_template_from_file("src/templates/email_templates/welcome.html", **context)
+        welcome_email_template = os.path.join(ROOT_DIR, "src/templates/email_templates/welcome.html")
+        html_body = render_template_from_file(welcome_email_template, **context)
         send_smpt_email(email, subject, html_body, is_html=True)
 
         # Add and commit the new user to get the correct user ID
@@ -131,7 +134,8 @@ def delete_user(username):
             "deletion_time": datetime.datetime.now(),
             "current_user": current_user.username,
         }
-        html_body = render_template_from_file("src/templates/email_templates/deletion_email.html", **context)
+        deletion_email_template = os.path.join(ROOT_DIR, "src/templates/email_templates/deletion_email.html")
+        html_body = render_template_from_file(deletion_email_template, **context)
         send_smpt_email(admin_email_address, subject, html_body, is_html=True)
 
     db.session.delete(user)