Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2026-04-01 14:52:29 -04:00
Code Issues Packages Projects Releases Wiki Activity

Nicer temp file names

Browse Source
This commit is contained in:
Kovid Goyal 2025-10-05 12:39:28 +05:30
parent 5a5ff4e86a
commit 48a6c767ec
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/calibre/db/cache.py
View File

@ -3155,9 +3155,9 @@ class Cache:
poff += 1
@contextmanager
def tempfile_for_export(name: str) -> Iterator[str]:
def tempfile_for_export(which: str) -> Iterator[str]:
import tempfile
fd, ans = tempfile.mkstemp(suffix=name, dir=exporter.base)
fd, ans = tempfile.mkstemp(suffix=f'-{which}.db', dir=exporter.base)
os.close(fd)
try:
yield ans
@ -3165,19 +3165,19 @@ class Cache:
os.remove(ans)
report_progress('metadata.db')
with tempfile_for_export('-export.db') as tf:
with tempfile_for_export('metadata') as tf:
self.backend.backup_database(tf)
dbkey = key_prefix + ':::' + 'metadata.db'
with open(tf, 'rb') as f:
exporter.add_file(f, dbkey)
if has_fts:
report_progress('full-text-search.db')
with tempfile_for_export('-export.db') as tf:
with tempfile_for_export('fts') as tf:
self.backend.backup_fts_database(tf)
ftsdbkey = key_prefix + ':::full-text-search.db'
with open(tf, 'rb') as f:
exporter.add_file(f, ftsdbkey)
with tempfile_for_export('-export.db') as tf, open(tf, 'r+b') as pt:
with tempfile_for_export('notes') as tf, open(tf, 'r+b') as pt:
self.backend.export_notes_data(pt)
pt.flush()
pt.seek(0)