Mirrors/Plex-Meta-Manager
2
0
Fork 0
mirror of https://github.com/meisnate12/Plex-Meta-Manager synced 2024-11-29 07:50:24 +00:00
Code Issues Projects Releases Packages Wiki Activity

[15] add load_templates

Browse source
This commit is contained in:
meisnate12 2022-04-09 21:45:33 -04:00
parent 2adab8d700
commit d9ff00d46b
2 changed files with 54 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
VERSION
View file

@ -1 +1 @@
1.16.3-develop14
1.16.3-develop15

62
modules/meta.py
View file

@ -74,20 +74,20 @@ class DataFile:
else:
return data
def load_file(self):
def load_file(self, file_type, file_path):
try:
if self.type in ["URL", "Git", "Repo"]:
if self.type == "Repo" and not self.config.custom_repo:
if file_type in ["URL", "Git", "Repo"]:
if file_type == "Repo" and not self.config.custom_repo:
raise Failed("Config Error: No custom_repo defined")
content_path = self.path if self.type == "URL" else f"{self.config.custom_repo if self.type == 'Repo' else github_base}{self.path}.yml"
content_path = file_path if file_type == "URL" else f"{self.config.custom_repo if file_type == 'Repo' else github_base}{file_path}.yml"
response = self.config.get(content_path)
if response.status_code >= 400:
raise Failed(f"URL Error: No file found at {content_path}")
content = response.content
elif os.path.exists(os.path.abspath(self.path)):
content = open(self.path, encoding="utf-8")
elif os.path.exists(os.path.abspath(file_path)):
content = open(file_path, encoding="utf-8")
else:
raise Failed(f"File Error: File does not exist {os.path.abspath(self.path)}")
raise Failed(f"File Error: File does not exist {os.path.abspath(file_path)}")
data, _, _ = yaml.util.load_yaml_guess_indent(content)
return data
except yaml.scanner.ScannerError as ye:
@ -235,6 +235,47 @@ class DataFile:
continue
return new_attributes
def load_templates(self):
if "load_templates" in self.templates and self.templates["load_templates"]:
template_files = []
for template_file in util.get_list(self.templates["load_templates"], split=False):
if isinstance(template_file, dict):
temp_vars = {}
if "template_variables" in template_file and template_file["template_variables"] and isinstance(template_file["template_variables"], dict):
temp_vars = template_file["template_variables"]
def check_dict(attr, name):
if attr in template_file:
if template_file[attr] is None:
logger.error(f"Config Error: metadata_path {attr} is blank")
else:
template_files.append((name, template_file[attr], temp_vars))
check_dict("url", "URL")
check_dict("git", "Git")
check_dict("repo", "Repo")
check_dict("file", "File")
if "folder" in template_file:
if template_file["folder"] is None:
logger.error(f"Config Error: metadata_path folder is blank")
elif not os.path.isdir(template_file["folder"]):
logger.error(f"Config Error: Folder not found: {template_file['folder']}")
else:
yml_files = util.glob_filter(os.path.join(template_file["folder"], "*.yml"))
if yml_files:
template_files.extend([("File", yml, temp_vars) for yml in yml_files])
else:
logger.error(f"Config Error: No YAML (.yml) files found in {template_file['folder']}")
else:
template_files.append(("File", template_file, {}))
for file_type, template_file, temp_vars in template_files:
temp_data = self.load_file(file_type, template_file)
if temp_data and isinstance(temp_data, dict) and "templates" in temp_data and temp_data["templates"] and isinstance(temp_data["templates"], dict):
for temp_key, temp_value in temp_data["templates"].items():
if temp_key not in self.templates:
self.templates[temp_key] = temp_value
for tk, tv in temp_vars.items():
if tk not in self.temp_vars:
self.temp_vars[tk] = tv
self.templates.pop("load_templates")
class MetadataFile(DataFile):
def __init__(self, config, library, file_type, path, temp_vars):
@ -249,9 +290,11 @@ class MetadataFile(DataFile):
logger.info("")
logger.info(f"Loading Metadata {file_type}: {path}")
logger.info("")
data = self.load_file()
data = self.load_file(self.type, self.path)
self.metadata = get_dict("metadata", data, library.metadata_files)
self.templates = get_dict("templates", data)
self.load_templates()
self.collections = get_dict("collections", data, library.collections)
self.dynamic_collections = get_dict("dynamic_collections", data)
col_names = library.collections + [c for c in self.collections]
@ -1057,9 +1100,10 @@ class PlaylistFile(DataFile):
self.playlists = {}
logger.info("")
logger.info(f"Loading Playlist File {file_type}: {path}")
data = self.load_file()
data = self.load_file(self.type, self.path)
self.playlists = get_dict("playlists", data, self.config.playlist_names)
self.templates = get_dict("templates", data)
self.load_templates()
if not self.playlists:
raise Failed("YAML Error: playlists attribute is required")
logger.info(f"Playlist File Loaded Successfully")