Skip to content

Commit

Permalink
Merge pull request #1819 from glensc/imdb_import-skip-rating
Browse files Browse the repository at this point in the history
Fix: Skip rating over imdb_import items with same value
  • Loading branch information
glensc authored Feb 22, 2024
2 parents 23d669a + 7687d50 commit ab4bb16
Showing 1 changed file with 16 additions and 5 deletions.
21 changes: 16 additions & 5 deletions plextraktsync/commands/imdb_import.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
from __future__ import annotations

import csv
from dataclasses import dataclass
from dataclasses import dataclass, fields
from functools import cached_property
from typing import TYPE_CHECKING

Expand Down Expand Up @@ -43,23 +43,31 @@ class Ratings:
# 'Directors': 'directors',
}

def __post_init__(self):
# cast "int" fields
fieldnames = [f.name for f in fields(self) if f.type == "int"]
for name in fieldnames:
value = self.__dict__[name]
if value is not None and not isinstance(value, int):
self.__dict__[name] = int(value)

@cached_property
def media_type(self):
if self.type == "tvSeries":
return "show"

return self.type

@staticmethod
def from_csv(row):
mapping = Ratings.FIELD_MAPPING
@classmethod
def from_csv(cls, row):
mapping = cls.FIELD_MAPPING
data = {}
for k, v in row.items():
if k not in mapping:
continue
data[mapping[k]] = v

return Ratings(**data)
return cls(**data)


def imdb_import(input: PathLike, dry_run: bool):
Expand All @@ -70,6 +78,9 @@ def imdb_import(input: PathLike, dry_run: bool):
print(f"Importing [blue]{r.media_type} {r.imdb}[/]: {r.title} ({r.year}), rated at {r.rate_date}")
m = trakt.search_by_id(r.imdb, "imdb", r.media_type)
rating = trakt.rating(m)
if r.rating == rating:
print(f"Rating {rating} already exists")
continue
print(f"{'Would rate' if dry_run else 'Rating'} {m} with {r.rating} (was {rating})")
if not dry_run:
trakt.rate(m, r.rating)

0 comments on commit ab4bb16

Please sign in to comment.