klette/main.py

#!/usr/bin/python3

import pyamf
import shutil
import json
import os
import argparse
import unicodedata
import random
import re
import sys
import readline

DATA_DIR = os.environ['XDG_DATA_HOME'] + '/klette'
VOKABELN_FILE = DATA_DIR + '/vokabeln.json'
STATUS_FILE = DATA_DIR + '/status.json'
AUDIO_BASE = DATA_DIR + '/audio/'

class bcolors:
    HEADER = '\033[95m'
    OKBLUE = '\033[94m'
    OKCYAN = '\033[96m'
    OKGREEN = '\033[92m'
    WARNING = '\033[93m'
    FAIL = '\033[91m'
    ENDC = '\033[0m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'
    LINK_START = '\033]8;;'
    LINK_MIDDLE = '\033\\'
    LINK_END = '\033]8;;\033\\'

def import_vokabeln(file_name, audio_base):
    os.makedirs(AUDIO_BASE, exist_ok=True)
    palabras = []
    with open(file_name, 'rb') as reader:
        obj = pyamf.decode(reader.read())
        for unidad_data in obj.readElement()['lektionList']:
            unidad = unidad_data['name']
            for paso_data in unidad_data['lektionsTeilList']:
                paso = paso_data['name']
                for palabra in paso_data['entryList']:
                    audio = palabra['vorderSeite']['swTonAbgleichwort']
                    palabras.append({
                            'id': palabra['matId'],
                            'de': palabra['rueckSeite']['text'],
                            'es': palabra['vorderSeite']['text'],
                            'audio': audio,
                            'unidad': unidad,
                            'paso': paso
                    })
                    shutil.copy2(audio_base + audio + '.bin',
                                 AUDIO_BASE + audio + '.aac')
    with open(VOKABELN_FILE, 'x') as writer:
        writer.write(json.dumps(palabras))

def import_status(file_name):
    os.makedirs(DATA_DIR, exist_ok=True)
    statuses = {}
    with open(file_name, 'rb') as reader:
        obj = pyamf.decode(reader.read()).readElement()[0]
        for idx in obj:
            item = obj[idx]
            statuses[item['matId']] = item['statusWortschatz']
    with open(STATUS_FILE, 'x') as writer:
        writer.write(json.dumps(statuses))

def vergleiche(respuesta, expecta):
    if respuesta == "": return 4
    if respuesta[0] == '!' and expecta[0] == '¡':
        respuesta = '¡' + respuesta[1:]
    if respuesta[0] == '?' and expecta[0] == '¿':
        respuesta = '¿' + respuesta[1:]
    expecta = re.sub(' \((irr.|pl.)\)', '', expecta)
    respuesta = unicodedata.normalize('NFD', respuesta)
    expecta = unicodedata.normalize('NFD', expecta)
    if respuesta == expecta: return 0
    respuesta = re.sub('[!?¿¡…. ]', '', respuesta)
    expecta = re.sub('[!?¿¡…. ]', '', expecta)
    if respuesta == expecta: return 1
    respuesta = respuesta.encode('ASCII', 'ignore').decode('ASCII')
    expecta = expecta.encode('ASCII', 'ignore').decode('ASCII')
    if respuesta == expecta: return 2
    respuesta = respuesta.casefold()
    expecta = expecta.casefold()
    if respuesta == expecta: return 3
    return 4

assert vergleiche("e", "e") == 0
assert vergleiche("é", "e") == 2
assert vergleiche("Bien", "bien") == 3
assert vergleiche("quétal", "Qué tal") == 3
assert vergleiche("Buenas noches", "¡Buenas noches!") == 1
assert vergleiche("ser", "ser (irr.)") == 0
assert vergleiche("hacer algo", "hacer (irr.) algo") == 0
assert vergleiche("padres", "padres (pl.)") == 0

class Resultado:
    BIEN = 1
    MAL = 2
    ADIOS = 3

def einzelne_abfrage(palabra, status):
    status_text = f"({status})" if status > 0 else f"{bcolors.OKGREEN}neu{bcolors.ENDC}"
    try:
        respuesta = input(f"{status_text} {bcolors.BOLD}{palabra['de']}{bcolors.ENDC} {bcolors.OKCYAN}")
    except EOFError:
        print()
        return Resultado.ADIOS
    finally:
        print(bcolors.ENDC, end="")
    print('\033[{}C\033[1A'.format(4 + len(palabra['de']) + 1 + len(respuesta)), end="")
    resultado = vergleiche(respuesta, palabra['es'])
    match status:
        case 0: bien = resultado < 4
        case 1: bien = resultado < 3
        case 2: bien = resultado < 3
        case 3: bien = resultado < 2
        case 4: bien = resultado < 1
    print(f" {bcolors.LINK_START}file://{AUDIO_BASE}{palabra['audio']}.aac{bcolors.LINK_MIDDLE}",end="")
    if bien and resultado == 0:
        print(f"{bcolors.OKGREEN}✓{bcolors.ENDC}",end="")
    elif bien:
        print(f"{bcolors.WARNING}{palabra['es']}{bcolors.ENDC}",end="")
    else:
        print(f"{bcolors.FAIL}{palabra['es']}{bcolors.ENDC}",end="")
    print(bcolors.LINK_END,end="")
    try:
        input(" ")
    except EOFError:
        print()
        return Resultado.ADIOS
    return Resultado.BIEN if bien else Resultado.MAL

def abfrage(parser, quiero_unidad):
    random.seed()
    try:
        with open(VOKABELN_FILE, 'r') as f:
            palabras = json.load(f)
        with open(STATUS_FILE, 'r') as f:
            status = json.load(f)
    except FileNotFoundError:
        print(f"{bcolors.FAIL}Daten können nicht geladen werden, hast du sie schon importiert?{bcolors.ENDC}")
        print()
        parser.print_help()
        return

    unidad = False
    unidad_no = -1
    paso = False
    cur_palabras = False
    bien = 0
    mal = 0
    for palabra in palabras:
        if paso != palabra['paso']:
            if cur_palabras != False and (quiero_unidad == None or quiero_unidad == unidad_no):
                c = [len(cur_palabras[x]) for x in cur_palabras]
                print(f"{bcolors.BOLD}{unidad}{bcolors.ENDC}: {paso} (",end="")
                print(*c, sep="/", end=")")
                print()
                for n in range(5): # 1..4, no 5
                    random.shuffle(cur_palabras[n])
                    for a_palabra in cur_palabras[n]:
                        s = status.get(a_palabra['id'], 0)
                        match einzelne_abfrage(a_palabra, s):
                            case Resultado.BIEN:
                                status[a_palabra['id']] = s + 1
                                bien+=1
                            case Resultado.MAL:
                                if s > 0:
                                    status[a_palabra['id']] = s - 1
                                    mal+=1
                                    continue
                            case Resultado.ADIOS:
                                print(f'{bcolors.OKGREEN}+{bien}{bcolors.ENDC} / {bcolors.FAIL}-{mal}{bcolors.ENDC}')
                                return
                        with open(STATUS_FILE + '.new', 'w') as f:
                            json.dump(status, f)
                            f.flush()
                            os.fsync(f.fileno())
                        with open(STATUS_FILE + '.new', 'r') as f:
                            status = json.load(f)
                        os.replace(STATUS_FILE + '.new', STATUS_FILE)

            cur_palabras = {0: [], 1: [], 2: [], 3: [], 4: [], 5: []}
            if unidad != palabra['unidad']:
                unidad_no += 1
            unidad = palabra['unidad']
            paso = palabra['paso']

        cur_palabras[status.get(palabra['id'], 0)].append(palabra)

parser = argparse.ArgumentParser()
default_data_file = os.environ['PWD'] + '/assets/amf/vokabelTrainer147.amf'
parser.add_argument('--import-data', type=str, help="Path to assets", metavar="DIR")
default_status_file = os.environ['HOME'] + '/klett/1266/vokabeltrainerData147'
parser.add_argument('--import-status', type=str, help="Path to AMF File, defaults to " + default_status_file, metavar="FILE", nargs='?', const=default_status_file)
parser.add_argument('--unidad', type=int)
args = parser.parse_args()

if args.import_data:
    import_vokabeln(args.import_data + '/amf/vokabelTrainer147.amf',
                    args.import_data + '/amf/medium/')
elif args.import_status:
    import_status(args.import_status)
else:
    abfrage(parser, args.unidad)