hashing support for urls (#10975)

sydney-runkle · sydney-runkle · commit 02229a6ab176 · 2024-11-26T07:20:37.000-05:00
diff --git a/pydantic/networks.py b/pydantic/networks.py
@@ -224,6 +224,9 @@ def __deepcopy__(self, memo: dict) -> Self:
     def __eq__(self, other: Any) -> bool:
         return self.__class__ is other.__class__ and self._url == other._url
 
+    def __hash__(self) -> int:
+        return hash(self._url)
+
     @classmethod
     def build(
         cls,
@@ -370,6 +373,9 @@ def __deepcopy__(self, memo: dict) -> Self:
     def __eq__(self, other: Any) -> bool:
         return self.__class__ is other.__class__ and self._url == other._url
 
+    def __hash__(self) -> int:
+        return hash(self._url)
+
     @classmethod
     def build(
         cls,
diff --git a/tests/test_networks.py b/tests/test_networks.py
@@ -1111,3 +1111,21 @@ def test_serialize_as_any() -> None:
     ta = TypeAdapter(Any)
     assert ta.dump_python(HttpUrl('https://example.com')) == HttpUrl('https://example.com/')
     assert ta.dump_json('https://example.com') == b'"https://example.com"'
+
+
+def test_any_url_hashable() -> None:
+    example_url_1a = AnyUrl('https://example1.com')
+    example_url_1b = AnyUrl('https://example1.com')
+    example_url_2 = AnyUrl('https://example2.com')
+
+    assert hash(example_url_1a) == hash(example_url_1b)
+    assert hash(example_url_1a) != hash(example_url_2)
+    assert len({example_url_1a, example_url_1b, example_url_2}) == 2
+
+    example_multi_host_url_1a = PostgresDsn('postgres://user:pass@host1:5432,host2:5432/app')
+    example_multi_host_url_1b = PostgresDsn('postgres://user:pass@host1:5432,host2:5432/app')
+    example_multi_host_url_2 = PostgresDsn('postgres://user:pass@host1:5432,host3:5432/app')
+
+    assert hash(example_multi_host_url_1a) == hash(example_multi_host_url_1b)
+    assert hash(example_multi_host_url_1a) != hash(example_multi_host_url_2)
+    assert len({example_multi_host_url_1a, example_multi_host_url_1b, example_multi_host_url_2}) == 2
