编辑文件 - delay_on_cron

✏️ 正在编辑: delay_on_cron_call.py

路径: /opt/imunify360/venv/share/imunify360/scripts/delay_on_cron_call.py
提示: 您可以编辑任何文件（包括二进制文件），但请注意不当修改可能导致文件损坏。

#!/opt/imunify360/venv/bin/python3
"""
Script for internal use in cron update scripts.

This script detects when parent call was maid from cron job and
apply the random delay in 60 minutes
"""

import os
import psutil
import random
import time

MAX_SLEEP_TIME = 3600

def is_current_process_cron_child():
    pid = os.getpid()
    try:
        process = psutil.Process(pid)
        parents = process.parents()

# True for cron cronie anacron and etc..
        return any(is_cron(proc.name().lower()) for proc in parents)
    except psutil.NoSuchProcess:
        print("Unable to determine parent process ID.")
        return False

def is_cron(proc_name: str):
    return "cron" in proc_name and "imunify" not in proc_name

def sleep_in_cron():
    try:
        if is_current_process_cron_child():
            sleep_time = random.randint(0, MAX_SLEEP_TIME)
            print(f"Wait for {sleep_time} seconds before starting updates")
            time.sleep(sleep_time)
        else:
            print("Not in cron, run update immediately")
    except Exception as e:
        print("An error occurred while waiting: ", e)

sleep_in_cron()

← 返回文件管理器