Merge pull request #62 from lidofinance/support-localhost-uri

Support localhost in uri

Author: F4ever

tests/test_util.py

@@ -13,12 +13,8 @@
     ("https://my.provider.example.io/path", "example.io"),
     ("my.provider.infura.io", "infura.io"),
     ("abc.co.uk", "co.uk"), # intentionally done this way
+    ("localhost:1010", "localhost:1010"),
 ])
 def test_normalize_provider_valid(input_uri, expected):
     assert normalize_provider(input_uri) == expected
 
-
-def test_normalize_provider_invalid_hostname():
-    with pytest.raises(ValueError, match="Unhandled hostname format"):
-        normalize_provider("localhost")
-

web3_multi_provider/util.py

@@ -31,15 +31,14 @@ def normalize_provider(uri: str) -> str:
     If uri is an IP address returns as is.
     If uri is a dns address returns two highest domains.
     """
-    uri = re.sub(r'^https?://', '', uri.lower())
+    stripped = re.sub(r'^https?://', '', uri.strip().lower())
+    host = stripped.split('/')[0]
 
-    ip_match = re.match(r'^(\d{1,3}\.){3}\d{1,3}(:\d+)?$', uri)
-    if ip_match:
-        return uri
+    if re.match(r'^((\d{1,3}\.){3}\d{1,3}|localhost)(:\d+)?$', host):
+        return host
 
-    hostname = uri.split('/')[0]
-    parts = hostname.split('.')
+    parts = host.split('.')
     if len(parts) >= 2:
         return '.'.join(parts[-2:])
 
-    raise ValueError(f'Unhandled hostname format: {{uri}}. Hostname must be either an IP address or a valid provider address.')
+    raise ValueError(f"Unhandled hostname format: {uri!r}. Hostname must be either an IP address or a valid provider address.")