Mobilizon-Reshare-condividi-sui-social/mobilizon_reshare/cli/__init__.py

import asyncio
import logging
import traceback
from logging.config import dictConfig
from pathlib import Path

from mobilizon_reshare.config.config import get_settings
from mobilizon_reshare.storage.db import tear_down, MoReDB

logger = logging.getLogger(__name__)


async def graceful_exit(code):
    await tear_down()
    exit(code)


async def init(settings_file):
    settings = get_settings(settings_file)
    dictConfig(settings["logging"])
    db_path = Path(settings.db_path)
    db = MoReDB(db_path)
    await db.setup()


async def _safe_execution(f, settings_file):
    await init(settings_file)
    return_code = 1
    try:
        return_code = await f()
    except Exception:
        traceback.print_exc()
    finally:
        logger.debug("Closing")
        await graceful_exit(return_code)


def safe_execution(f, settings_file):
    asyncio.run(_safe_execution(f, settings_file))