From 02df49102b0ca116b25766bff5c3b084fb07aff4 Mon Sep 17 00:00:00 2001 From: svalee Date: Sun, 28 Apr 2019 13:05:45 +0300 Subject: [PATCH 1/2] Added equality check for Connection class --- elasticsearch/connection/base.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/elasticsearch/connection/base.py b/elasticsearch/connection/base.py index d4fbe33de..225d28fb7 100644 --- a/elasticsearch/connection/base.py +++ b/elasticsearch/connection/base.py @@ -46,6 +46,13 @@ def __init__(self, host='localhost', port=9200, use_ssl=False, url_prefix='', ti def __repr__(self): return '<%s: %s>' % (self.__class__.__name__, self.host) + def __eq__(self, other): + if not isinstance(other, Connection): + raise TypeError( + "Unsupported equality check for %s and %s" % (self, other) + ) + return True + def _pretty_json(self, data): # pretty JSON in tracer curl logs try: From 2b7389d29e4a3a5f1df81a0369734b2d32fbc8c4 Mon Sep 17 00:00:00 2001 From: svalee Date: Mon, 13 May 2019 22:49:03 +0300 Subject: [PATCH 2/2] ES: make connection hashable --- elasticsearch/connection/base.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/elasticsearch/connection/base.py b/elasticsearch/connection/base.py index 225d28fb7..cb7ab532f 100644 --- a/elasticsearch/connection/base.py +++ b/elasticsearch/connection/base.py @@ -53,6 +53,9 @@ def __eq__(self, other): ) return True + def __hash__(self): + return id(self) + def _pretty_json(self, data): # pretty JSON in tracer curl logs try: