rkkautsar/psmon

import time
from psmon.base import Watcher


class WallTimeLimiter(Watcher):
    watched_attrs = []

    def __init__(self, limit):
        self._limit = limit

    def fallback(self, res):
        return time.perf_counter() - self._start_time

    def register_root(self, pid):
        self._start_time = time.perf_counter()

    def update(self, processes_info):
        pass

    def get_stats(self, pid):
        return time.perf_counter() - self._start_time

    def should_terminate(self, pid):
        return time.perf_counter() - self._start_time > self._limit

    @classmethod
    def get_error(cls, pid):
        return TimeoutError, "Wall time limit exceeded!"