diff --git a/SAS/ResourceAssignment/ResourceAssigner/lib/raservice.py b/SAS/ResourceAssignment/ResourceAssigner/lib/raservice.py index d802a9ca2546876f86a48dad64c9a04a242a5f15..66709f74dffe1411ff16e8be70de95ad1eee5745 100755 --- a/SAS/ResourceAssignment/ResourceAssigner/lib/raservice.py +++ b/SAS/ResourceAssignment/ResourceAssigner/lib/raservice.py @@ -57,7 +57,7 @@ class SpecifiedTaskEventMessageHandler(RATaskSpecifiedEventMessageHandler): except Exception as e: logger.error(str(e)) -__all__ = ["SpecifiedTaskListener"] +__all__ = ["SpecifiedTaskEventMessageHandler"] def main(): @@ -91,8 +91,10 @@ def main(): with ResourceAssigner(exchange=options.exchange, broker=options.broker, - radb_dbcreds=radb_dbcreds): - with RATaskSpecifiedBusListener(exchange=options.exchange, + radb_dbcreds=radb_dbcreds) as assigner: + with RATaskSpecifiedBusListener(handler_type=SpecifiedTaskEventMessageHandler, + handler_kwargs={"assigner": assigner}, + exchange=options.exchange, broker=options.broker): with ScheduleChecker(exchange=options.exchange, broker=options.broker):