diff --git a/misskaty/plugins/imdb_search.py b/misskaty/plugins/imdb_search.py index 408e0d09..a6f7af70 100644 --- a/misskaty/plugins/imdb_search.py +++ b/misskaty/plugins/imdb_search.py @@ -291,12 +291,8 @@ async def imdb_id_callback(bot, query): if duration := sop.select('li[data-testid="title-techspec_runtime"]'): durasi = duration[0].find(class_="ipc-metadata-list-item__content-container").text res_str += f"Durasi: {GoogleTranslator('auto', 'id').translate(durasi)}\n" - if cat := sop.select('ul[data-testid="hero-title-block__metadata"]'): - try: - cat = cat[0].find("span", class_="sc-8c396aa2-2 jwaBvf").text - except: - cat = "N/A" - res_str += f"Kategori: {cat} \n" + if r_json.get("contentRating"): + res_str += f"Kategori: {r_json['contentRating']} \n" if r_json.get("aggregateRating"): res_str += f"Peringkat: {r_json['aggregateRating']['ratingValue']}⭐️ dari {r_json['aggregateRating']['ratingCount']} pengguna \n" if release := sop.select('li[data-testid="title-details-releasedate"]'): @@ -420,12 +416,8 @@ async def imdb_en_callback(bot, query): if durasi := sop.select('li[data-testid="title-techspec_runtime"]'): durasi = durasi[0].find(class_="ipc-metadata-list-item__content-container").text res_str += f"Duration: {durasi}\n" - if cat := sop.select('ul[data-testid="hero-title-block__metadata"]'): - try: - cat = cat[0].find("span", class_="sc-8c396aa2-2 jwaBvf").text - except: - cat = "N/A" - res_str += f"Category: {cat} \n" + if r_json.get("contentRating"): + res_str += f"Category: {r_json['contentRating']} \n" if r_json.get("aggregateRating"): res_str += f"Rating: {r_json['aggregateRating']['ratingValue']}⭐️ from {r_json['aggregateRating']['ratingCount']} user \n" if release := sop.select('li[data-testid="title-details-releasedate"]'): diff --git a/misskaty/plugins/inline_search.py b/misskaty/plugins/inline_search.py index 6460847b..6dabd5b2 100644 --- a/misskaty/plugins/inline_search.py +++ b/misskaty/plugins/inline_search.py @@ -452,12 +452,8 @@ async def imdb_inl(_, query): if durasi := sop.select('li[data-testid="title-techspec_runtime"]'): durasi = durasi[0].find(class_="ipc-metadata-list-item__content-container").text res_str += f"Durasi: {GoogleTranslator('auto', 'id').translate(durasi)}\n" - if cat := sop.select('ul[data-testid="hero-title-block__metadata"]'): - try: - cat = cat[0].find("span", class_="sc-8c396aa2-2 jwaBvf").text - except: - cat = "N/A" - res_str += f"Kategori: {cat} \n" + if r_json.get("contentRating"): + res_str += f"Kategori: {r_json['contentRating']} \n" if r_json.get("aggregateRating"): res_str += f"Peringkat: {r_json['aggregateRating']['ratingValue']}⭐️ dari {r_json['aggregateRating']['ratingCount']} pengguna \n" if release := sop.select('li[data-testid="title-details-releasedate"]'):