Disable cleanup at the beginning of the buffer for blob connections

indi/client/client.py

@@ -351,7 +351,7 @@ async def start(self):
             self.process_message
         )
         self.blob_connection_handler = await self.blob_connection.connect(
-            self.process_message
+            self.process_message, for_blobs=True
         )
 
         asyncio.get_running_loop().create_task(

indi/transport/buffer.py

@@ -1,7 +1,7 @@
 import logging
 import xml.etree.ElementTree as ET
 from io import StringIO
-from typing import Callable
+from typing import Callable, Optional
 
 from indi.message import IndiMessage
 
@@ -10,7 +10,7 @@
 
 class Buffer:
     def __init__(self) -> None:
-        self.max_buffer_size_before_frontal_cleanup = 2048
+        self.max_buffer_size_before_frontal_cleanup: Optional[int] = 2048
         self.buffer = StringIO()
         self.allowed_tags = [m.tag_name() for m in IndiMessage.all_message_classes()]
 
@@ -101,7 +101,7 @@ def process(self, callback: Callable[[IndiMessage], None]):
         while self.data_len:
             message, end = self._find_message_in_buffer()
 
-            if not message:
+            if not message and self.max_buffer_size_before_frontal_cleanup is not None:
                 if self.data_len > self.max_buffer_size_before_frontal_cleanup:
                     self._cleanup_beginning()
                     continue

indi/transport/client/tcp.py

@@ -14,8 +14,12 @@ def __init__(
         reader: asyncio.StreamReader,
         writer: asyncio.StreamWriter,
         callback: Callable[[IndiMessage], None],
+        for_blobs=False,
     ):
         self.buffer = Buffer()
+        if for_blobs:
+            self.buffer.max_buffer_size_before_frontal_cleanup = None
+
         self.reader, self.writer = reader, writer
         self.callback = callback
         self.sender_lock = asyncio.Lock()
@@ -53,7 +57,7 @@ def __init__(self, address: str = "127.0.0.1", port: int = 7624):
         self.address = address
         self.port = port
 
-    async def connect(self, callback: Callable[[IndiMessage], None]):
+    async def connect(self, callback: Callable[[IndiMessage], None], for_blobs=False):
         reader, writer = await asyncio.open_connection(self.address, self.port)
-        handler = ConnectionHandler(reader, writer, callback)
+        handler = ConnectionHandler(reader, writer, callback, for_blobs=for_blobs)
         return handler