feat: Update models and routes for SMTP configuration, application general settings, user dashboard settings, dashboard network settings, network speed test result, and feature toggle settings

Author: codeperfectplus

app.py

@@ -2,7 +2,7 @@
 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 User
+from src.models import UserProfile
 from src.scripts.email_me import send_smpt_email
 
 def register_routes():
@@ -20,7 +20,7 @@ def register_routes():
 
 def server_up_email():
     with app.app_context():
-        admin_emails = [user.email for user in User.query.filter_by(user_level="admin", receive_email_alerts=True).all()]
+        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 = {

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 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 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(settings=None, card_settings=None)
+    general_settings = ApplicationGeneralSettings.query.first()
+    card_settings = UserCardSettings.query.filter_by(user_id=current_user.id).first()
+    settings = UserDashboardSettings.query.filter_by(
+        user_id=current_user.id
+    ).first()  # Retrieve user-specific settings from DB
+    feature_toggles_settings = FeatureToggleSettings.query.filter_by(
+        user_id=current_user.id
+    ).first()
+    all_settings = dict(
+        settings=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)

src/models/feature_toggle_settings.py

@@ -0,0 +1,16 @@
+import datetime
+
+from src.config import db
+
+class NetworkSpeedTestResult(db.Model):
+    __tablename__ = "network_speed_test_result"
+    id = db.Column(db.Integer, primary_key=True)
+    download_speed = db.Column(db.String(50))
+    upload_speed = db.Column(db.String(50))
+    ping = db.Column(db.String(50))
+    timestamp = db.Column(db.DateTime, default=datetime.datetime.now())
+
+    def __repr__(self):
+        return (
+            f"<SpeedTestResult {self.download_speed}, {self.upload_speed}, {self.ping}>"
+        )

src/models/network_speed_test_result.py

@@ -0,0 +1,8 @@
+from src.config import db
+
+
+class SMTPSettings(db.Model):
+    __tablename__ = "smtp_settings"
+    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)

src/models/smtp_configuration.py

@@ -0,0 +1,28 @@
+import datetime
+
+from src.config import db
+
+
+class SystemInformation(db.Model):
+    __tablename__ = "system_information"
+    id = db.Column(db.Integer, primary_key=True)
+    username = db.Column(db.String(50))
+    cpu_percent = db.Column(db.Float)
+    memory_percent = db.Column(db.Float)
+    disk_usage = db.Column(db.Float)
+    battery_percent = db.Column(db.Float)
+    cpu_core = db.Column(db.Integer)
+    boot_time = db.Column(db.String(50))
+    network_sent = db.Column(db.Float)
+    network_received = db.Column(db.Float)
+    process_count = db.Column(db.Integer)
+    swap_memory = db.Column(db.Float)
+    uptime = db.Column(db.String(50))
+    ipv4_connections = db.Column(db.String(50))
+    dashboard_memory_usage = db.Column(db.String(50))
+    timestamp = db.Column(db.DateTime, default=datetime.datetime.now())
+    cpu_frequency = db.Column(db.String(50))
+    current_temp = db.Column(db.String(50))
+
+    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}>"

src/models/system_information.py

@@ -0,0 +1,22 @@
+from src.config import db
+
+
+class UserCardSettings(db.Model):
+    __tablename__ = "user_card_settings"
+
+    id = db.Column(db.Integer, primary_key=True)
+    user_id = db.Column(db.Integer, db.ForeignKey("users.id"))
+
+    # 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)
+    is_cpu_core_card_enabled = db.Column(db.Boolean, default=True)
+    is_cpu_usage_card_enabled = db.Column(db.Boolean, default=True)
+    is_cpu_temp_card_enabled = db.Column(db.Boolean, default=True)
+    is_dashboard_memory_card_enabled = db.Column(db.Boolean, default=True)
+    is_memory_usage_card_enabled = db.Column(db.Boolean, default=True)
+    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=True)