Mobilizon/Mobilizon-Reshare-condividi-sui-social
1
0
Fork 0
mirror of https://github.com/Tech-Workers-Coalition-Italia/mobilizon-reshare.git synced 2025-02-06 04:13:27 +01:00
Code Issues Projects Releases Wiki Activity

fixed teardown (#42)

Browse Source
This commit is contained in:
Simone Robutti 2021-07-21 09:08:19 +02:00 committed by GitHub
parent 8a27de8981
commit 7c5f263c2b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mobilizon_bots/main.py
View File

@ -2,17 +2,21 @@ import logging.config
from pathlib import Path
from mobilizon_bots.config.config import update_settings_files
from mobilizon_bots.event.event_selection_strategies import select_event_to_publish
from mobilizon_bots.mobilizon.events import get_unpublished_events
from mobilizon_bots.publishers import get_active_publishers
from mobilizon_bots.publishers.coordinator import PublisherCoordinator
from mobilizon_bots.storage.db import MobilizonBotsDB
from mobilizon_bots.storage.db import MobilizonBotsDB, tear_down
from mobilizon_bots.storage.query import get_published_events, create_unpublished_events
logger = logging.getLogger(__name__)
async def graceful_exit(code):
await tear_down()
exit(code)
async def main(settings_file):
"""
STUB
@ -39,7 +43,8 @@ async def main(settings_file):
result = PublisherCoordinator(event).run()
logger.debug("Closing")
exit(0 if result.successful else 1)
await graceful_exit(0 if result.successful else 1)
else:
logger.debug("Closing")
exit(0)
await graceful_exit(0)

6
mobilizon_bots/storage/db.py
View File

@ -48,4 +48,8 @@ def gracefully_tear_down():
logger.info("Shutting down DB")
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
asyncio.run(Tortoise.close_connections())
asyncio.run(tear_down())
async def tear_down():
return await Tortoise.close_connections()