remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-07-24 09:54:31 +02:00
Code Issues Projects Releases Packages Wiki Activity

sqlite-utils memory handles files with same filename, closes #325

Browse source
This commit is contained in:
Simon Willison 2021-09-22 13:45:37 -07:00
commit c1b26eed03
3 changed files with 35 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
sqlite_utils/cli.py
View file

@ -1275,6 +1275,7 @@ def memory(
if (dump or save or schema or analyze) and not paths:
paths = [sql]
sql = None
stem_counts = {}
for i, path in enumerate(paths):
# Path may have a :format suffix
if ":" in path and path.rsplit(":", 1)[-1].upper() in Format.__members__:
@ -1287,7 +1288,12 @@ def memory(
csv_table = "stdin"
else:
csv_path = pathlib.Path(path)
csv_table = csv_path.stem
stem = csv_path.stem
if stem_counts.get(stem):
csv_table = "{}_{}".format(stem, stem_counts[stem])
else:
csv_table = stem
stem_counts[stem] = stem_counts.get(stem, 1) + 1
csv_fp = csv_path.open("rb")
rows, format_used = rows_from_file(csv_fp, format=format, encoding=encoding)
tracker = None