enterer tool added

This commit is contained in:
stupidcomputer 2024-06-24 04:57:31 -05:00
parent 8fccf4a397
commit 3842b6c3b7
1 changed files with 241 additions and 0 deletions

241
enterer.py Normal file
View File

@ -0,0 +1,241 @@
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)