recap_generator.py

#!/usr/bin/env python3

"""
Daily recap generator (French version)

Emile Villette - March 2021
"""
import json
import locale
import os
from datetime import date, timedelta

from translator import trans


def generate_recap(
        recap_file, path, user_data, total, yesterday_missing, uuid, language, test_mode=False
):
    """

    :param language: user preferred language
    :param uuid: user unique id
    :param test_mode: test mode
    :param yesterday_missing: user's yesterday missing data
    :param total: the filename with the total cases in it
    :param recap_file: The recap file's name (with the extension) (string)
    :param path: the directory with the recap file in it (string)
    :param user_data: file of the user with the preferences
    :return: recap, a string of the daily recap.
    """
    # Set locale separator to the local preference
    locale.setlocale(locale.LC_ALL, "")

    # Open json file with list of chosen countries
    with open(path + recap_file, "r") as file:
        data = json.loads(file.read())

    with open("languages/countries.json", "r") as countries_file:
        countries_data = json.load(countries_file)

    # Open json file with the language data in it, according to the user's preferred language
    with open("languages/EN.json", "r", encoding="utf-8") as language_file:
        language_data = json.load(language_file)
        yesterday_sentence = language_data["yesterday_missed_data"]

    # Initialize recap
    recap = ""

    today_recap = generate_per_country_recap(
        path, user_data, language_data, countries_data, data, test_mode, uuid
    )

    recap += today_recap[0]
    to_be_translated = today_recap[1]

    with open(path + total, "r") as total_file:
        total_cases = json.loads(total_file.read())

    with open("emaildata/email3.html", "r") as inner_global:
        html_email_global = inner_global.read()

    with open(
            path + "/vaccine/" + "AA_DAILY_TOTAL_GLOBAL_VACCINE.json", "r"
    ) as vaccine_total_file:
        total_vaccine = json.loads(vaccine_total_file.read())

    total_vaccine_data = []

    for global_vaccine in total_vaccine.values():
        total_vaccine_data.append(global_vaccine)

    recap += html_email_global

    to_be_translated.append("Globally:")
    to_be_translated.append("{:n} (+{:n}) cases and {:n} (+{:n}) deaths worldwide.".format(
        total_cases["cases"],
        total_cases["todayCases"],
        total_cases["deaths"],
        total_cases["todayDeaths"]))
    to_be_translated.append("A total of {:n} vaccines have been administrated globally (+{:n} today).".format(
        total_vaccine_data[1],
        int(total_vaccine_data[1]) - int(total_vaccine_data[0])))

    if len(yesterday_missing) > 0:
        with open("emaildata/email2.html", "r") as yesterday_file:
            temp_yesterday = yesterday_file.read()
        recap += temp_yesterday
        to_be_translated.append(yesterday_sentence)
        to_be_translated.append("")

        new_path = f"data/{str(date.today() - timedelta(days=1))}/"
        yesterday_recap = generate_per_country_recap(
            new_path,
            user_data,
            language_data,
            countries_data,
            data,
            test_mode,
            uuid,
            treating_missing=True,
            missing_list=yesterday_missing,
        )
        recap += yesterday_recap[0]
        for translation_string in yesterday_recap[1]:
            to_be_translated.append(translation_string)
    translation_raw = ""
    for element in range(len(to_be_translated)):
        to_be_translated[element] = to_be_translated[element].replace("\n", "")
        translation_raw += to_be_translated[element] + "\n"

    translated = trans(translation_raw, "en", language).split("\n")
    recap = recap.format(*translated)
    recap = recap.replace("<li>",
                          """<li style="-webkit-text-size-adjust:none;-ms-text-size-adjust:none;mso-line-height-rule:exactly;font-family:arial, 'helvetica neue', helvetica, sans-serif;line-height:21px;Margin-bottom:15px;color:#999999;font-size:14px">""")

    return recap


def generate_per_country_recap(
        inner_path,
        user_data,
        language_data,
        countries_data,
        data,
        test_mode,
        uuid,
        treating_missing=False,
        missing_list=[],
):
    inner_recap = ""

    today_missing_countries = []
    if treating_missing:
        user_cases = missing_list.copy()
    else:
        user_cases = user_data["countries"].split(",")
    user_vaccine = user_data["vaccine detail"].split(",")
    user_critical = user_data["critical"].split(",")
    user_100k = user_data["active cases per 100k"].split(",")
    user_recovered = user_data["recovered today"].split(",")

    case_sentence = language_data["cases"]
    critical_sentence = language_data["critical"]
    vaccine_sentence = language_data["vaccine"]
    no_data_sentence = language_data["no_data"]
    still_no_data_sentence = language_data["still_no_data"]

    with open("emaildata/email2.html", "r") as inner_country:
        html_email_country = inner_country.read()

    normal_inner = """<li>{}</li>
{}
{}"""
    inner_vaccine = "<li>{}</li>"

    missing_inner = "<li>{}</li>"

    translation_list = []

    with open(f"{inner_path}/AA_to_download.json", "r") as missing_data_file:
        missing_countries = json.load(missing_data_file)

    for country in range(len(user_cases)):
        if os.path.isfile(f"{inner_path}/vaccine/{user_cases[country]}_VACCINE.json"):
            with open(
                    inner_path + "vaccine/" +
                    user_cases[country] + "_VACCINE.json", "r"
            ) as vaccine_file:
                vaccine_data = json.loads(vaccine_file.read())
        else:
            vaccine_data = None

        if vaccine_data is not None:
            vaccine_delta = []
            for day in vaccine_data["timeline"].values():
                vaccine_delta.append(day)

        if treating_missing:
            if (
                    data[user_cases[country]]["todayDeaths"] == 0
                    and data[user_cases[country]]["todayCases"] == 0
            ):
                inner_recap += html_email_country
                translation_list.append(countries_data["couples"][user_cases[country]])
                translation_list.append(missing_inner.format(still_no_data_sentence))
            else:
                if (
                        vaccine_data is not None and os.path.isfile(
                    f"{inner_path}/vaccine/{user_cases[country]}_VACCINE.json") and (
                        user_cases[country] in user_vaccine)
                ):
                    if int(vaccine_delta[1]) - int(vaccine_delta[0]) > 0:

                        temp_vacc = inner_vaccine.format(
                            "A total of {:n} vaccines have been administrated (+ {:n}).".format(vaccine_delta[1], int(
                                vaccine_delta[1]) - int(vaccine_delta[0])), )
                    else:
                        temp_vacc = ""
                else:
                    temp_vacc = ""

                if user_cases[country] in user_critical:
                    temp_crit = "<li>{}</li>".format(
                        "{:n} people are in a critical state right now.".format(data[user_cases[country]]["critical"]))
                else:
                    temp_crit = ""
                inner_recap += html_email_country
                translation_list.append(countries_data["couples"][user_cases[country]])
                translation_list.append(
                    normal_inner.format(case_sentence.format(data[user_cases[country]]["todayDeaths"],
                                                             data[user_cases[country]]["todayCases"]),
                                        temp_crit,
                                        temp_vacc))
        else:
            if user_cases[country] not in missing_countries:
                if (
                        vaccine_data is not None and os.path.isfile(
                    f"{inner_path}/vaccine/{user_cases[country]}_VACCINE.json") and (
                        user_cases[country] in user_vaccine)
                ):
                    if int(vaccine_delta[1]) - int(vaccine_delta[0]) > 0:

                        temp_vacc = inner_vaccine.format(
                            "A total of {:n} vaccines have been administrated (+ {:n}).".format(vaccine_delta[1], int(
                                vaccine_delta[1]) - int(vaccine_delta[0])), )
                    else:
                        temp_vacc = ""
                else:
                    temp_vacc = ""

                if user_cases[country] in user_critical:
                    temp_crit = "<li>{}</li>".format(
                        "{:n} people are in a critical state right now.".format(data[user_cases[country]]["critical"]))

                else:
                    temp_crit = ""
                inner_recap += html_email_country
                translation_list.append(countries_data["couples"][user_cases[country]])
                translation_list.append(
                    normal_inner.format(case_sentence.format(data[user_cases[country]]["todayDeaths"],
                                                             data[user_cases[country]]["todayCases"]),
                                        temp_crit,
                                        temp_vacc))
            else:
                inner_recap += html_email_country
                translation_list.append(countries_data["couples"][user_cases[country]])
                translation_list.append(missing_inner.format(no_data_sentence))
                today_missing_countries.append(user_cases[country])

        if not treating_missing and not test_mode:
            with open(f"User_data/Users/{uuid}.json", "r") as read_user_data:
                old_data = json.load(read_user_data)
            old_data["yesterday_missing"] = today_missing_countries
            with open(f"User_data/Users/{uuid}.json", "w") as write_user_data:
                json.dump(old_data, write_user_data)

    return [inner_recap, translation_list]