main.py

import os
import redis

from importlib import import_module

from telegram.ext import Updater, Dispatcher

import configurations.settings as settings
import utils.logger as logger
from splitwise_telegram import SplitwiseTelegramBot

BOT_TOKEN = os.environ.get("BOT_TOKEN")

splitwise = SplitwiseTelegramBot()
redis = redis.from_url(os.environ.get("REDIS_URL"), decode_responses=True)
# r = redis.Redis(host='localhost', port=6379, db=0)
# redis://h:p895874526617dd19e8163ef5bba8f2f3ee66e85664635528765b0eeca4cc1010@ec2-34-198-97-40.compute-1.amazonaws.com:16339


def load_handlers(dispatcher: Dispatcher):
    """Load handlers from files in a 'bot' directory."""
    base_path = os.path.join(os.path.dirname(__file__), 'bot')
    files = os.listdir(base_path)

    for file_name in files:
        handler_module, _ = os.path.splitext(file_name)

        module = import_module(f'.{handler_module}', 'bot')
        module.init(dispatcher)


# TODO - find some way to store the last notification id and keep a job running which will check for new notifications
def get_notifications(update, context):
    # _initialize_bot(update)

    print(splitwise.getNotifications())


def main():
    logger.init_logger(f'logs/{settings.NAME}.log')
    updater = Updater(BOT_TOKEN, use_context=True)
    dispatcher = updater.dispatcher

    load_handlers(updater.dispatcher)

    # # dispatcher.add_handler(CommandHandler(GET_NOTIFICATIONS, get_notifications_callback))

    if settings.WEBHOOK:
        # signal.signal(signal.SIGINT, graceful_exit)
        updater.start_webhook(**settings.WEBHOOK_OPTIONS)
        # updater.bot.set_webhook(url=settings.WEBHOOK_URL)
        updater.bot.set_webhook(os.environ.get("APP_NAME") + BOT_TOKEN)
    else:
        updater.start_polling()

    # Run the bot until you press Ctrl-C or the process receives SIGINT,
    # SIGTERM or SIGABRT. This should be used most of the time, since
    # start_polling() is non-blocking and will stop the bot gracefully.
    updater.idle()


if __name__ == '__main__':
    main()