#!/usr/bin/python3
# coding: utf-8

"""
    author: 猪猪侠 https://github.com/ring04h

"""

import logging
import subprocess

logging.basicConfig(level=logging.DEBUG)


#
# (crontab -l;echo '0 2 * * * /usr/local/bin/python3 /data/script/zombie_clean.py') | crontab -
#

import logging
import subprocess

# 判断是否超时的函数
def is_timeout(etime):
    """
    判断进程的运行时间是否超时。

    参数：
    etime (str)：进程的运行时间字符串，格式可能为 "天数-小时数" 或直接是小时数。

    返回：
    bool：如果天数大于等于 1 或者小时数超过一定阈值（这里没有具体指定阈值），返回 True，表示超时；否则返回 False。
    """
    if '-' in etime:
        day, hour = etime.split('-')
        return True if int(day) >= 1 else False
    else:
        return False

# 执行命令的函数
def cmdprocess(cmdline):
    """
    执行给定的命令行。

    参数：
    cmdline (str)：要执行的命令行字符串。

    返回：
    tuple：包含命令的标准输出、标准错误输出和返回码。
    """
    # 使用 subprocess.Popen 执行命令，捕获标准输出和标准错误输出
    pipe = subprocess.Popen(cmdline, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    # 等待命令执行完成，获取标准输出和标准错误输出
    output, stderr = pipe.communicate()
    # 获取命令的返回码
    return_code = pipe.returncode
    # 将标准错误输出解码为字符串
    stderr = stderr.decode(errors='replace')
    # 将标准输出解码为字符串
    output = output.decode(errors='replace')
    return output, stderr, return_code

def main():
    # 构建查找 crawlergo 进程 ID 的命令行
    cmdline = "ps -ef | grep crawlergo | grep -v grep | awk '{print $2}'"
    # 执行命令，获取输出、错误输出和返回码
    output, stderr, return_code = cmdprocess(cmdline)

    # 如果返回码不为 0，表示命令执行失败，直接返回
    if return_code!= 0:
        return

    # 将输出的进程 ID 按行分割成列表
    zombie_pids = output.splitlines()

    # 遍历每个 crawlergo 的进程 ID
    for zombie_pid in zombie_pids:
        # 构建查找指定进程 ID 的运行时间的命令行
        cmdline = f'''ps -eo pid,etime | grep {zombie_pid}'''
        # 执行命令，获取输出、错误输出和返回码
        ps_output, ps_stderr, ps_return_code = cmdprocess(cmdline)

        # 如果返回码不为 0，表示命令执行失败，继续下一个进程 ID 的处理
        if ps_return_code!= 0:
            continue

        # 遍历输出的每一行（包含进程 ID 和运行时间）
        for line in ps_output.splitlines():
            # 将每行按空格分割，获取进程 ID 和运行时间
            pid, etime = line.split()
            # 判断运行时间是否超时
            status = is_timeout(etime)
            # 打印调试信息，包括进程 ID、运行时间和是否超时
            logging.debug(f"PID: {pid:<8} ETIME: {etime:<15} TIMEOUT: {status}")

            # 如果没有超时，继续下一个进程 ID 的处理
            if not status:
                continue

            # 构建杀死进程的命令行
            kill_cmdline = f"kill -9 {pid}"
            # 打印调试信息，表示要执行的杀死进程的命令
            logging.debug(f"call kill : [{kill_cmdline}]")

            # 执行杀死进程的命令
            cmdprocess(kill_cmdline)


if __name__ == "__main__":
    main()