diff --git a/unwind/imdb.py b/unwind/imdb.py index 0da9091..666f8ed 100644 --- a/unwind/imdb.py +++ b/unwind/imdb.py @@ -45,7 +45,7 @@ async def refresh_user_ratings_from_imdb(stop_on_dupe=True): log.info("⚡️ Loading data for %s ...", user.name) async for rating, is_updated in load_ratings(user.imdb_id): - assert rating.user == user + assert rating.user.id == user.id if stop_on_dupe and not is_updated: break @@ -156,7 +156,10 @@ async def parse_page(url) -> Tuple[list[Rating], Optional[str]]: meta = soup.find("meta", property="pageId") headline = soup.h1 assert meta is not None and headline is not None - user = User(imdb_id=meta["content"], name="") + imdb_id = meta["content"] + user = await db.get(User, imdb_id=imdb_id) or User( + imdb_id=imdb_id, name="", secret="" + ) if match := find_name(headline.string): user.name = match["name"]