fix(ws): exclude connection_name from data

Atem18 · Nov 3, 2021 · 8176b07 · 8176b07
1 parent 2af83cd
commit 8176b07
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,6 +1,7 @@
 __pycache__
 .vscode
 env
+.env
 dist
 build
 .mypy_cache

diff --git a/kraky/ws.py b/kraky/ws.py
@@ -223,7 +223,7 @@ async def add_order(
         data = {
             arg: value
             for arg, value in locals().items()
-            if arg != "self" and value is not None
+            if arg != "self" and arg != "connection_name" and value is not None
         }
         await self._send(data=data, connection_name=connection_name)
 
@@ -239,7 +239,7 @@ async def cancel_order(
         data = {
             arg: value
             for arg, value in locals().items()
-            if arg != "self" and value is not None
+            if arg != "self" and arg != "connection_name" and value is not None
         }
         await self._send(data=data, connection_name=connection_name)
 
@@ -254,7 +254,7 @@ async def cancel_all(
         data = {
             arg: value
             for arg, value in locals().items()
-            if arg != "self" and value is not None
+            if arg != "self" and arg != "connection_name" and value is not None
         }
         await self._send(data=data, connection_name=connection_name)
 
@@ -270,6 +270,6 @@ async def cancel_all_orders_after(
         data = {
             arg: value
             for arg, value in locals().items()
-            if arg != "self" and value is not None
+            if arg != "self" and arg != "connection_name" and value is not None
         }
         await self._send(data=data, connection_name=connection_name)