Minor changes

Author: codeSamuraii

lib/metadata.py

@@ -4,11 +4,11 @@
 
 
 class FileMetadata(BaseModel):
-    name: StrictStr = Field(description="File name", min_length=2, max_length=255, validation_alias=AliasChoices('name', 'file_name'))
-    size: ByteSize = Field(description="Size in bytes", gt=0,validation_alias=AliasChoices('size', 'file_size'))
-    type: StrictStr = Field(description="MIME type", default='application/octet-stream',validation_alias=AliasChoices('type', 'file_type', 'content_type'))
+    name: StrictStr = Field(description="File name", min_length=2, max_length=255)
+    size: ByteSize = Field(description="Size in bytes", gt=0)
+    type: StrictStr = Field(description="MIME type", default='application/octet-stream')
 
-    model_config = ConfigDict(validate_by_name=True, populate_by_name=True)
+    model_config = ConfigDict(title="File transfer metadata", alias_generator=lambda s: f'file_{s}', populate_by_name=True, validate_by_name=True)
 
     @field_validator('name')
     @classmethod

lib/transfer.py

@@ -137,6 +137,9 @@ async def supply_download(self, on_error: Callable[[Exception | str], Awaitable[
             self.warning(f"▼ Timeout during download.")
             await on_error("Timeout during download.")
 
+        else:
+            await asyncio.sleep(1.0)
+
     async def cleanup(self):
         try:
             await asyncio.wait_for(self.store.cleanup(), timeout=30.0)

tests/helpers.py

@@ -1,6 +1,7 @@
+import asyncio
 from string import ascii_letters
 from itertools import islice, repeat, chain
-from typing import Tuple, Iterable, Iterator
+from typing import Tuple, Iterable, AsyncIterator
 from annotated_types import T
 
 from lib.metadata import FileMetadata
@@ -19,12 +20,8 @@ def generate_test_file(size_in_kb: int = 10) -> tuple[bytes, FileMetadata]:
     return content, metadata
 
 
-def batched(iterable: Iterable[T], chunk_size: int) -> Iterator[Tuple[T, ...]]:
-    "Batch data into lists of length n. The last batch may be shorter."
-    # batched('ABCDEFG', 3) --> ABC DEF G
-    it = iter(iterable)
-    while True:
-        batch = bytes(islice(it, chunk_size))
-        if not batch:
-            return
-        yield batch
+async def chunks(data: bytes, chunk_size: int = 1024) -> AsyncIterator[bytes]:
+    """Yield successive chunks of data."""
+    for i in range(0, len(data), chunk_size):
+        yield data[i:i + chunk_size]
+        await asyncio.sleep(0)