Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

More robust atexit worker close

Browse Source
This commit is contained in:
Kovid Goyal 2025-04-05 12:57:01 +05:30
parent 2ccec1c8e2
commit 2ebf2af44f
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/calibre/utils/safe_atexit.py
View File

@ -47,15 +47,22 @@ def unlink(path):
oss.remove(path) oss.remove(path)
def close_worker(worker):
import subprocess
worker.stdin.close()
try:
worker.wait(10)
except subprocess.TimeoutExpired:
worker.kill()
worker.wait()
def ensure_worker(): def ensure_worker():
global worker global worker
if worker is None: if worker is None:
from calibre.utils.ipc.simple_worker import start_pipe_worker from calibre.utils.ipc.simple_worker import start_pipe_worker
worker = start_pipe_worker('from calibre.utils.safe_atexit import main; main()', stdout=None) worker = start_pipe_worker('from calibre.utils.safe_atexit import main; main()', stdout=None)
def close_worker(): atexit.register(close_worker, worker)
worker.stdin.close()
worker.wait(10)
atexit.register(close_worker)
return worker return worker