fix

sonic182 · Jan 21, 2024 · ba3d43d · ba3d43d
1 parent e272e0e
commit ba3d43d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/aiosonic/__init__.py b/aiosonic/__init__.py
@@ -442,7 +442,7 @@ async def _do_request(
 
         # get response code and version
         try:
-            line = await wait_for(connection.readline(), timeouts.sock_read)
+            line = await wait_for(connection.readuntil(), timeouts.sock_read)
             if not line:
                 raise HttpParsingError(f"response line parsing error: {line}")
             response._set_response_initial(line)

diff --git a/aiosonic/connection.py b/aiosonic/connection.py
@@ -88,7 +88,7 @@ async def readexactly(self, size: int):
             raise MissingReaderException("reader not set.")
         return await self.reader.readexactly(size)
 
-    async def readuntil(self, separator: bytes):
+    async def readuntil(self, separator: bytes = b'\n'):
         """Read until separator"""
         if not self.reader:
             raise MissingReaderException("reader not set.")