# pyruse is intended as a replacement to both fail2ban and epylog
# Copyright © 2017–2018 Y. Gablin
# Full licensing information in the LICENSE file, or gnu.org/licences/gpl-3.0.txt if the file is missing.
import datetime
from pyruse import base, counter

class Action(base.Action, counter.Counter):
    def __init__(self, args):
        base.Action.__init__(self)
        counter.Counter.__init__(self, args["counter"])
        self.keyName = args["for"]
        self.save = args.get("save", None)
        graceSeconds = args.get("graceSeconds", None)
        if graceSeconds:
            self.graceSeconds = datetime.timedelta(seconds = graceSeconds)
        else:
            self.graceSeconds = None

    def act(self, entry):
        self.reset(entry[self.keyName], self.graceSeconds)
        if self.save:
            entry[self.save] = 0