How can i traverse to particular key dynamically in python? - python-3.x

Question

Probelm statement:
response= [{'HotelId': 8, 'IsFast': False, 'Payload': {'HotelInstaceId': 8, 'IsResetNeeded': False, 'HotelType': 'test', 'Product': {'Family': 'Times', 'Model': 'Roman', 'Type': 'Visible', 'Vendor': 'Royal'}, 'Hotel': {'DisplayBrightness': 80, 'LedColor': None, 'Location': '', 'Name': 'testing'}}}]
I want to verify the value of name "testing".
My code:
result = response["Data"][0]["Payload"]["Name"]
if result == "testing":
print("pass")
else:
print("fail")
New code:
tests = [{'a': 1},
{'b': {'a': 2}},
{'a': 3, 'b': {'a': 4}},
{'a': {'b': 5}},
{'a': {'a': 6}}]
def recursively_find_dict_values(self, d, tests, key="HotelType"):
output = []
for i, v in d.items():
if i == key:
output.append(v)
if isinstance(v, dict):
output.extend(self.recursively_find_dict_values(v, key, tests))
return output
for t in tests:
recursively_find_dict_values(t, key="HotelType")
i will get the response from another function and that response i will send to "recursively_find_dict_values" func as a argument..
Example inputs:
Input keys Output
"Name" "testing"
"DisplayBrightness" 80
"HotelType" "test"
but i want to do in generic way like if i pass "Name" or "DisplayBrightness" or any key to the function it should return the value and also if keys in nested dictionary then also it should return the value.
How can i achieve this?

Answer 1

I think this may be able to help you:
def recursively_find_dict_values(d, key):
output = []
for i,v in d.items():
if i == key:
output.append(v)
if isinstance(v, dict):
output.extend(recursively_find_dict_values(v, key))
return output
tests = [{'a': 1},
{'b': {'a': 2}},
{'a': 3, 'b': {'a': 4}},
{'a': {'b': 5}},
{'a': {'a': 6}}]
for t in tests:
print(recursively_find_dict_values(t, 'a'))
Try it online!
I would check all the examples to see if the functionality is to your liking. Note that if the value of 'a' is a dict, it will add a dict to the list, and if the dict value of 'a' also has a key 'a', both levels will be added to the dict (see the fifth example). Also, even if there is only one match, it will still return a list.
In this example, d is the name of your input dict:
# Getting all the values of key in d:
values = recursively_find_dict_values(d, key)
# Checking if key has value v in d:
if v in recursively_find_dict_values(d):
print('Success!')