enterer tool added
This commit is contained in:
parent
8fccf4a397
commit
3842b6c3b7
|
|
@ -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)
|
||||
Loading…
Reference in New Issue