feat: Add different email alerts and dashboards for different users

- Implement different email alerts for different actions
- Create separate dashboards for different user roles (Developer, Admin, IT Manager, Manager)
- Update relevant templates to reflect the changes

Author: codeperfectplus

README.md

@@ -57,6 +57,8 @@ chmod +x install_miniconda.sh && sudo ./install_miniconda.sh
 - Update itself to the latest version.
 - Easy download and installation using a bash script.
 - Logged user and admin user will get the notification if the user kill some process manully on dashbaord.
+- Different email alerts for different actions.
+- Different Dashboards for different users.(Deveoper, Admin, IT Manager, Manager)
 
 ## Email Feature 📧
 

src/assets/predefine_user.json

@@ -0,0 +1,18 @@
+[
+    {
+        "username": "admin",
+        "email": "codeperfectplus@gmail.com",
+        "password": "admin",
+        "user_level": "admin",
+        "profession": "developer",
+        "receive_email_alerts": true
+    },
+    {
+        "username": "manager",
+        "email": "dr47806@gmail.com",
+        "password": "manager",
+        "user_level": "user",
+        "profession": "manager",
+        "receive_email_alerts": true
+    }
+]

src/models.py

@@ -1,9 +1,10 @@
+import json
 import datetime
-from flask_login import UserMixin
+from flask_login import UserMixin, current_user
 from werkzeug.security import generate_password_hash, check_password_hash
 
-from src.config import db, app
 
+from src.config import db, app
 
 class SpeedTestResult(db.Model):
     __tablename__ = "SpeedTestResult"
@@ -18,28 +19,37 @@ def __repr__(self):
             f"<SpeedTestResult {self.download_speed}, {self.upload_speed}, {self.ping}>"
         )
 
-class DashboardSettings(db.Model):
-    __tablename__ = "DashboardSettings"
+class User(db.Model, UserMixin):
+    __tablename__ = 'users'
+    
     id = db.Column(db.Integer, primary_key=True)
+    username = db.Column(db.String(50), unique=True, nullable=False)
+    email = db.Column(db.String(100), unique=True, nullable=False)
+    password = db.Column(db.String(100), nullable=False)
+    user_level = db.Column(db.String(10), nullable=False, default='user')
+    receive_email_alerts = db.Column(db.Boolean, default=False)
+    profession = db.Column(db.String(50), nullable=True)
 
-    # speedtest setting
-    speedtest_cooldown = db.Column(db.Integer, default=1)
-    number_of_speedtests = db.Column(db.Integer, default=1)
+    # Backref renamed to avoid conflict
+    dashboard_settings = db.relationship('DashboardSettings', backref='user', uselist=False)
 
-    # general settings
-    timezone = db.Column(db.String(50), default="Asia/Kolkata")
-    enable_cache = db.Column(db.Boolean, default=True)
-    enable_alerts = db.Column(db.Boolean, default=False)
+class DashboardSettings(db.Model):
+    __tablename__ = 'dashboard_settings'
 
-    # page enable/disable
+    id = db.Column(db.Integer, primary_key=True)
+    user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
+    
+    speedtest_cooldown = db.Column(db.Integer, default=3600)
+    number_of_speedtests = db.Column(db.Integer, default=3)
+    
+    # Feature Toggles
     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=False)
+    is_process_info_enabled = db.Column(db.Boolean, default=True)
 
-
-    # card enable/disable
+    # Card Toggles
     is_user_card_enabled = db.Column(db.Boolean, default=True)
     is_server_card_enabled = db.Column(db.Boolean, default=True)
     is_battery_card_enabled = db.Column(db.Boolean, default=True)
@@ -51,10 +61,15 @@ class DashboardSettings(db.Model):
     is_disk_usage_card_enabled = db.Column(db.Boolean, default=True)
     is_system_uptime_card_enabled = db.Column(db.Boolean, default=True)
     is_network_statistic_card_enabled = db.Column(db.Boolean, default=True)
-    is_speedtest_enabled = db.Column(db.Boolean, default=False)
-
-    def __repr__(self):
-        return f"<DashboardSettings {self.speedtest_cooldown}, {self.timezone}, {self.number_of_speedtests}>"
+    is_speedtest_enabled = db.Column(db.Boolean, default=True)
+    
+class GeneralSettings(db.Model):
+    __tablename__ = 'general_settings'
+    
+    id = db.Column(db.Integer, primary_key=True)
+    enable_alerts = db.Column(db.Boolean, default=False)
+    timezone = db.Column(db.String(50), default='UTC')
+    enable_cache = db.Column(db.Boolean, default=False)
 
 
 class SystemInfo(db.Model):
@@ -81,13 +96,6 @@ class SystemInfo(db.Model):
     def __repr__(self):
         return f"<SystemInfo {self.username}, {self.cpu_percent}, {self.memory_percent}, {self.disk_usage}, {self.battery_percent}, {self.cpu_core}, {self.boot_time}, {self.network_sent}, {self.network_received}, {self.process_count}, {self.swap_memory}, {self.uptime}, {self.ipv4_connections}, {self.ipv6_connections}, {self.dashboard_memory_usage}>"
 
-class User(db.Model, UserMixin):
-    id = db.Column(db.Integer, primary_key=True)
-    username = db.Column(db.String(150), unique=True, nullable=False)
-    email = db.Column(db.String(150), unique=True, nullable=False)
-    password = db.Column(db.String(150), nullable=False)
-    user_level = db.Column(db.String(50), nullable=False, default='user')
-    receive_email_alerts = db.Column(db.Boolean, default=True)
 
 
 class SmptEamilPasswordConfig(db.Model):
@@ -102,32 +110,37 @@ class SmptEamilPasswordConfig(db.Model):
     db.create_all()
 
     # Initialize default settings
-    settings = DashboardSettings.query.first()
-    if not settings:
-        db.session.add(DashboardSettings())
-        db.session.commit()
-
-    # Create admin user if not exists
-    if not User.query.filter_by(username='admin').first():
-        hashed_password = generate_password_hash('adminpassword')
-        admin_user = User(username='admin', email="codeperfectplus@gmail.com", password=hashed_password, user_level='admin',
-                          receive_email_alerts=True)
-        
-        db.session.add(admin_user)
+    general_settings = GeneralSettings.query.first()
+    if not general_settings:
+        db.session.add(GeneralSettings())
         db.session.commit()
 
-    # create a user if not exists
-    if not User.query.filter_by(username='user').first():
-        hashed_password = generate_password_hash('userpassword')
-        user = User(username='user', email="test@mail.com",
-                    password=hashed_password, user_level='user', receive_email_alerts=False)
-        
-        db.session.add(user)
-        db.session.commit()
+    # initialize default dashboard settings for users
+    users = User.query.all()
+    for user in users:
+        if not user.dashboard_settings:
+            db.session.add(DashboardSettings(user_id=user.id))
+            db.session.commit()
+
+
+    pre_defined_users_json = "src/assets/predefine_user.json"
+    with open(pre_defined_users_json, "r") as file:
+        pre_defined_users = json.load(file)
+    for user in pre_defined_users:
+        if not User.query.filter_by(user_level=user['user_level']).first():
+            hashed_password = generate_password_hash(user['password'])
+            user = User(username=user['username'], email=user['email'], password=hashed_password, user_level=user['user_level'],
+                        receive_email_alerts=user['receive_email_alerts'], profession=user['profession'])
+            
+            db.session.add(user)
+            db.session.commit()
 
 # ibject for all templates
 @app.context_processor
 def inject_settings():
-    settings = DashboardSettings.query.first()
-    return dict(settings=settings)
-
+    if current_user.is_anonymous:
+        return dict(settings=None)
+    settings = DashboardSettings.query.filter_by(user_id=current_user.id).first()  # Retrieve user-specific settings from DB
+    general_settings = GeneralSettings.query.first()
+    all_settings = dict(settings=settings, general_settings=general_settings)
+    return all_settings

src/routes/auth.py

@@ -47,19 +47,6 @@ def get_email_addresses(user_level=None, receive_email_alerts=True, fetch_all_us
         # Return list of email addresses
         return [user.email for user in users]
 
-# Get Admin Emails with Alerts Enabled:
-# admin_emails = get_email_addresses(user_level='admin', receive_email_alerts=True)
-# Get All Admin Emails Regardless of Alert Preference:
-# all_admin_emails = get_email_addresses(user_level='admin', fetch_all_users=True)
-
-# Get All Users with Alerts Enabled:
-# all_user_emails = get_email_addresses(receive_email_alerts=True)
-# Get All Users Regardless of Alert Preference:
-# all_users_emails = get_email_addresses(fetch_all_users=True)
-
-
-
-
 @app.route('/login', methods=['GET', 'POST'])
 def login():
     if request.method == 'POST':
@@ -109,6 +96,7 @@ def signup():
         confirm_password = request.form['confirm_password']
         user_level = request.form.get('user_level', 'user')  # Default to 'user' if not provided
         receive_email_alerts = 'receive_email_alerts' in request.form  # Checkbox is either checked or not
+        profession = request.form.get('profession', None)
 
         if password != confirm_password:
             flash('Passwords do not match')
@@ -120,7 +108,12 @@ def signup():
             return redirect(url_for('signup'))
 
         hashed_password = generate_password_hash(password)
-        new_user = User(username=username, email=email, password=hashed_password, user_level=user_level, receive_email_alerts=receive_email_alerts)
+        new_user = User(username=username, 
+                        email=email, 
+                        password=hashed_password, 
+                        user_level=user_level, 
+                        receive_email_alerts=receive_email_alerts,
+                        profession=profession)
 
         # Get Admin Emails with Alerts Enabled:
         admin_email_address = get_email_addresses(user_level='admin', receive_email_alerts=True)
@@ -202,8 +195,23 @@ def delete_user(username):
         return redirect(url_for('view_users'))  # Redirect to the users page
 
     user = User.query.filter_by(username=username).first_or_404()
+
+    # Get Admin Emails with Alerts Enabled:
+    admin_email_address = get_email_addresses(user_level='admin', receive_email_alerts=True)
+    if admin_email_address:
+        subject = "User Deletion Alert"
+        context = {
+            "username": 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)
+        send_smpt_email(admin_email_address, subject, html_body, is_html=True)
+
     db.session.delete(user)
     db.session.commit()
+
+            
 
     flash(f'User {username} has been deleted successfully!', 'success')
     return redirect(url_for('view_users'))

src/routes/homepage.py

@@ -1,6 +1,6 @@
 import datetime
 from flask import render_template, blueprints
-from flask_login import login_required
+from flask_login import login_required, current_user
 
 from src.config import app
 from src.models import SpeedTestResult, DashboardSettings, SystemInfo
@@ -46,13 +46,47 @@ def dashboard():
         source = None
         show_prompt = True
         remaining_time_for_next_test = None
-
+    
     return render_template(
-        "dashboard.html",
-        system_info=system_info,
-        speedtest_result=speedtest_result,
-        source=source,
-        last_timestamp=last_timestamp,
-        next_test_time=remaining_time_for_next_test,
-        show_prompt=show_prompt,
-    )
+            "dashboard/developer.html",
+            system_info=system_info,
+            speedtest_result=speedtest_result,
+            source=source,
+            last_timestamp=last_timestamp,
+            next_test_time=remaining_time_for_next_test,
+            show_prompt=show_prompt,
+            current_user=current_user,
+        )
+    # if current_user.profession == "developer":
+    #     return render_template(
+    #         "dashboard/developer.html",
+    #         system_info=system_info,
+    #         speedtest_result=speedtest_result,
+    #         source=source,
+    #         last_timestamp=last_timestamp,
+    #         next_test_time=remaining_time_for_next_test,
+    #         show_prompt=show_prompt,
+    #         current_user=current_user,
+    #     )
+    # elif current_user.profession == "manager":
+    #     return render_template(
+    #         "dashboard/manager.html",
+    #         system_info=system_info,
+    #         speedtest_result=speedtest_result,
+    #         source=source,
+    #         last_timestamp=last_timestamp,
+    #         next_test_time=remaining_time_for_next_test,
+    #         show_prompt=show_prompt,
+    #         current_user=current_user,
+    #     )
+    # else:
+    #     return render_template(
+    #         "dashboard/developer.html",
+    #         system_info=system_info,
+    #         speedtest_result=speedtest_result,
+    #         source=source,
+    #         last_timestamp=last_timestamp,
+    #         next_test_time=remaining_time_for_next_test,
+    #         show_prompt=show_prompt,
+    #         current_user=current_user,
+    #     )