functions.py

# -*- coding: utf-8  -*-


EVENTS = [
    "earth",
    "monuments",
    "africa",
    "public_art",
    "science",
    "food",
    "folklore",
]


def get_country_data(db):
    country_data = {}
    for edition_slug in db:
        scope_slug = edition_slug[:-4]
        year = edition_slug[-4:]
        for country in db[edition_slug]:
            country_data.setdefault(country, {}).setdefault(scope_slug, {}).update(
                {
                    year: {
                        "count": db[edition_slug][country]["count"],
                        "usercount": db[edition_slug][country]["usercount"],
                        "usage": db[edition_slug][country]["usage"],
                        "userreg": db[edition_slug][country]["userreg"],
                    }
                }
            )
    return country_data


def get_events_data(db):
    return {
        name: {
            e[-4:]: {
                "count": sum(db[e][c]["count"] for c in db[e]),
                "usercount": sum(db[e][c]["usercount"] for c in db[e]),
                "userreg": sum(db[e][c]["userreg"] for c in db[e]),
                "usage": sum(db[e][c]["usage"] for c in db[e]),
                "country_count": len(db[e]),
            }
            for e in db
            if e[:-4] == name
        }
        for name in set(e[:-4] for e in db)
    }


def get_edition_data(db, edition_slug):
    return {
        country: {
            field: db[edition_slug][country][field]
            for field in db[edition_slug][country]
            if field != "users"
        }
        for country in db[edition_slug]
    }


def get_instance_users_data(db, edition_slug, country):
    return sorted(
        list(db[edition_slug][country]["users"].items()),
        key=lambda i: (i[1]["count"], i[0]),
        reverse=True,
    )


def get_menu(db):
    return {
        name: sorted(e[-4:] for e in db if e[:-4] == name)
        for name in set(e[:-4] for e in db)
    }


def get_country_summary(country_data):
    return {
        c: [
            (
                sorted(country_data[c][event].keys())
                if event in country_data[c]
                else None
            )
            for event in EVENTS
        ]
        for c in country_data
    }


def normalize_country_name(country_name):
    return country_name.replace("_", " ")


def get_event_name(event_slug):
    """
    Generate a name from the label.

    Returns title case with underscore replaced.
    """
    default = "Wiki Loves %s" % event_slug.replace("_", " ").title()
    return event_exceptions.get(event_slug.lower(), default)


def get_edition_name(scope_slug, year):
    default = "%s %s" % (get_event_name(scope_slug), year)
    return edition_exceptions.get((scope_slug.lower(), str(year)), default)


def get_instance_name(scope_slug, year, country):
    return "%s %s in %s" % (get_event_name(scope_slug), year, country)


def get_wikiloves_category_name(event_slug, year, country):
    if (event_slug, year, country) in special_exceptions:
        return special_exceptions[(event_slug, year, country)]

    edition = get_edition_name(event_slug, year)
    template = get_event_category_template()
    country_name = catExceptions.get(country, country)
    return template.format(edition=edition, country=country_name).replace(" ", "_")


def get_event_category_template():
    return "Images_from_{edition}_in_{country}"


event_exceptions = {
    "science": "Wiki Science Competition",
}

catExceptions = {
    "Netherlands": "the_Netherlands",
    "Central African Republic": "the_Central_African_Republic",
    "Comoros": "the_Comoros",
    "Czech Republic": "the_Czech_Republic",
    "Democratic Republic of the Congo": "the_Democratic_Republic_of_the_Congo",
    "Republic of the Congo": "the_Republic_of_the_Congo",
    "Dutch Caribbean": "the_Dutch_Caribbean",
    "Philippines": "the_Philippines",
    "Seychelles": "the_Seychelles",
    "United Arab Emirates": "the_United_Arab_Emirates",
    "United Kingdom": "the_United_Kingdom",
    "United States": "the_United_States",
}

edition_exceptions = {
    ("science", "2015"): "European_Science_Photo_Competition_2015",
}

special_exceptions = {
    (
        "Monuments",
        "2024",
        "Austria",
    ): "Media_from_WikiDaheim_2024_in_Austria/Cultural_heritage_monuments",
    (
        "Monuments",
        "2023",
        "Austria",
    ): "Media_from_WikiDaheim_2023_in_Austria/Cultural_heritage_monuments",
    (
        "Monuments",
        "2022",
        "Austria",
    ): "Media_from_WikiDaheim_2022_in_Austria/Cultural_heritage_monuments",
    (
        "Monuments",
        "2021",
        "Austria",
    ): "Media_from_WikiDaheim_2021_in_Austria/Cultural_heritage_monuments",
    (
        "Monuments",
        "2020",
        "Austria",
    ): "Media_from_WikiDaheim_2020_in_Austria/Cultural_heritage_monuments",
    (
        "Monuments",
        "2019",
        "Austria",
    ): "Media_from_WikiDaheim_2019_in_Austria/Cultural_heritage_monuments",
    (
        "Monuments",
        "2018",
        "Austria",
    ): "Media_from_WikiDaheim_2018_in_Austria/Cultural_heritage_monuments",
    (
        "Monuments",
        "2017",
        "Austria",
    ): "Media_from_WikiDaheim_2017_in_Austria/Cultural_heritage_monuments",
    (
        "Monuments",
        "2015",
        "Armenia",
    ): "Images_from_Wiki_Loves_Monuments_2015_in_Armenia_&_Nagorno-Karabakh",
    (
        "Monuments",
        "2017",
        "Armenia",
    ): "Images_from_Wiki_Loves_Monuments_2017_in_Armenia_&_Nagorno-Karabakh",
    ("Earth", "2014", "Armenia"): "Images_from_Wiki_Loves_Earth_2014_in_Armenia_&_Nagorno-Karabakh",
    (
        "Earth",
        "2021",
        "United Arab Emirates",
    ): "Images_from_Wiki_Loves_Earth_2021_in_United_Arab_Emirates",
    (
        "Earth",
        "2022",
        "United Arab Emirates",
    ): "Images_from_Wiki_Loves_Earth_2022_in_United_Arab_Emirates",
    (
        "Earth",
        "2022",
        "Democratic Republic of the Congo",
    ): "Images_from_Wiki_Loves_Earth_2022_in_DR_Congo",
    (
        "Earth",
        "2023",
        "Democratic Republic of the Congo",
    ): "Images_from_Wiki_Loves_Earth_2023_in_DR_Congo",
    (
        "Science",
        "2015",
        "Armenia",
    ): "Images_from_European_Science_Photo_Competition_2015_in_Armenia",
    ("Science", "2013", "Estonia"): "Images_from_Teadusfoto_2013",
    ("Science", "2012", "Estonia"): "Images_from_Teadusfoto_2012",
    ("Science", "2011", "Estonia"): "Images_from_Teadusfoto_2011",
}