|
|
|
|
|
|
|
|
|
|
|
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 [] |
|
|
|
