Code formatting

Author: Adam-Color

src/core/logic/app_tracker.py

@@ -2,29 +2,39 @@
 App Tracking logic class. Used for app tracking, filtering, and listing.
 """
 
-import threading
 import os
-import psutil # type: ignore
 import sys
+import threading
 
-if os.name == 'nt':
+import psutil  # type: ignore
+
+if os.name == "nt":
     from pywinauto import Desktop
     from pywinauto.findwindows import ElementNotFoundError
-    windows = Desktop(backend="uia").windows()
-elif sys.platform == 'darwin':
-    from AppKit import NSWorkspace # type: ignore
 
-from core.utils.file_utils import read_file, write_file, apps_file, user_dir_exists, config_file
+    windows = Desktop(backend="uia").windows()
+elif sys.platform == "darwin":
+    from AppKit import NSWorkspace  # type: ignore
 
 import logging
+
+from core.utils.file_utils import (
+    apps_file,
+    config_file,
+    read_file,
+    user_dir_exists,
+    write_file,
+)
+
 logger = logging.getLogger(__name__)
 
 EXCLUDED_APP_PIDS = []
 INCLUDED_APP_PIDS = []
 
 if user_dir_exists() and os.path.exists(apps_file()):
-    EXCLUDED_APP_PIDS = read_file(apps_file())['excluded_app_pids']
-    INCLUDED_APP_PIDS = read_file(apps_file())['included_app_pids']
+    EXCLUDED_APP_PIDS = read_file(apps_file())["excluded_app_pids"]
+    INCLUDED_APP_PIDS = read_file(apps_file())["included_app_pids"]
+
 
 class AppTracker:
     def __init__(self, parent, logic_controller):
@@ -34,7 +44,7 @@ def __init__(self, parent, logic_controller):
         self.stop_event = threading.Event()  # Used to stop the thread gracefully
         self.cached_process_count = 0  # Tracks the last known process count
         try:
-            self.is_filter_enabled = read_file(config_file())['is_filter_enabled']
+            self.is_filter_enabled = read_file(config_file())["is_filter_enabled"]
         except (KeyError, FileNotFoundError):
             self.is_filter_enabled = True
 
@@ -48,17 +58,19 @@ def __init__(self, parent, logic_controller):
 
     def _start_tracking(self):
         if self.update_thread is None:
-            self.update_thread = threading.Thread(target=self._monitor_processes, name="app_tracker")
+            self.update_thread = threading.Thread(
+                target=self._monitor_processes, name="app_tracker"
+            )
             self.update_thread.start()
 
     def _fetch_app_names(self):
         apps = []
         seen_names = ["AppUsageGUI", "Python"]
 
-        for process in psutil.process_iter(['pid', 'name']):
+        for process in psutil.process_iter(["pid", "name"]):
             try:
-                pid = process.info['pid']
-                app_name = process.info['name']
+                pid = process.info["pid"]
+                app_name = process.info["name"]
                 app_name = app_name.split(".")[0]  # Use the base name of the process
                 if (
                     app_name not in seen_names
@@ -70,11 +82,11 @@ def _fetch_app_names(self):
                     if app_name == self.selected_app:
                         logger.debug(f"Seleced App found: {app_name}")
                         break
-                    #print(app_name)  # Debugging line to help optimize
+                    # print(app_name)  # Debugging line to help optimize
             except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
                 # Skip processes that terminate mid-iteration or are inaccessible
                 pass
-        if os.name == 'nt':
+        if os.name == "nt":
             return sorted(apps, key=str.casefold)
         return sorted(apps)
 
@@ -86,14 +98,16 @@ def _monitor_processes(self):
                 # Process count has changed; update app names
                 self.cached_process_count = current_process_count
                 self.app_names = self._fetch_app_names()
-            self.stop_event.wait(timeout=1)  # Check periodically to avoid excessive CPU usage
+            self.stop_event.wait(
+                timeout=1
+            )  # Check periodically to avoid excessive CPU usage
 
     def get_app_names(self):
         return self.app_names
 
     def get_selected_app(self):
         return self.selected_app
-    
+
     def set_selected_app(self, app):
         self.selected_app = app
 
@@ -113,10 +127,13 @@ def start(self):
     def reset(self):
         self.selected_app = None
         self.update_thread = None
-    
+
     def start_filter_reset(self, refresh=False, update_pids=False):
-        if os.name == 'nt':
-            self.temp_reset_thread = threading.Thread(target=self._reset_excluded_pids(refresh, update_pids), name="reset_filter")
+        if os.name == "nt":
+            self.temp_reset_thread = threading.Thread(
+                target=self._reset_excluded_pids(refresh, update_pids),
+                name="reset_filter",
+            )
             self.temp_reset_thread.start()
 
     def _reset_excluded_pids(self, refresh, update_pids):
@@ -136,34 +153,40 @@ def _update_excluded_apps(self):
             return
         seen_pids = []
         i = 0
-        for process in psutil.process_iter(['pid', 'status']):
+        for process in psutil.process_iter(["pid", "status"]):
             try:
-                pid = process.info['pid']
+                pid = process.info["pid"]
                 if pid in seen_pids:
                     continue
-                #print(f"Checking process: (PID: {pid})")  # Debugging line
+                # print(f"Checking process: (PID: {pid})")  # Debugging line
                 seen_pids.append(pid)
-                if process.info['status'] == psutil.STATUS_RUNNING and pid not in INCLUDED_APP_PIDS and pid not in EXCLUDED_APP_PIDS:
+                if (
+                    process.info["status"] == psutil.STATUS_RUNNING
+                    and pid not in INCLUDED_APP_PIDS
+                    and pid not in EXCLUDED_APP_PIDS
+                ):
                     if self._has_gui(pid):
                         INCLUDED_APP_PIDS.append(pid)
                     else:
                         i += 1
                         EXCLUDED_APP_PIDS.append(pid)
-                if len(seen_pids) > (400 if os.name == 'nt' else 10000):
+                if len(seen_pids) > (400 if os.name == "nt" else 10000):
                     # Limit the number of seen processes to avoid long loading times
                     break
             except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
                 # Skip processes that terminate mid-iteration or are inaccessible
                 pass
 
-        #print(f"\nExcluded app PIDs: {EXCLUDED_APP_PIDS}")  # Debugging line
+        # print(f"\nExcluded app PIDs: {EXCLUDED_APP_PIDS}")  # Debugging line
         logger.info(f"New exlusions: {i}")
-        data = {'excluded_app_pids': EXCLUDED_APP_PIDS,
-                'included_app_pids': INCLUDED_APP_PIDS}
+        data = {
+            "excluded_app_pids": EXCLUDED_APP_PIDS,
+            "included_app_pids": INCLUDED_APP_PIDS,
+        }
         write_file(apps_file(), data)
 
     def _has_gui(self, process_id):
-        if os.name == 'nt':
+        if os.name == "nt":
             try:
                 # Enumerate all top-level windows
                 for win in windows:
@@ -173,8 +196,8 @@ def _has_gui(self, process_id):
             except (ElementNotFoundError, RuntimeError):
                 return True  # Handle the case where the process is not found
             return False
-        elif sys.platform == 'darwin':
-            #TODO: Implement a better GUI check for macOS
+        elif sys.platform == "darwin":
+            # TODO: Implement a better GUI check for macOS
             try:
                 apps = NSWorkspace.sharedWorkspace().runningApplications()
                 for app in apps: