From 2ee0ed933a88516e300eb5d16b1d863641b0f505 Mon Sep 17 00:00:00 2001 From: Thephaseless Date: Wed, 18 Dec 2024 13:32:37 +0000 Subject: [PATCH] fix utils --- src/utils/__init__.py | 42 +----------------------------------------- 1 file changed, 1 insertion(+), 41 deletions(-) diff --git a/src/utils/__init__.py b/src/utils/__init__.py index fc6b8d1..e0dde19 100644 --- a/src/utils/__init__.py +++ b/src/utils/__init__.py @@ -1,48 +1,8 @@ import logging -import time -import psutil - -from src.utils.consts import LOG_LEVEL, MAX_CHROME_LIFETIME +from src.utils.consts import LOG_LEVEL logger = logging.getLogger("uvicorn.error") logger.setLevel(LOG_LEVEL) if len(logger.handlers) == 0: logger.addHandler(logging.StreamHandler()) - - -def kill_chromium_processes(): - # Define the prefix and time threshold - """ - Kill all chromium processes that have been running longer than the specified time threshold. - - This is used to clean up any rogue chromium processes that may be left behind. - """ - prefix = "chromium" - time_threshold = MAX_CHROME_LIFETIME - - # Get the current time - current_time = time.time() - - # Iterate through all processes - for proc in psutil.process_iter(["pid", "name", "create_time"]): - try: - # Extract process details - pid = proc.info["pid"] - name: str = proc.info["name"] - create_time = proc.info["create_time"] - - # Check if the process name starts with the prefix and has been running longer than the threshold - if ( - name - and name.startswith(prefix) - and (current_time - create_time > time_threshold) - ): - logger.info( - f"Terminating process {name} (PID: {pid}) running for {int(current_time - create_time)} seconds" - ) - psutil.Process(pid).terminate() # Terminate the process - - except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess): - # Ignore processes that no longer exist or can't be accessed - pass