From bd55569ad0698c37d680065c6a6cdcbae1503c15 Mon Sep 17 00:00:00 2001
From: Robert Latta <robert.latta@geant.org>
Date: Mon, 15 Feb 2021 10:59:51 +0000
Subject: [PATCH] added extra check against IMS converted equipment name

---
 inventory_provider/routes/ims_data.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/inventory_provider/routes/ims_data.py b/inventory_provider/routes/ims_data.py
index 4f70c3f6..589bf57b 100644
--- a/inventory_provider/routes/ims_data.py
+++ b/inventory_provider/routes/ims_data.py
@@ -1,6 +1,7 @@
 from flask import Blueprint, Response
 
 from inventory_provider.routes import common
+from inventory_provider.routes.ims_classifier import get_ims_equipment_name
 
 routes = Blueprint("ims-inventory-data-query-routes", __name__)
 
@@ -17,6 +18,10 @@ def equipment_location(equipment_name):
     redis = common.get_current_redis()
     result = redis.get(f'ims:location:{equipment_name}')
 
+    if not result:
+        equipment_name = get_ims_equipment_name(equipment_name)
+        result = redis.get(f'ims:location:{equipment_name}')
+
     if not result:
         return Response(
             response="no available info for {}".format(equipment_name),
-- 
GitLab