mirror of
https://github.com/chatmail/core.git
synced 2026-04-06 15:42:10 +03:00
77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
|
|
from queue import Queue
|
|
from threading import Event
|
|
|
|
from .hookspec import account_hookimpl
|
|
|
|
|
|
class ImexFailed(RuntimeError):
|
|
""" Exception for signalling that import/export operations failed."""
|
|
|
|
|
|
class ImexTracker:
|
|
def __init__(self):
|
|
self._imex_events = Queue()
|
|
|
|
@account_hookimpl
|
|
def process_ffi_event(self, ffi_event):
|
|
if ffi_event.name == "DC_EVENT_IMEX_PROGRESS":
|
|
self._imex_events.put(ffi_event.data1)
|
|
elif ffi_event.name == "DC_EVENT_IMEX_FILE_WRITTEN":
|
|
self._imex_events.put(ffi_event.data1)
|
|
|
|
def wait_finish(self, progress_timeout=60):
|
|
""" Return list of written files, raise ValueError if ExportFailed. """
|
|
files_written = []
|
|
while True:
|
|
ev = self._imex_events.get(timeout=progress_timeout)
|
|
if isinstance(ev, str):
|
|
files_written.append(ev)
|
|
elif ev == 0:
|
|
raise ImexFailed("export failed, exp-files: {}".format(files_written))
|
|
elif ev == 1000:
|
|
return files_written
|
|
|
|
|
|
class ConfigureFailed(RuntimeError):
|
|
""" Exception for signalling that configuration failed."""
|
|
|
|
|
|
class ConfigureTracker:
|
|
ConfigureFailed = ConfigureFailed
|
|
|
|
def __init__(self):
|
|
self._configure_events = Queue()
|
|
self._smtp_finished = Event()
|
|
self._imap_finished = Event()
|
|
self._ffi_events = []
|
|
|
|
@account_hookimpl
|
|
def process_ffi_event(self, ffi_event):
|
|
self._ffi_events.append(ffi_event)
|
|
if ffi_event.name == "DC_EVENT_SMTP_CONNECTED":
|
|
self._smtp_finished.set()
|
|
elif ffi_event.name == "DC_EVENT_IMAP_CONNECTED":
|
|
self._imap_finished.set()
|
|
|
|
@account_hookimpl
|
|
def configure_completed(self, success):
|
|
self._configure_events.put(success)
|
|
|
|
def wait_smtp_connected(self):
|
|
""" wait until smtp is configured. """
|
|
self._smtp_finished.wait()
|
|
|
|
def wait_imap_connected(self):
|
|
""" wait until smtp is configured. """
|
|
self._imap_finished.wait()
|
|
|
|
def wait_finish(self):
|
|
""" wait until configure is completed.
|
|
|
|
Raise Exception if Configure failed
|
|
"""
|
|
if not self._configure_events.get():
|
|
content = "\n".join(map(str, self._ffi_events))
|
|
raise ConfigureFailed(content)
|