From f9cfc311fa8adfeeee6d2f464c1ae59f4c669b4e Mon Sep 17 00:00:00 2001 From: Rilic Date: Sun, 16 Jul 2023 16:22:28 +0100 Subject: [PATCH] Simplify playlist file location code --- src/importer/sources/retroarch_source.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/src/importer/sources/retroarch_source.py b/src/importer/sources/retroarch_source.py index e9a460e..ec78a52 100644 --- a/src/importer/sources/retroarch_source.py +++ b/src/importer/sources/retroarch_source.py @@ -34,10 +34,7 @@ class RetroarchSourceIterator(SourceIterator): source: "RetroarchSource" def generator_builder(self) -> SourceIterationResult: - playlist_files = [] - for file in os.listdir(self.source.config_location["playlists"]): - if file.endswith(".lpl"): - playlist_files.append(file) + playlist_files = self.source.config_location["playlists"].glob("*.lpl") for playlist_file in playlist_files: try: @@ -82,7 +79,7 @@ class RetroarchSourceIterator(SourceIterator): # Get boxart boxart_image_name = item["label"].split(".", 1)[0] + ".png" - boxart_folder_name = playlist_file.split(".", 1)[0] + boxart_folder_name = playlist_file.name.split(".", 1)[0] image_path = ( self.source.config_location["thumbnails"] / boxart_folder_name