feat: email sending script added for process kill

Author: codeperfectplus

README.md

@@ -56,6 +56,7 @@ chmod +x install_miniconda.sh && sudo ./install_miniconda.sh
 - Responsive design that works on mobile, tablet, and desktop.
 - 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.
 
 
 ## Product Screenshots 📸

requirements.txt

@@ -3,4 +3,5 @@ psutil==5.9.5
 gunicorn
 flask-sqlalchemy
 flask-login
-flask-session
+flask-session
+python-dotenv

src/models.py

@@ -18,7 +18,6 @@ def __repr__(self):
             f"<SpeedTestResult {self.download_speed}, {self.upload_speed}, {self.ping}>"
         )
 
-
 class DashboardSettings(db.Model):
     __tablename__ = "DashboardSettings"
     id = db.Column(db.Integer, primary_key=True)
@@ -84,10 +83,18 @@ def __repr__(self):
 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')
 
 
+class EmailPassword(db.Model):
+    __tablename__ = "EmailPassword"
+    id = db.Column(db.Integer, primary_key=True)
+    email = db.Column(db.String(150), unique=True, nullable=False)
+    password = db.Column(db.String(150), nullable=False)
+
+
 with app.app_context():
     print("Creating tables")
     db.create_all()
@@ -101,14 +108,17 @@ class User(db.Model, UserMixin):
     # 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', password=hashed_password, user_level='admin')
+        admin_user = User(username='admin', email="codeperfectplus@gmail.com", password=hashed_password, user_level='admin')
+        
         db.session.add(admin_user)
         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', password=hashed_password, user_level='user')
+        user = User(username='user', email="test@mail.com",
+                    password=hashed_password, user_level='user')
+        
         db.session.add(user)
         db.session.commit()
 

src/routes/auth.py

@@ -2,9 +2,10 @@
 from flask_login import LoginManager, UserMixin, login_user, login_required, logout_user, current_user
 from flask_sqlalchemy import SQLAlchemy
 from werkzeug.security import generate_password_hash, check_password_hash
+from src.scripts.email_me import send_email
 
 from src.config import app, db
-from src.models import User
+from src.models import User, EmailPassword
 
 auth_bp = blueprints.Blueprint('auth', __name__)
 
@@ -28,7 +29,7 @@ def login():
         if user and check_password_hash(user.password, password):
             login_user(user)
             return redirect(url_for('dashboard'))
-        flash('Invalid username or password')
+        flash('Invalid username or password', 'danger')
     return render_template('login.html')
 
 @app.route('/signup', methods=['GET', 'POST'])
@@ -88,3 +89,68 @@ def logout():
 #             return redirect(url_for('login'))
 #         return f(*args, **kwargs)
 #     return decorated_function
+
+@app.route('/user/<username>', methods=['GET', 'POST'])
+@login_required
+def change_user_settings(username):
+    user = User.query.filter_by(username=username).first_or_404()
+
+    if request.method == 'POST':
+        new_username = request.form['username']
+        new_user_level = request.form['user_level']
+        
+        # Update user details
+        user.username = new_username
+        user.user_level = new_user_level
+        db.session.commit()
+        
+        flash('User settings updated successfully!', 'success')
+        return redirect(url_for('change_user_settings', username=user.username))
+
+    return render_template('change_user.html', user=user)
+
+
+@app.route("/update-email-password", methods=["GET", "POST"])
+@login_required
+def update_email_password():
+    email_password = EmailPassword.query.first()
+
+    if request.method == "POST":
+        new_email = request.form.get("email")
+        new_password = request.form.get("password")
+
+        if new_email:
+            email_password.email = new_email
+        if new_password:
+            email_password.password = new_password
+
+        db.session.commit()
+        flash("Email and password updated successfully!", "success")
+        return redirect(url_for("update_email_password"))
+
+    return render_template("update_email_password.html", email_password=email_password)
+
+@app.route("/send_email", methods=["GET", "POST"])
+@login_required
+def send_email_page():
+    if request.method == "POST":
+        receiver_email = request.form.get("receiver_email")
+        subject = request.form.get("subject")
+        body = request.form.get("body")
+        attachment = request.files.get("attachment")
+
+        # Save attachment if any
+        attachment_path = None
+        if attachment:
+            attachment_path = f"/tmp/{attachment.filename}"
+            attachment.save(attachment_path)
+
+        try:
+            send_email(receiver_email, subject, body, attachment_path)
+            flash("Email sent successfully!", "success")
+        except Exception as e:
+            flash(f"Failed to send email: {str(e)}", "danger")
+        
+        return redirect(url_for('send_email_page'))
+
+    return render_template("send_email.html")

src/routes/cpu_info.py

@@ -1,13 +1,16 @@
 import psutil
 from flask import render_template, blueprints, flash
 
+from flask_login import login_required
+
 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")
+@login_required
 def cpu_usage():
     settings = DashboardSettings.query.first()
     if not settings.is_cpu_info_enabled:

src/routes/disk_info.py

@@ -1,14 +1,15 @@
 import psutil
 
 from flask import render_template, blueprints, flash
-
+from flask_login import login_required
 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")
+@login_required
 def disk_usage():
     settings = DashboardSettings.query.first()
     if not settings.is_disk_info_enabled:

src/routes/homepage.py

@@ -1,5 +1,6 @@
 import datetime
-from flask import render_template, blueprints, session
+from flask import render_template, blueprints
+from flask_login import login_required
 
 from src.config import app
 from src.models import SpeedTestResult, DashboardSettings, SystemInfo
@@ -8,6 +9,7 @@
 homepage_bp = blueprints.Blueprint("homepage", __name__)
 
 @app.route("/")
+@login_required
 def dashboard():
     settings = DashboardSettings.query.first()
     SPEEDTEST_COOLDOWN_IN_HOURS = settings.speedtest_cooldown

src/routes/memory_info.py

@@ -1,5 +1,6 @@
 import psutil
 from flask import render_template, blueprints, flash
+from flask_login import login_required
 
 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
@@ -9,6 +10,7 @@
 
 
 @app.route("/memory_usage")
+@login_required
 def memory_usage():
     settings = DashboardSettings.query.first()
     if not settings.is_memory_info_enabled:

src/routes/network_info.py

@@ -1,13 +1,14 @@
 import psutil
 from flask import render_template, blueprints, flash
-
+from flask_login import login_required
 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")
+@login_required
 def network_stats():
     settings = DashboardSettings.query.first()
     if not settings.is_network_info_enabled:

src/routes/process.py

@@ -4,6 +4,7 @@
 from src.config import app
 from src.utils import get_top_processes
 from src.models import DashboardSettings
+from src.scripts.email_me import send_email
 
 process_bp = blueprints.Blueprint("process", __name__)
 
@@ -31,6 +32,10 @@ def process():
             try:
                 os.kill(int(pid_to_kill), 9)  # Sends a SIGKILL signal
                 flash(f"Process '{process_name}' (PID {pid_to_kill}) killed successfully.", "success")
+                receiver_email = current_user.email
+                subject = f"Process '{process_name}' (PID {pid_to_kill}) killed successfully."
+                body = f"Process '{process_name}' (PID {pid_to_kill}) was killed successfully by {current_user.username}."
+                send_email(receiver_email, subject, body)
             except Exception as e:
                 flash(f"Failed to kill process '{process_name}' (PID {pid_to_kill}). Error: {e}", "danger")
             return redirect(url_for("process"))  # Refresh the page after killing process