import asyncio import signal from concurrent.futures import Future from pathlib import Path from threading import Thread from typing import Any, Coroutine, Dict, List, Optional, Sequence from uuid import uuid4 from appdirs import AppDirs from . import __about__ from .events.app import CoroutineDone, ExitRequested class App: def __init__(self) -> None: self.appdirs = AppDirs(appname=__about__.__pkg_name__, roaming=True) from .backend import Backend self.backend = Backend(app=self) self.loop = asyncio.get_event_loop() self.loop_thread = Thread(target=self._loop_starter) self.loop_thread.start() def is_debug_on(self, cli_flags: Sequence[str] = ()) -> bool: return "-d" in cli_flags or "--debug" in cli_flags def _loop_starter(self) -> None: asyncio.set_event_loop(self.loop) self.loop.run_forever() def run_in_loop(self, coro: Coroutine) -> Future: return asyncio.run_coroutine_threadsafe(coro, self.loop) def _call_coro(self, coro: Coroutine) -> str: uuid = str(uuid4()) self.run_in_loop(coro).add_done_callback( lambda future: CoroutineDone(uuid=uuid, result=future.result()) ) return uuid def call_backend_coro(self, name: str, args: Optional[List[str]] = None, kwargs: Optional[Dict[str, Any]] = None) -> str: return self._call_coro( getattr(self.backend, name)(*args or [], **kwargs or {}) ) def call_client_coro(self, account_id: str, name: str, args: Optional[List[str]] = None, kwargs: Optional[Dict[str, Any]] = None) -> str: client = self.backend.clients[account_id] return self._call_coro( getattr(client, name)(*args or [], **kwargs or {}) ) def pdb(self, additional_data: Sequence = ()) -> None: # pylint: disable=all ad = additional_data rl = self.run_in_loop ba = self.backend cl = self.backend.clients tcl = lambda user: cl[f"@test_{user}:matrix.org"] import json jd = lambda obj: print(json.dumps(obj, indent=4, ensure_ascii=False)) print("\n=> Run `socat readline tcp:127.0.0.1:4444` in a terminal " "to connect to pdb.") import remote_pdb remote_pdb.RemotePdb("127.0.0.1", 4444).set_trace() # Make CTRL-C work again signal.signal(signal.SIGINT, signal.SIG_DFL) APP = App()