1. import psutil
  2. import time
  3. import logging
  4. import os
  6. logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
  8. def teardown_processes(process_arrays, retry_interval=1):
  9. """
  10. Tear down processes from given arrays with retry intervals.
  12. Args:
  13. process_arrays (list of lists): A list of lists, where each inner list
  14. contains process names to terminate.
  15. retry_interval (int): The interval (in seconds) between retry attempts.
  16. """
  17. for process_list in process_arrays:
  18. for process_name in process_list:
  19. max_retries = 5 # Set a maximum number of retries
  20. retries = 0
  22. while retries < max_retries:
  23. try:
  24. for proc in psutil.process_iter(['name']):
  25. if proc.info['name'] == process_name:
  26. proc.kill()
  27. logging.info(f"Killed process: {process_name}")
  28. break # Exit inner loop after killing the process
  29. else:
  30. logging.warning(f"Process {process_name} not found.")
  31. break # exit retry loop if process not found
  33. break # Exit outer loop if process is killed
  34. except psutil.NoSuchProcess:
  35. logging.warning(f"Process {process_name} already terminated.")
  36. break
  37. except Exception as e:
  38. logging.error(f"Error terminating process {process_name}: {e}")
  39. retries += 1
  40. if retries < max_retries:
  41. time.sleep(retry_interval)
  43. if __name__ == '__main__':
  44. # Example usage
  45. process_arrays = [
  46. ['my_process_1', 'my_process_2'],
  47. ['another_process']
  48. ]
  50. teardown_processes(process_arrays, retry_interval=0.5)

Add your comment