Рекурсивный поиск в словаре

site = {
    'html': {
        'head': {
            'title': 'Мой сайт'
        },
        'body': {
            'h2': 'Здесь будет мой заголовок',
            'div': 'Тут, наверное, какой-то блок',
            'p': 'А вот здесь новый абзац'
        }
    }
}

Введите искомый ключ: head
Хотите ввести максимальную глубину? Y/N: n
Значение ключа: {'title': 'Мой сайт'}

Введите искомый ключ: head
Хотите ввести максимальную глубину? Y/N: y
Введите максимальную глубину: 1
Значение ключа: None

def find_key(key, dict):
    if key in dict:
        return dict[key]
    for i in dict.values():
        return find_key(key, i)
 
 
key = input('Введите искомый ключ: ')
print(find_key(key, site))

def get_value(object: any, searched_key: str) -> any:
  if type(object) is not dict:
    return None
 
  if searched_key in object:
    return object.get(searched_key)
 
  for value in object.values():    
    if (result := get_value(value, searched_key)) is not None:
      break
  else:
    result = None   
 
  return result

def get_value(object: any, searched_key: str, nested_value: int | None=None) -> any:
  if type(object) is not dict:
    return None
 
  if searched_key in object:
    return object.get(searched_key)
 
  if nested_value is not None and nested_value == 0:
    return None
 
  for value in object.values():
    if (result := get_value(
      value, 
      searched_key, 
      (nested_value-1 if nested_value is not None else None)
    )) is not None:
      break
  else:
    result = None
  return result
 
print(get_value(site, 'h2', 1))  # None
print(get_value(site, 'h2', 2))  # Здесь будет мой заголовок

def get_value(object, searched_key, nested_value=None):
  if type(object) is not dict:
    return None
 
  if searched_key in object:
    return object.get(searched_key)
 
  if nested_value is not None and nested_value == 0:
    return None
 
  for value in object.values():
    if (result := get_value(
      value, 
      searched_key, 
      (nested_value-1 if nested_value is not None else None)
    )) is not None:
      break
  else:
    result = None
  return result

def enter():
  def get_value(object: any, searched_key: str, nested_value: int | None=None) -> any:
    if type(object) is not dict:
      return None
 
    if searched_key in object:
      return object.get(searched_key)
 
    if nested_value is not None and nested_value == 0:
      return None
 
    for value in object.values():
      if (result := get_value(
        value, 
        searched_key, 
        (nested_value-1 if nested_value is not None else None)
      )) is not None:
        break
    else:
      result = None
    return result
 
  def get_user_input(message: str) -> str:
    return input(message)
  
  def get_number() -> int:
    try:
      return int(get_user_input('Введите максимальную глубину: '))
    except ValueError:
      print('Введите число!')
      return get_number()
 
  site = {
    'html': {
      'head': {
          'title': 'Мой сайт'
      },
      'body': {
        'h2': 'Здесь будет мой заголовок',
        'div': 'Тут, наверное, какой-то блок',
        'p': 'А вот здесь новый абзац'
      }
    }
  }
 
  to_search = get_user_input('Введите искомый ключ: ')
  while (use_nested_value := get_user_input('Хотите ввести максимальную глубину? Y/N: ').lower()) not in 'yn':
    ...
    
  nested_value_input = get_number() if use_nested_value == 'y' else None
  print('Значение ключа: ', get_value(site, to_search, nested_value_input))
  
 
if __name__ == '__main__':
  enter()