demographic-manager-sanitized/enterer.py

import curses
import logging
import json
import sys

# logging.basicConfig(filename="enterer.log", level=logging.DEBUG)

attrtable = {
#   attr, sel[i]
    "a": 0,
    "c": 0,
    "m": 0,
    "f": 0,
    "u/o": 0,
    "h": 0,
    "nh": 0,
    "ai/an": 1,
    "as": 1,
    "b/aa": 1,
    "nh/opi": 3,
    "wh": 0,
    "1+": 0,
    "unk": 2,
}

age = ["a", "c"]
gender = ["m", "f", "u/o"]
ethnicity = ["h", "nh"]
race = ["ai/an", "as", "b/aa", "nh/opi", "wh", "1+", "unk"]

def serialize_finalsels_to_json(finalsels, filename):
    json_wrapper = {"payload": []}
    json_objs = []
#    logging.debug("lllllllllllll {}".format(finalsels))
    for i in finalsels:
        new_json_obj = {"age": None, "gender": None, "ethnicity": None, "race": None}
        for j in i[1]:
            if j in age:
                new_json_obj["age"] = j

            if j in gender:
                new_json_obj["gender"] = j

            if j in ethnicity:
                new_json_obj["ethnicity"] = j

            if j in race:
                new_json_obj["race"] = j

        for _ in range(i[0]):
            json_objs.append(new_json_obj)

    json_wrapper["payload"] = json_objs
#    logging.debug((json_objs, json_wrapper))
    with open(filename, "w", encoding="utf-8") as f:
        f.write(json.dumps(json_wrapper, indent=4))

## TODO: add padding for window elements
## TODO: add exportation
def pad_out_rest_of_line(window):
    window.refresh()
    height, width = window.getmaxyx()
    y, x = window.getyx()
    diff = width - x
#    logging.debug("lakfdjalskfjalskdfjalkfjsd {}".format(str(diff)))
    window.addstr(y, x, " " * diff, curses.A_REVERSE)

def check_if_attr_selected(attr, sel):
#    logging.debug(attrtable[attr])
#    logging.debug((attrtable[attr], attr[attrtable[attr]], sel))
    if attr[attrtable[attr]] == sel:
        return True
    return False

def equality_sel_check(attr, sel):
    return attr == sel

def generate_outstring_format_values(sels):
    output = []
    for i in attrtable.keys(): # will always be sorted, in order
        for j in sels:
            if check_if_attr_selected(i, j):
                output.append("[{}]".format(i))
                break
        if not "[{}]".format(i) in output:
            output.append("{}".format(i))

    return output

def convert_sels_arr_to_attrs(sels):
    output = []
    for i in attrtable.keys(): # will always be sorted, in order
        for j in sels:
#            logging.debug(str((i, j)))
            if check_if_attr_selected(i, j):
                output.append(i)
                break
    return output

def get_attr_cata_counts(sels):
    ages = 0
    genders = 0
    ethnicities = 0
    races = 0
    for i in sels:
        if i in age:
            ages += 1
        if i in gender:
            genders += 1
        if i in ethnicity:
            ethnicities += 1
        if i in race:
            races += 1

    return (ages, genders, ethnicities, races)

def resolve_sel_conflict(sels):
    final = convert_sels_arr_to_attrs(sels)
    ages, genders, ethnicities, races = get_attr_cata_counts(final)

#    logging.debug(final)
#    logging.debug((ages, genders, ethnicities, races))

    if ages > 1:
        for i in final:
            if i in age:
                final.remove(i)

    if genders > 1:
        for i in final:
            if i in gender:
                final.remove(i)

    if ethnicities > 1:
        for i in final:
            if i in ethnicity:
                final.remove(i)

    if races > 1:
        for i in final:
            if i in race:
                final.remove(i)

    finalfinal = []
    for i in final:
        finalfinal.append(i[attrtable[i]])

    return finalfinal

def main(stdscr):
    stdscr.clear()
    outputsels = []
    currentsels = []
    previoussels = []
    lines = 1
    count = ""
    errstring = ""
    status = ""
    outfile = sys.argv[1]

    while True:
        stdscr.addstr(0, 0, "enterer - normal mode - {}".format(status), curses.A_REVERSE)
        pad_out_rest_of_line(stdscr)
        stdscr.move(lines, 0)
        stdscr.clrtoeol()
        stdscr.addstr(lines, 0, "{}{} | {} {} | {} {} {} | {} {} | {} {} {} {} {} {} {}".format(
            errstring,
            count,
            *generate_outstring_format_values(currentsels)
        ))
        newsel = stdscr.getkey()

        if errstring != "":
            errstring = ""

        if status != "":
            status = ""

        if newsel == '\\':
            currentsels, previoussels = previoussels, currentsels
            continue

        if newsel in [str(i) for i in range(10)]:
            count += newsel
            continue


        if newsel in ('KEY_BACKSPACE', '\b', '\x7f'):
            count = count[:-1]
            continue

        if newsel == ']':
            final_output = []
            for i in outputsels:
                temp = convert_sels_arr_to_attrs(i[1])
                final_output.append([i[0], temp])

#            logging.debug("selsconverter" + str(final_output))
            try:
                serialize_finalsels_to_json(final_output, outfile)
            except FileNotFoundError:
                status = "failed to write. changing outfile to __RESCUE_OUTFILE.json. write again"
                outfile = "__RESCUE_OUTFILE.json"

            status = "written to {}".format(outfile)
            continue

        if newsel == ' ':
            catagories_data = get_attr_cata_counts(convert_sels_arr_to_attrs(currentsels))
            not_all_one = False
            for i in catagories_data:
                if not i == 1:
                    not_all_one = True
                    break

            if not_all_one:
                errstring += "!"
                continue

            if count == "":
                count = 1

            outputsels.append((int(count), currentsels))
            count = ""
            previoussels = currentsels
            currentsels = []
            lines += 1
            continue

        if not newsel in currentsels:
            currentsels.append(newsel)
        else:
            currentsels.remove(newsel)

        currentsels = resolve_sel_conflict(currentsels)


    stdscr.refresh()
    stdscr.getkey()

curses.wrapper(main)