build.py

from lib import *

countries = {}
def init_countries():
    cj = load_json('countries')
    for country in cj:
        countries[country['alpha2_code'].lower()] = country
    countries['online'] = {}
    countries['online']['arabic_name'] = 'عن بعد'
    countries['online']['english_name'] = 'Online'

members = {}
members_j = load_json('people')
participations = load_json('participations')
olympiads_j = load_json('olympiads')
olympiads = {}

def init_members():
    global members
    for mem in members_j:
        members[mem['id']] = mem
        members[mem['id']]['participations'] = {}

    for oly in participations:
        for mem_id in oly['participants']:
            # Should give a fatal error or just create a new person with default data
            if mem_id not in members:
                members[mem_id] = {}
                members[mem_id]['participations'] = {}
                members[mem_id]['arname'] = mem_id
                members[mem_id]['enname'] = mem_id
                members[mem_id]['graduation'] = None
                members[mem_id]['codeforces'] = None
            members[mem_id]['participations'][oly['name'] + '_' + oly['start'].split('/')[0]] = oly['participants'][mem_id]

def init_olympiads():
    for oly in olympiads_j:
        olympiads[oly['id']] = oly
        olympiads[oly['id']]['gold'] = 0
        olympiads[oly['id']]['silver'] = 0
        olympiads[oly['id']]['bronze'] = 0
        olympiads[oly['id']]['hm'] = 0
        olympiads[oly['id']]['participations'] = 0

def build_participations():
    for oly in participations:
        oly['country_arname'] = countries[oly['country']]['arabic_name'] + ' ' + flag_emoji(oly['country'])
        oly['country_enname'] = countries[oly['country']]['english_name'] + ' ' + flag_emoji(oly['country'])
        olympiads[oly['name']]['participations'] += 1

        oly['arname'] = olympiads[oly['name']]['arname']
        oly['enname'] = olympiads[oly['name']]['enname']

        filename = oly['name'] + '_' + oly['start'].split('/')[0]
        parts = []
        enparts = []
        awards = ''
        for mem_id, award in oly['participants'].items():
            if award != None:
                olympiads[oly['name']][award] += 1
            parts.append({'id': mem_id, 'name': members[mem_id]['arname'], 'award': award_emoji(award, dashing_none=True)})
            enparts.append({'id': mem_id, 'name': members[mem_id]['enname'], 'award': award_emoji(award, dashing_none=True)})
            awards += award_emoji(award)

        oly['awards'] = awards
        write_file(f'participations/{filename}.html', {
            'layout': 'participation',
            'lang': 'ar',
            'title': oly['name'].upper() + ' ' + oly['start'].split('/')[0],
            'olympiad': oly['name'],
            'olympiad_arname': oly['arname'],
            'start_date': oly['start'],
            'end_date': oly['end'],
            'country_arname': oly['country_arname'],
            'country_enname': oly['country_enname'],
            'participants': parts,
            'website': oly['website']
        })
        write_file(f'en/participations/{filename}.html', {
            'layout': 'participation',
            'lang': 'en',
            'title': oly['name'].upper() + ' ' + oly['start'].split('/')[0],
            'olympiad': oly['name'],
            'olympiad_enname': oly['enname'],
            'country_arname': oly['country_arname'],
            'country_enname': oly['country_enname'],
            'start_date': oly['start'],
            'end_date': oly['end'],
            'participants': enparts,
            'website': oly['website']
        })

def build_members():
    for memid, mem in members.items():
        participations = []
        for oly, award in mem['participations'].items():
            olymp = {}
            olymp['olympiad'] = oly.split("_")[0]
            olymp['year'] = oly.split("_")[1]
            olymp['award'] = award
            
            participations.append(olymp)

        write_file(f'members/{memid}.html', {
            'layout': 'person',
            'title': mem['arname'],
            'lang': 'ar',
            'full_name': mem['arname'],
            'graduation': mem['graduation'],
            'codeforces': mem['codeforces'],
            'participations': participations
        })
        write_file(f'en/members/{memid}.html', {
            'layout': 'person',
            'title': mem['enname'],
            'lang': 'en',
            'full_name': mem['enname'],
            'graduation': mem['graduation'],
            'codeforces': mem['codeforces'],
            'participations': participations
        })

def build_participations_index():
    olympiads = {}
    min_year = 3000
    max_year = 2000
    for oly in participations:
        year = oly['start'].split('/')[0]
        min_year = min(int(year), min_year)
        max_year = max(int(year), max_year)
        if year not in olympiads:
            olympiads[year] = []

        del oly["participants"]
        olympiads[year].append(oly)

    written = {
        'layout': 'participations',
        'lang': 'ar',
        'title': 'المشاركات',
        'start_year': min_year,
        'last_year': max_year
    }

    for year, list in olympiads.items():
        written[year] = list
    
    write_file("participations/index.html", written)

    written['lang'] = 'en'
    written['title'] = 'Participations'

    write_file("en/participations/index.html", written)

def build_hall_of_fame():
    official_olympiads = []
    for oly in olympiads_j:
        if oly['official']:
            official_olympiads.append(oly['id'])
    fame = {}
    for memid, data in members.items():
        fame[memid] = {
            'gold': 0,
            'silver': 0,
            'bronze': 0,
            'hm': 0,
            None: 0,
        }
        for oly, award in data['participations'].items():
            oly = oly.split('_')[0]
            if oly in official_olympiads:
                fame[memid][award] += 1
    
    filtered_fame = filter(lambda dic: dic[1]['gold'] + dic[1]['silver'] + dic[1]['bronze'] + dic[1]['hm'] > 0, fame.items())
    sorted_fame = sorted(filtered_fame, key=lambda x: (-x[1]['gold'], -x[1]['silver'], -x[1]['bronze'], -x[1]['hm']))
    # TODO: should be printed to a file

    written = []
    for it in sorted_fame:
        mem = it[0]
        dic = it[1]
        written.append({
            'id': mem,
            'arname': members[mem]['arname'],
            'enname': members[mem]['enname'],
            'gold': dic['gold'],
            'silver': dic['silver'],
            'bronze': dic['bronze'],
            'hm': dic['hm'],
        })
    
    write_file('./hall-of-fame.html', {
        'layout': 'halloffame',
        'lang': 'ar',
        'title': 'لائحة الشرف',
        'list': written,
    })

    write_file('en/hall-of-fame.html', {
        'layout': 'halloffame',
        'lang': 'en',
        'title': 'Fame',
        'list': written,
    })

def build_images():
    imgs = load_json('images')
    count = len(imgs)
    written = {
        'layout': 'images',
        'title': "مكتبة الصور",
        'lang': 'ar',
        'count': count
    }
    idx = 1
    for img in imgs:
        written[idx] = img
        idx += 1
    write_file('./images.html', written)
    written['lang'] = 'en'
    written['title'] = 'Image library'
    write_file('en/images.html', written)

def build_contact():
    contact = load_json('contact')
    realcontact = {
        'maintainers': [],
        'developers': [],
        'admins': [],
    }
    for mem in contact['maintainers']:
        try:
            del members[mem]['participations_count']
        except:
            ...
        try:
            del members[mem]['graduation']
        except:
            ...
        try:
            del members[mem]['codeforces']
        except:
            ...
        try:
            del members[mem]['level']
        except:
            ...
        realcontact['maintainers'].append(members[mem])
        if members[mem]['email'] == None:
            exit(1)

    for mem in contact['developers']:
        try:
            del members[mem]['participations_count']
        except:
            ...
        try:
            del members[mem]['graduation']
        except:
            ...
        try:
            del members[mem]['codeforces']
        except:
            ...
        try:
            del members[mem]['level']
        except:
            ...
        realcontact['developers'].append(members[mem])
        if members[mem]['email'] == None:
            exit(1)

    for mem in contact['admins']:
        try:
            del members[mem]['participations_count']
        except:
            ...
        try:
            del members[mem]['graduation']
        except:
            ...
        try:
            del members[mem]['codeforces']
        except:
            ...
        try:
            del members[mem]['level']
        except:
            ...
        realcontact['admins'].append(members[mem])
        if members[mem]['email'] == None:
            exit(1)

    written = {
        'layout': 'contact',
        'lang': 'ar',
        'title': 'تواصل',
        'maintainers': realcontact['maintainers'],
        'admins': realcontact['admins'],
        'developers': realcontact['developers']
    }
    write_file('./contact.html', written)
    written['lang'] = 'en'
    written['title'] = 'Contact'
    write_file('en/contact.html', written)


def test_utils():
    assert flag_emoji('sa') == '🇸🇦'
    assert countries['sa']['english_name'] == 'Saudi Arabia'
    assert write_yml({
        'Hi': 'Hello',
        'list': ['Hi', 'Hello'],
        'dict': {
            'sub1': 'sub2'
        },
        'nullval': None
    }) == """Hi: "Hello"
list:
  count: 2
  1: "Hi"
  2: "Hello"
dict:
  sub1: "sub2"
nullval: null
"""

def build_members_index():
    members_list = members.values()
    levels = {1: [], 2: [], 3: [], 4: []}
    for mem in members_list:
        mem['participations_count'] = len(mem['participations'])
        del mem['participations']
        if 1 <= mem['level'] <= 4:
            levels[mem['level']].append(mem)

    data = {
        'lang': 'ar',
        'title': 'قائمة الأعضاء',
        'layout': 'members',
        'levels': levels
    }
    write_file("./members/index.html", data)
    data['lang'] = 'en'
    data['title'] = 'Members list'
    write_file("en/members/index.html", data)

def build_olympiads():
    for oly in olympiads_j:
        oly['gold'] = olympiads[oly['id']]['gold']
        oly['silver'] = olympiads[oly['id']]['silver']
        oly['bronze'] = olympiads[oly['id']]['bronze']
        oly['hm'] = olympiads[oly['id']]['hm']
        oly['participations'] = olympiads[oly['id']]['participations']
    write_file('./olympiads.html', {
        'title': 'الأولمبيادات',
        'layout': 'olympiads',
        'lang': 'ar',
        'olympiads': olympiads_j
    })
    write_file('en/olympiads.html', {
        'title': 'Olympiads',
        'layout': 'olympiads',
        'lang': 'en',
        'olympiads': olympiads_j
    })

def main():
    init_countries()
    init_members()
    init_olympiads()
    print("Init: OK")
    test_utils()

    # Pls don't change the order
    build_members()
    print("Built members")

    build_participations()
    print("Built participations")

    build_participations_index()
    print("Built participations index")

    build_hall_of_fame()
    print("Built hall of fame")

    build_images()
    print("Built participations index")

    build_olympiads()
    print("Built olympiads")

    # build_calendar()
    build_members_index()
    print("Built members index")

    build_contact()
    print("Built contact")

if __name__ == '__main__':
    main()