Plex
/
Plex-Meta-Manager
mirror of https://github.com/meisnate12/Plex-Meta-Manager.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix templates

Browse Source
pull/240/head
meisnate12 4 years ago
parent a0963181dc
commit 69b8403309
1 changed files with 58 additions and 61 deletions
Show Stats Download Patch File Download Diff File

119
modules/builder.py
Unescape View File

@ -209,23 +209,25 @@ class CollectionBuilder:
elif not self.data[methods["template"]]: elif not self.data[methods["template"]]:
raise Failed("Collection Error: template attribute is blank") raise Failed("Collection Error: template attribute is blank")
else: else:
for data_template in util.get_list(self.data[methods["template"]], split=False): for variables in util.get_list(self.data[methods["template"]], split=False):
if not isinstance(data_template, dict): if not isinstance(variables, dict):
raise Failed("Collection Error: template attribute is not a dictionary") raise Failed("Collection Error: template attribute is not a dictionary")
elif "name" not in data_template: elif "name" not in variables:
raise Failed("Collection Error: template sub-attribute name is required") raise Failed("Collection Error: template sub-attribute name is required")
elif not data_template["name"]: elif not variables["name"]:
raise Failed("Collection Error: template sub-attribute name is blank") raise Failed("Collection Error: template sub-attribute name is blank")
elif data_template["name"] not in self.metadata.templates: elif variables["name"] not in self.metadata.templates:
raise Failed(f"Collection Error: template {data_template['name']} not found") raise Failed(f"Collection Error: template {variables['name']} not found")
elif not isinstance(self.metadata.templates[data_template["name"]], dict): elif not isinstance(self.metadata.templates[variables["name"]], dict):
raise Failed(f"Collection Error: template {data_template['name']} is not a dictionary") raise Failed(f"Collection Error: template {variables['name']} is not a dictionary")
else: else:
for tm in data_template: for tm in variables:
if not data_template[tm]: if not variables[tm]:
raise Failed(f"Collection Error: template sub-attribute {tm} is blank") raise Failed(f"Collection Error: template sub-attribute {tm} is blank")
if "collection_name" not in variables:
variables["collection_name"] = str(self.name)
template_name = data_template["name"] template_name = variables["name"]
template = self.metadata.templates[template_name] template = self.metadata.templates[template_name]
default = {} default = {}
@ -256,60 +258,55 @@ class CollectionBuilder:
else: else:
raise Failed("Collection Error: template sub-attribute optional is blank") raise Failed("Collection Error: template sub-attribute optional is blank")
for method_name, attr_data in template.items(): def check_data(_data):
if method_name not in self.data and method_name not in ["default", "optional"]: if isinstance(_data, dict):
if attr_data is not None: final_data = {}
def replace_txt(txt): for sm, sd in _data.items():
txt = str(txt)
for option in optional:
if option not in data_template and f"<<{option}>>" in txt:
raise Failed("remove attribute")
for template_method in data_template:
if template_method != "name" and txt == f"<<{template_method}>>":
return data_template[template_method]
elif template_method != "name" and f"<<{template_method}>>" in txt:
txt = txt.replace(f"<<{template_method}>>", str(data_template[template_method]))
if "<<collection_name>>" in txt:
txt = txt.replace("<<collection_name>>", str(self.name))
for dm in default:
if txt == f"<<{dm}>>":
txt = default[dm]
elif f"<<{dm}>>" in txt:
txt = txt.replace(f"<<{dm}>>", str(default[dm]))
if txt in ["true", "True"]: return True
elif txt in ["false", "False"]: return False
else:
try: return int(txt)
except (ValueError, TypeError): return txt
try: try:
if isinstance(attr_data, dict): final_data[sm] = check_data(sd)
final_data = {}
for sm in attr_data:
if isinstance(attr_data[sm], list):
temp_list = []
for li in attr_data[sm]:
temp_list.append(replace_txt(li))
final_data[sm] = temp_list
else:
final_data[sm] = replace_txt(attr_data[sm])
elif isinstance(attr_data, list):
final_data = []
for li in attr_data:
if isinstance(li, dict):
temp_dict = {}
for sm in li:
temp_dict[sm] = replace_txt(li[sm])
final_data.append(temp_dict)
else:
final_data.append(replace_txt(li))
else:
final_data = replace_txt(attr_data)
except Failed: except Failed:
continue continue
self.data[method_name] = final_data elif isinstance(_data, list):
methods[method_name.lower()] = method_name final_data = []
for li in _data:
try:
final_data.append(check_data(li))
except Failed:
continue
else:
txt = str(_data)
def idk_yet(og_txt, var, var_value):
if og_txt == f"<<{var}>>":
return var_value
elif f"<<{var}>>" in str(og_txt):
return str(og_txt).replace(f"<<{var}>>", str(var_value))
else:
return og_txt
for option in optional:
if option not in variables and f"<<{option}>>" in txt:
raise Failed
for variable, variable_data in variables.items():
if variable != "name":
txt = idk_yet(txt, variable, variable_data)
for dm, dd in default.items():
txt = idk_yet(txt, dm, dd)
if txt in ["true", "True"]: final_data = True
elif txt in ["false", "False"]: final_data = False
else: else:
raise Failed(f"Collection Error: template attribute {method_name} is blank") try: final_data = int(txt)
except (ValueError, TypeError): final_data = txt
return final_data
for method_name, attr_data in template.items():
if method_name not in self.data and method_name not in ["default", "optional"]:
if attr_data is None:
logger.error(f"Template Error: template attribute {method_name} is blank")
continue
try:
self.data[method_name] = check_data(attr_data)
methods[method_name.lower()] = method_name
except Failed:
continue
skip_collection = True skip_collection = True
if "schedule" not in methods: if "schedule" not in methods:

Write Preview
Loading…
Cancel
Save