diff --git a/inventory_provider/tasks/worker.py b/inventory_provider/tasks/worker.py index 328594950566047033bac348a55dd1a7ba5d7e65..df66c2c5d4d5020dc236e4939ae17a042117d27b 100644 --- a/inventory_provider/tasks/worker.py +++ b/inventory_provider/tasks/worker.py @@ -461,8 +461,15 @@ def update_entry_point(self): def _get_router_vendor(router): return ims_rv.get(router.lower().split('.geant.')[0], 'unknown') + def _get_lab_router_vendor(router): + _rv = ims_rv.get(router.lower().split('.geant.')[0]) + if not _rv: + _rv = ims_rv.get(router.lower().split('.office.')[0], + 'unknown') + return _rv + rv = {r: _get_router_vendor(r) for r in routers} - lab_rv = {r: _get_router_vendor(r) for r in lab_routers} + lab_rv = {r: _get_lab_router_vendor(r) for r in lab_routers} chord( ( ims_task.s().on_error(task_error_handler.s()),