App.py

#!/usr/bin/env python3
"""===============================================================================

        FILE: App.py
       USAGE: (not intended to be directly executed)

 DESCRIPTION: 

     OPTIONS: ---
REQUIREMENTS: ---
        BUGS: ---
       NOTES: ---
      AUTHOR: Alex Leontiev (alozz1991@gmail.com)
ORGANIZATION: 
     VERSION: --
     CREATED: 2021-01-31T17:12:33.975745
    REVISION: ---

==============================================================================="""

from telegram import InlineKeyboardButton, InlineKeyboardMarkup, KeyboardButton, ReplyKeyboardMarkup
from telegram.ext import Updater, CommandHandler, CallbackQueryHandler, MessageHandler, Filters
from pymongo import MongoClient
import click
import json
import os
from os import path
import atexit
from py.eldrinko.el_drinko_pub_bot import ElDrinkoPubBot
import logging
from datetime import datetime


class _AtExitHook:
    def __init__(self, pidfile):
        self._pidfile = pidfile

    def __call__(self):
        click.echo("_atexit_hook")
        os.system(f"rm -rf {self._pidfile}")


@click.command()
@click.option("--mongo-url", envvar="MONGO_URL")
@click.option("--template-folder", type=click.Path(), envvar="TEMPLATE_FOLDER")
@click.option("-e", "--environment", type=click.Choice(["ElDrinkoPubBot", "ProtoElDrinkoPubBot", "DevElDrinkoPubBot"]), default="DevElDrinkoPubBot", envvar="BEERBOT_ENVIRONMENT")
@click.option("--debug/--no-debug", default=True)
def App(mongo_url, environment, debug, template_folder):
    pidfile = f".tmp/{environment}.txt"

    basic_config_kwargs = {"handlers": [], "level": logging.DEBUG}
    _handler = logging.FileHandler(
        filename=f".log/{environment}_{datetime.now().strftime('%Y%m%d%H%M%S')}.log.txt",
    )
    _handler.setFormatter(logging.Formatter(
        fmt='%(asctime)s,%(msecs)d %(levelname)-8s %(name)s [%(filename)s:%(lineno)d] %(message)s',
        datefmt='%Y-%m-%d:%H:%M:%S',
    ))
    _handler.setLevel(logging.DEBUG)
    basic_config_kwargs["handlers"].append(_handler)
    _handler = logging.StreamHandler()
    if debug:
        _handler.setLevel(logging.INFO)
    else:
        _handler.setLevel(logging.WARNING)
    basic_config_kwargs["handlers"].append(_handler)
    logging.basicConfig(**basic_config_kwargs)

    assert not path.isfile(
        pidfile), "only one instance of ElDrinkoPubBot allowed to run"
    with open(pidfile, "w") as f:
        f.write(str(os.getpid()))

    mongo_client = MongoClient(mongo_url)
    settings, keyring = [
        {k: v for k, v in mongo_client.beerbot[kk].find_one(
            {"id": environment}).items() if k not in ["_id", "id"]}
        for kk
        in ["_settings", "_keyring"]
    ]
    atexit.register(_AtExitHook(pidfile))
    click.echo("here")

    updater = Updater(keyring["telegram"]["token"], use_context=True)
    bot = updater.bot
    edbp = ElDrinkoPubBot(
        settings={**settings, "id": environment},
        bot=bot,
        mongo_url=mongo_url,
        template_folder=template_folder
    )
    updater.dispatcher.add_handler(
        MessageHandler(filters=Filters.all, callback=edbp))
    updater.dispatcher.add_handler(
        CallbackQueryHandler(callback=edbp))
    updater.start_polling()
    updater.idle()


if __name__ == "__main__":
    App()