remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-09-23 18:34:13 +02:00
Code Issues Projects Releases Packages Wiki Activity

Handle empty input in rows_from_file

Browse source
This commit is contained in:
Rami Abdelrazzaq 2026-08-12 13:06:25 -05:00
commit b143fc4913
2 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
sqlite_utils/utils.py
View file

@ -368,6 +368,8 @@ def rows_from_file(
raise TypeError( raise TypeError(
"rows_from_file() requires a file-like object that supports peek(), such as io.BytesIO" "rows_from_file() requires a file-like object that supports peek(), such as io.BytesIO"
) )
if not first_bytes:
return (), Format.CSV
if first_bytes.startswith((b"[", b"{")): if first_bytes.startswith((b"[", b"{")):
# TODO: Detect newline-JSON # TODO: Detect newline-JSON
return rows_from_file(buffered, format=Format.JSON) return rows_from_file(buffered, format=Format.JSON)

7
tests/test_rows_from_file.py
View file

@ -20,6 +20,13 @@ def test_rows_from_file_detect_format(input, expected_format):
assert rows_list == [{"id": "1", "name": "Cleo"}] assert rows_list == [{"id": "1", "name": "Cleo"}]
@pytest.mark.parametrize("input", (b"", b" \n\t"))
def test_rows_from_file_empty_input(input):
rows, format = rows_from_file(BytesIO(input))
assert format == Format.CSV
assert list(rows) == []
@pytest.mark.parametrize( @pytest.mark.parametrize(
"ignore_extras,extras_key,expected", "ignore_extras,extras_key,expected",
( (