74 lines
1.9 KiB
Python
74 lines
1.9 KiB
Python
import contextlib
|
|
import logging.config
|
|
from argparse import ArgumentParser
|
|
from logging import Logger
|
|
from os import getpid, makedirs
|
|
from pathlib import Path
|
|
from sys import exit
|
|
|
|
from libbot.utils import json_read
|
|
|
|
from classes.pyroclient import PyroClient
|
|
from modules.logging_utils import get_logger, get_logging_config
|
|
|
|
# Main uses MongoDB implementation of DB,
|
|
# but you can also select SQLite one below
|
|
# from modules.migrator_sqlite import migrate_database
|
|
from modules.migrator_mongo import migrate_database
|
|
from modules.scheduler import scheduler
|
|
|
|
# Uncomment this and the line below client declaration
|
|
# in order to use context manager in your commands.
|
|
# Don't forget to disable PyroClient workers if you want
|
|
# to use convopyro, because workers create more event loops.
|
|
# from convopyro import Conversation
|
|
|
|
|
|
makedirs(Path("logs/"), exist_ok=True)
|
|
|
|
logging.config.dictConfig(get_logging_config())
|
|
|
|
logger: Logger = get_logger(__name__)
|
|
|
|
parser = ArgumentParser(
|
|
prog="__name__",
|
|
description="__description__",
|
|
)
|
|
|
|
# Remove if no database is being used
|
|
parser.add_argument("--migrate", action="store_true")
|
|
|
|
args = parser.parse_args()
|
|
|
|
with contextlib.suppress(ImportError):
|
|
import uvloop
|
|
|
|
uvloop.install()
|
|
|
|
|
|
def main():
|
|
# Remove if no database is being used
|
|
if args.migrate:
|
|
migrate_database()
|
|
logger.info("Migration finished. Exiting...")
|
|
exit()
|
|
|
|
client = PyroClient(scheduler=scheduler, commands_source=json_read(Path("commands.json")))
|
|
# Conversation(client)
|
|
|
|
try:
|
|
client.run()
|
|
except KeyboardInterrupt:
|
|
logger.warning("Forcefully shutting down with PID %s...", getpid())
|
|
except Exception as exc:
|
|
logger.error("An unexpected exception has occurred: %s", exc, exc_info=exc)
|
|
exit(1)
|
|
finally:
|
|
if client.scheduler is not None:
|
|
client.scheduler.shutdown()
|
|
exit()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|