import requests
import json
import time
import os


def get_vocabulary(level_id: str):
    body = {"id": level_id, "type": "builtin"}
    response = requests.post(
        "https://api.relingo.net/api/getVocabulary", json=body)
    response.raise_for_status()
    return response.json()


def process(level_id: str):
    vocabulary = get_vocabulary(level_id)
    if 'data' in vocabulary and 'words' in vocabulary['data']:
        words = vocabulary['data']['words']
        print(f'size of {level_id}: {len(words)}')
        sorted_words = sorted(words)
        # write json file in dated directory
        cur_time = time.localtime()
        date_named_dir = f'{cur_time.tm_year}-{cur_time.tm_mon}-{cur_time.tm_mday}'
        if not os.path.exists(date_named_dir):
            os.makedirs(date_named_dir)
        with open(f'{date_named_dir}/level-{level_id}.json', 'w') as json_file:
            json.dump(sorted_words, json_file, indent=4, ensure_ascii=True)


# all levels' dictionary names
levels = ['Beginner', 'Intermediate', 'Advanced', 'Proficient',
          'GRE', 'IELTS', 'TOEFL', 'SAT', 'CET4', 'CET6', 'KaoYan',
          'GMAT', 'Medical']
for level in levels:
    process('relingo-' + level)