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

37 lines
786 B
Python

import asyncio
import functools
import logging
import sys
import traceback
from mobilizon_reshare.config.command import CommandConfig
from mobilizon_reshare.storage.db import tear_down, init
logger = logging.getLogger(__name__)
async def graceful_exit():
await tear_down()
async def _safe_execution(function):
await init()
return_code = 1
try:
return_code = await function()
except Exception:
traceback.print_exc()
finally:
logger.debug("Closing")
await graceful_exit()
return return_code
def safe_execution(function, command_config: CommandConfig = None):
if command_config:
function = functools.partial(function, command_config)
code = asyncio.run(_safe_execution(function))
sys.exit(code)