Add __slots__ to Connection classes

edgedb · Jan 11, 2022 · 447e585 · 447e585
1 parent 97d212b
commit 447e585
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/edgedb/asyncio_client.py b/edgedb/asyncio_client.py
@@ -46,6 +46,7 @@
 
 
 class AsyncIOConnection(base_client.BaseConnection):
+    __slots__ = ("_loop", "_holder")
 
     def __init__(self, loop, *args, **kwargs):
         super().__init__(*args, **kwargs)

diff --git a/edgedb/base_client.py b/edgedb/base_client.py
@@ -39,6 +39,15 @@ class BaseConnection:
     _log_listeners: typing.Set[
         typing.Callable[[BaseConnection_T, errors.EdgeDBMessage], None]
     ]
+    __slots__ = (
+        "__weakref__",
+        "_protocol",
+        "_addr",
+        "_addrs",
+        "_config",
+        "_params",
+        "_log_listeners",
+    )
 
     def __init__(
         self,

diff --git a/edgedb/blocking_client.py b/edgedb/blocking_client.py
@@ -33,6 +33,7 @@
 
 
 class BlockingIOConnection(base_client.BaseConnection):
+    __slots__ = ()
 
     def connect(self, *, single_attempt=False):
         start = time.monotonic()