main.py

import os
import time

from dotenv import load_dotenv
import database
from twilio.rest import Client


def get_time_remaining() -> str:
    epoch = 1717657200
    current = int(time.time())
    remaining = epoch - current

    days = remaining // (24 * 3600)
    remaining = remaining % (24 * 3600)
    hours = remaining // 3600
    remaining %= 3600
    minutes = remaining // 60

    return f'{days} days {hours} hours {minutes} minutes'


def get_body() -> str:
    db = database.DatabaseManager()
    total_count = db.get_corpus_count()
    total_entity_count = db.get_entity_count()
    total_entity_count_unique = db.get_entity_count_unique()
    latest = db.get_latest_100_platforms()
    db.close_connection()

    platform_counts = {}
    for platform in latest:
        platform = platform.split('/', 1)[1]
        platform_counts.update({platform: platform_counts.get(platform, 0) + 1})

    b = '*' + str(total_count) + '*\n\n'
    b += '*Remaining Time:* ' + get_time_remaining() + '\n\n'
    b += '*Latest 100 Subreddits:*\n\n'

    platform_counts = {k: v for k, v in sorted(platform_counts.items(), key=lambda item: item[1], reverse=True)}
    for platform, count in platform_counts.items():
        b += f'- {platform}: {count}\n'

    b += '\n'

    b = f"Your appointment is coming up on {total_count},{total_entity_count},{total_entity_count_unique}"
    return b


def get_twilio_client() -> Client:
    account_sid = os.getenv('TWILIO_ACCOUNT_SID')
    auth_token = os.getenv('TWILIO_AUTH_TOKEN')
    return Client(account_sid, auth_token)


def send_message(c: Client, b: str, f: str, t: str) -> None:
    """
    Send a message using the Twilio client.

    :param c:   Twilio client
    :param b:   Body of the message
    :param f:   From number
    :param t:   To number
    :return:
    """
    c.messages.create(
        from_='whatsapp:' + f,
        body=b,
        to='whatsapp:' + t
    )


if __name__ == '__main__':
    load_dotenv()

    body = get_body()
    client = get_twilio_client()

    from_ = os.getenv('TWILIO_FROM')
    to_ = os.getenv('TO_PHONE_NUMBERS')

    to_numbers = to_.split(',')
    for to_number in to_numbers:
        send_message(client, body, from_, to_number)