jinja and remove_affix: fix bot class

certtools · Apr 28, 2023 · 46a9b7a · 46a9b7a
1 parent 47c7b20
commit 46a9b7a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/intelmq/bots/experts/jinja/expert.py b/intelmq/bots/experts/jinja/expert.py
@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: AGPL-3.0-or-later
 
-from intelmq.lib.bot import Bot
+from intelmq.lib.bot import ExpertBot
 from intelmq.lib.exceptions import MissingDependencyError
 
 import pathlib
@@ -15,7 +15,7 @@
     Template = None
 
 
-class JinjaExpertBot(Bot):
+class JinjaExpertBot(ExpertBot):
     """
     Modify the message using the Jinja templating engine
     Example:

diff --git a/intelmq/bots/experts/remove_affix/expert.py b/intelmq/bots/experts/remove_affix/expert.py
@@ -4,10 +4,10 @@
 SPDX-FileCopyrightText: 2021 Marius Karotkis <[email protected]>
 SPDX-License-Identifier: AGPL-3.0-or-later
 """
-from intelmq.lib.bot import Bot
+from intelmq.lib.bot import ExpertBot
 
 
-class RemoveAffixExpertBot(Bot):
+class RemoveAffixExpertBot(ExpertBot):
     remove_prefix: bool = True  # True - from start, False - from end
     affix: str = 'www.'
     field: str = 'source.fqdn'