department_tracking_system/app/app.py

"""General functions module"""

import random
import math
import re

from app import LOGGER, TELEGRAM_BOT, database, api


def update_department(state_id, department_type):
    """Update department professors"""
    LOGGER.info('"%s": Run update for "%s" department for state', state_id, department_type)
    latest_professor = database.get_latest_professor(state_id, department_type)
    date = None
    if latest_professor:
        date = latest_professor.date_time
    professors = api.get_professors(state_id, department_type, date)
    LOGGER.info(
        '"%s": Found "%s" new professors in "%s" department',
        state_id, len(professors), department_type
    )
    # print_professors(professors)
    database.save_professors(state_id, department_type, professors)
    LOGGER.info('"%s": saved professors', state_id)


def send_progress_message(state_id, department_type, language):
    """Update department professors"""
    LOGGER.info('"%s": Send progress message for "%s" department', state_id, department_type)
    yesterday_professors = database.get_yesterday_professors(state_id, department_type)
    if not yesterday_professors:
        LOGGER.warning('"%s": 0 professor yesterday in "%s" department', state_id, department_type)
        return
    yesterday_total = 0
    for professor in yesterday_professors:
        yesterday_total += professor.points

    institutes = api.get_institutes()
    uranium_institutes = []
    for institute in institutes:
        if institute['department_type'] == department_type:
            uranium_institutes.append(institute)
            if institute['state_id'] == state_id:
                state_institute = institute
    top_department = uranium_institutes[0]
    top_value = math.ceil(top_department['value'] / 10) * 10
    points_per_day = round(top_value / 14)
    msg_current = "Huidige uranium bonus is {} % met {} punten.".format(
        state_institute['current_bonus'],
        state_institute['value']
    )
    if state_institute['current_bonus'] == 10:
        msg_required = "Dagelijks zijn er {} punten nodig.".format(
            points_per_day
        )
    else:
        msg_required = \
            "Benodigde punten voor 10 % bonus: {} wat dagelijks {} punten zijn.".format(
                top_value,
                points_per_day
            )
    msg_yesterday = \
        "Aantal punten gisteren: {}, wat {} % van de benodigde aantal punten is.".format(
            yesterday_total,
            round(100 / points_per_day * yesterday_total)
        )
    message = ' '.join([
        msg_current,
        msg_required,
        msg_yesterday
    ])
    print(message)
    api.send_message(language, message)

def send_lotery_message(state_id, department_type, language, amount):
    """Send lotery message"""
    LOGGER.info('"%s": Send lotery message for "%s" department', state_id, department_type)
    yesterday_professors = database.get_yesterday_professors(state_id, department_type)
    professor_count = len(yesterday_professors)
    random_index = random.randint(0, professor_count) - 1
    winning_professor = yesterday_professors[random_index]
    winner = winning_professor.player
    winner_name = re.sub(r'\[.*]\s', '', winner.name)
    amount_of_points = database.get_amount_of_points(state_id, department_type, winner.id)
    LOGGER.info(
        '"%s": candidates "%s", winner "%s" with "%s" points',
        state_id, professor_count, winner_name, amount_of_points
    )
    tg_message = '. '.join([
        "The daily department lotery is won by: {:}".format(winner_name),
        "With {:} points you won $ {:,.0f}".format(
            amount_of_points, amount_of_points * amount
        ).replace(',', '.'),
        "Send @bergjnl a message to receive your price.",
    ])
    print(tg_message)
    TELEGRAM_BOT.sendMessage(chat_id='@vn_lottery', text=tg_message)
    rr_message = '. '.join([
        "De department loterij is gewonnen door: {:}".format(winner_name),
        "Met {:} punten heb je $ {:,.0f} gewonnen".format(
            amount_of_points, amount_of_points * amount
        ).replace(',', '.'),
        "Stuur me een bericht om de prijs te ontvangen.",
    ])
    print(rr_message)
    api.send_message(language, rr_message)