Skip to content

Commit

Permalink
Fix linting
Browse files Browse the repository at this point in the history
  • Loading branch information
downey-lv committed Jun 4, 2024
1 parent 588f88f commit f8c1336
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 12 deletions.
11 changes: 2 additions & 9 deletions custom_components/trakt_tv/const.py
Original file line number Diff line number Diff line change
Expand Up @@ -172,13 +172,6 @@
"zu",
]

SORT_BY_OPTIONS = [
"rating",
"rating_trakt",
"rank",
"runtime",
"released",
"listed_at"
]
SORT_BY_OPTIONS = ["rating", "rating_trakt", "rank", "runtime", "released", "listed_at"]

SORT_HOW_OPTIONS = ["asc", "desc"]
SORT_HOW_OPTIONS = ["asc", "desc"]
8 changes: 6 additions & 2 deletions custom_components/trakt_tv/models/media.py
Original file line number Diff line number Diff line change
Expand Up @@ -283,14 +283,18 @@ def to_homeassistant(self) -> Dict[str, Any]:
class Medias:
items: List[Media]

def to_homeassistant(self, sort_by = 'released', sort_order = 'asc') -> Dict[str, Any]:
def to_homeassistant(self, sort_by="released", sort_order="asc") -> Dict[str, Any]:
"""
Convert the List of medias to recommendation data.
:return: The dictionary containing all necessary information for upcoming media
card
"""
medias = sorted(self.items, key=lambda media: getattr(media, sort_by), reverse=sort_order == "desc")
medias = sorted(
self.items,
key=lambda media: getattr(media, sort_by),
reverse=sort_order == "desc",
)
medias = [media.to_homeassistant() for media in medias]
return [first_item] + medias

Expand Down
7 changes: 6 additions & 1 deletion custom_components/trakt_tv/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -93,8 +93,13 @@ def cache_retrieve(cache: Dict[str, Any], key: str) -> Optional[Any]:
else:
return None


def parse_utc_date(date_str: Optional[str]) -> Optional[datetime]:
"""
Parse an ISO date string (all dates returned from Trakt) to a datetime object.
"""
return datetime.fromisoformat(date_str).replace(tzinfo=timezone.utc) if date_str else None
return (
datetime.fromisoformat(date_str).replace(tzinfo=timezone.utc)
if date_str
else None
)

0 comments on commit f8c1336

Please sign in to comment.