diff --git a/packages/envied/src/envied/services/CBC/__init__.py b/packages/envied/src/envied/services/CBC/__init__.py index 3bfe4fb..de20c2e 100644 --- a/packages/envied/src/envied/services/CBC/__init__.py +++ b/packages/envied/src/envied/services/CBC/__init__.py @@ -230,12 +230,20 @@ class CBC(Service): # Service specific def _show(self, data: dict) -> list[Episode]: - lineups: list = next((x["lineups"] for x in data["content"] if x.get("title", "").lower() == "episodes"), None) - if not lineups: - try: - lineups = next(x["lineups"] for x in data["content"] if x.get("title", "").lower() == "parts") - except StopIteration: - pass + #lineups: list = next((x["lineups"] for x in data["content"] if x.get("title", "").lower() == "episodes"), None) + #if not lineups: + # try: + # lineups = next(x["lineups"] for x in data["content"] if x.get("title", "").lower() == "parts") + # except StopIteration: + # pass + lineups = next( + ( + x["lineups"] + for x in data["content"] + if x.get("title", "").lower() in ("episodes", "parts") + ), + None + ) if not lineups: self.log.warning("No episodes found for: {}".format(data.get("title")))