40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from core.concept import TooManySuccessConcept
|
|
from core.sheerka import ReturnValue
|
|
from evaluators.BaseEvaluator import BaseEvaluator
|
|
import logging
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class DefaultEvaluator(BaseEvaluator):
|
|
"""
|
|
Used to filter the responses of the parsers
|
|
"""
|
|
|
|
def __init__(self):
|
|
super().__init__("Default Evaluator", 90)
|
|
|
|
def matches(self, context, items):
|
|
return True
|
|
|
|
def eval(self, context, items):
|
|
successful_results = [item for item in items if item.status]
|
|
number_of_successful = len(successful_results)
|
|
total_items = len(items)
|
|
|
|
# remove errors when a winner is found
|
|
if number_of_successful == 1:
|
|
log.debug(f"1 / {total_items} good item found.")
|
|
return successful_results
|
|
|
|
# too many winners, which one to choose ?
|
|
if number_of_successful > 1:
|
|
log.debug(f"{number_of_successful} / {total_items} good items. Too many success")
|
|
return ReturnValue(self.name,
|
|
False,
|
|
context.sheerka.new(TooManySuccessConcept.NAME, body=items))
|
|
|
|
# only errors, i cannot help you
|
|
log.debug(f"{total_items} items. Only errors")
|
|
return items
|