Merge pull request #3 from codeperfectplus/features/ui_design

Features/UI design

Author: codeperfectplus

README.md

@@ -56,6 +56,25 @@ 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.
+- Different email alerts for different actions.
+- Different Dashboards for different users.(Deveoper, Admin, IT Manager, Manager)
+
+## Email Feature 📧
+
+| Email Alert | Is implemented | who will get the email |
+| ----------- | -------------- | ---------------------- |
+| Process Killed | Yes | Logged User |
+| Login | Yes | Admin User and Logged User |
+| Logout | Yes | Logged User |
+| Signup | Yes | Admin User & signed up User |
+| Create User | Yes | Admin User & Created User |
+| Delete User | No | Admin User & Deleted User |
+| Speed Test | Yes | Logged User |
+| Server Up | Yes | Admin User |
+| Notification Settings Change | Yes | Admin User |
+| Signup | Yes(few changes required) | Admin User & Logged User |
+
 
 
 ## Product Screenshots 📸

app.py

@@ -1,8 +1,12 @@
-from src.config import app, db
+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.models import UserProfile
+from src.scripts.email_me import send_smpt_email
 
 def register_routes():
-    app.register_blueprint(routes.homepage_bp)
+    app.register_blueprint(routes.dashboard_bp)
     app.register_blueprint(routes.settings_bp)
     app.register_blueprint(routes.system_health_bp)
     app.register_blueprint(routes.cpu_info_bp)
@@ -12,9 +16,31 @@ def register_routes():
     app.register_blueprint(routes.speedtest_bp)
     app.register_blueprint(routes.process_bp)
 
+app.config['is_server_up_email_sent'] = False
+
+def server_up_email():
+    with app.app_context():
+        admin_emails = [user.email for user in UserProfile.query.filter_by(user_level="admin", receive_email_alerts=True).all()]
+        if admin_emails:
+            subject = "SystemGuard Server Started"
+            context = {
+                "current_time": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+                "cpu_usage": cpu_usage_percent(),
+                "memory_usage": get_memory_percent(),
+                "app_memory_usage": get_flask_memory_usage(),
+
+            }
+            html_body = render_template_from_file("src/templates/email_templates/server_up.html", **context)
+            # send_smpt_email(admin_emails, subject, html_body, is_html=True)
+            print("Server up email sent to", admin_emails)
+
 
 if __name__ == "__main__":
     register_routes()
+    # TODO: fix this email alert sent twice
+    # if not app.config['is_server_up_email_sent']:
+    #     app.config['is_server_up_email_sent'] = True
+    #     server_up_email()
 
     # # Start the memory-consuming program in a separate thread
     # memory_thread = threading.Thread(target=memory_consuming_program, daemon=True)

requirements.txt

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

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/config.py

@@ -10,3 +10,7 @@
 
 # Initialize the database
 db = SQLAlchemy(app)
+
+@app.cli.command("run")
+def server_start():
+    print("Server started")

src/models.py

@@ -0,0 +1,74 @@
+from src.config import db, app
+from src.models.user_card_settings import UserCardSettings
+from src.models.dashboard_netowrk import DashboardNetworkSettings
+from src.models.user_dashboard_settings import UserDashboardSettings
+from src.models.feature_toggle_settings import FeatureToggleSettings
+from src.models.application_general_settings import ApplicationGeneralSettings
+from src.models.smtp_configuration import SMTPSettings
+from src.models.network_speed_test_result import NetworkSpeedTestResult
+from src.models.system_information import SystemInformation
+from src.models.user_profile import UserProfile
+from flask_login import current_user
+from werkzeug.security import generate_password_hash
+import json
+
+
+
+with app.app_context():
+    print("Creating tables")
+    db.create_all()
+
+    # initialize default dashboard user_dashboard_settings for users
+    users = UserProfile.query.all()
+    for user in users:
+        if not user.dashboard_settings:
+            db.session.add(UserDashboardSettings(user_id=user.id))
+            db.session.add(UserCardSettings(user_id=user.id))
+            db.session.add(FeatureToggleSettings(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 UserProfile.query.filter_by(user_level=user["user_level"]).first():
+            hashed_password = generate_password_hash(user["password"])
+            user = UserProfile(
+                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()
+
+    # Initialize default user_dashboard_settings
+    general_settings = ApplicationGeneralSettings.query.first()
+    if not general_settings:
+        db.session.add(ApplicationGeneralSettings())
+        db.session.commit()
+
+
+# ibject for all templates
+@app.context_processor
+def inject_settings():
+    if current_user.is_anonymous:
+        return dict(user_dashboard_settings=None, card_settings=None)
+    general_settings = ApplicationGeneralSettings.query.first()
+    card_settings = UserCardSettings.query.filter_by(user_id=current_user.id).first()
+    user_dashboard_settings = UserDashboardSettings.query.filter_by(
+        user_id=current_user.id
+    ).first()  # Retrieve user-specific user_dashboard_settings from DB
+    feature_toggles_settings = FeatureToggleSettings.query.filter_by(
+        user_id=current_user.id
+    ).first()
+    all_settings = dict(
+        user_dashboard_settings=user_dashboard_settings,
+        general_settings=general_settings,
+        card_settings=card_settings,
+        feature_toggles_settings=feature_toggles_settings,
+    )
+    return all_settings

src/models/__init__.py

@@ -0,0 +1,10 @@
+from src.config import db
+
+class ApplicationGeneralSettings(db.Model):
+    __tablename__ = 'general_settings'
+    
+    id = db.Column(db.Integer, primary_key=True)
+    enable_alerts = db.Column(db.Boolean, default=True)
+    timezone = db.Column(db.String(50), default='UTC')
+    enable_cache = db.Column(db.Boolean, default=False)
+

src/models/application_general_settings.py

@@ -0,0 +1,14 @@
+from src.config import db
+
+
+class DashboardNetworkSettings(db.Model):
+    __tablename__ = "DashboardGroup"
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(50), unique=True, nullable=False)
+    description = db.Column(db.String(150), nullable=True)
+    ip_address = db.Column(db.String(50), nullable=True)
+    port = db.Column(db.Integer, nullable=True)
+    link = db.Column(db.String(150), nullable=True)
+
+    def __repr__(self):
+        return f"<DashboardGroup {self.name}, {self.description}>"

src/models/dashboard_netowrk.py

@@ -0,0 +1,14 @@
+from src.config import db
+
+class FeatureToggleSettings(db.Model):
+    __tablename__ = 'feature_toggle_settings'
+    
+    id = db.Column(db.Integer, primary_key=True)
+    user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
+    
+    # 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=True)