56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
#
|
|
#
|
|
#
|
|
import os
|
|
from csv import DictReader
|
|
|
|
from modules import scripts
|
|
|
|
|
|
CSV_FILE_PATH = "csv/preset.tsv"
|
|
MYPRESET_PATH = "csv/preset_own.tsv"
|
|
HEADER = ["preset_name", "preset_weights"]
|
|
path_root = scripts.basedir()
|
|
|
|
|
|
class PresetWeights():
|
|
def __init__(self):
|
|
self.presets = {}
|
|
|
|
if os.path.exists(os.path.join(path_root, MYPRESET_PATH)):
|
|
with open(os.path.join(path_root, MYPRESET_PATH), "r") as f:
|
|
reader = DictReader(f, delimiter="\t")
|
|
lines_dict = [row for row in reader]
|
|
for line_dict in lines_dict:
|
|
_w = ",".join([f"{x.strip()}" for x in line_dict["preset_weights"].split(",")])
|
|
self.presets.update({line_dict["preset_name"]: _w})
|
|
|
|
with open(os.path.join(path_root, CSV_FILE_PATH), "r") as f:
|
|
reader = DictReader(f, delimiter="\t")
|
|
lines_dict = [row for row in reader]
|
|
for line_dict in lines_dict:
|
|
_w = ",".join([f"{x.strip()}" for x in line_dict["preset_weights"].split(",")])
|
|
self.presets.update({line_dict["preset_name"]: _w})
|
|
|
|
def get_preset_name_list(self):
|
|
return [k for k in self.presets.keys()]
|
|
|
|
def find_weight_by_name(self, preset_name=""):
|
|
if preset_name and preset_name != "" and preset_name in self.presets.keys():
|
|
return self.presets.get(preset_name, ",".join(["0.5" for _ in range(25)]))
|
|
else:
|
|
return ""
|
|
|
|
def find_names_by_weight(self, weights=""):
|
|
if weights and weights != "":
|
|
if weights in self.presets.values():
|
|
return [k for k, v in self.presets.items() if v == weights]
|
|
else:
|
|
_val = ",".join([f"{x.strip()}" for x in weights.split(",")])
|
|
if _val in self.presets.values():
|
|
return [k for k, v in self.presets.items() if v == _val]
|
|
else:
|
|
return []
|
|
else:
|
|
return []
|