Как найти элемент в дереве по data через js?

var code = prompt(); // Для примера вводим нужный код 
console.log($('[data-country-code="' + code + '"]').data('dial-code'));

<script src="https://ajax.googleapis.com/ajax/libs/jquery/2.1.1/jquery.min.js"></script> 
 
<ul> 
  <li class="country" data-dial-code="355" data-country-code="al"> 
    <span class="country-name">Albania</span> 
    <span class="dial-code">+355</span> 
  </li> 
  <li class="country" data-dial-code="7" data-country-code="ru"> 
    <span class="country-name">Russia</span> 
    <span class="dial-code">+7</span> 
  </li> 
</ul>