From 2fdc57acb055d8985dd4d8be3d85702e570457f8 Mon Sep 17 00:00:00 2001
From: yasir <yasiramunandar@gmail.com>
Date: Tue, 14 Feb 2023 11:01:24 +0700
Subject: [PATCH] Fix

---
 misskaty/plugins/imdb_search.py | 28 ++++++++++++++++++++++++----
 1 file changed, 24 insertions(+), 4 deletions(-)

diff --git a/misskaty/plugins/imdb_search.py b/misskaty/plugins/imdb_search.py
index 247575d3..dc8ad1c0 100644
--- a/misskaty/plugins/imdb_search.py
+++ b/misskaty/plugins/imdb_search.py
@@ -111,7 +111,12 @@ async def imdb_search_id(kueri, message):
         msg += f"🎬 Ditemukan ({len(res)}) hasil untuk kueri: <code>{kueri}</code>\n\n"
         for num, movie in enumerate(res, start=1):
             title = movie.get("l")
-            year = f"({movie.get('y', 'N/A')})"
+            if year := movie.get("yr"):
+                year = f"({year})"
+            elif year := movie.get("y"):
+                year = f"({year})"
+            else:
+                year = "N/A"
             typee = movie.get("q", "N/A").replace("feature", "movie").title()
             movieID = re.findall(r"tt(\d+)", movie.get("id"))[0]
             msg += f"{num}. {title} {year} - {typee}\n"
@@ -156,7 +161,12 @@ async def imdb_search_en(kueri, message):
         msg += f"🎬 Found ({len(res)}) result for keywords: <code>{kueri}</code>\n\n"
         for num, movie in enumerate(res, start=1):
             title = movie.get("l")
-            year = f"({movie.get('y', 'N/A')})"
+            if year := movie.get("yr"):
+                year = f"({year})"
+            elif year := movie.get("y"):
+                year = f"({year})"
+            else:
+                year = "N/A"
             typee = movie.get("q", "N/A").replace("feature", "movie").title()
             movieID = re.findall(r"tt(\d+)", movie.get("id"))[0]
             msg += f"{num}. {title} {year} - {typee}\n"
@@ -207,7 +217,12 @@ async def imdbcari(client, query):
             msg += f"🎬 Ditemukan ({len(res)}) hasil dari: <code>{kueri}</code> ~ {query.from_user.mention}\n\n"
             for num, movie in enumerate(res, start=1):
                 title = movie.get("l")
-                year = f"({movie.get('y', 'N/A')})"
+                if year := movie.get("yr"):
+                    year = f"({year})"
+                elif year := movie.get("y"):
+                    year = f"({year})"
+                else:
+                    year = "N/A"
                 typee = movie.get("q", "N/A").replace("feature", "movie").title()
                 movieID = re.findall(r"tt(\d+)", movie.get("id"))[0]
                 msg += f"{num}. {title} {year} - {typee}\n"
@@ -241,7 +256,12 @@ async def imdbcari(client, query):
             msg += f"🎬 Found ({len(res)}) result for keywords: <code>{kueri}</code> ~ {query.from_user.mention}\n\n"
             for num, movie in enumerate(res, start=1):
                 title = movie.get("l")
-                year = f"({movie.get('y', 'N/A')})"
+                if year := movie.get("yr"):
+                    year = f"({year})"
+                elif year := movie.get("y"):
+                    year = f"({year})"
+                else:
+                    year = "N/A"
                 typee = movie.get("q", "N/A").replace("feature", "movie").title()
                 movieID = re.findall(r"tt(\d+)", movie.get("id"))[0]
                 msg += f"{num}. {title} {year} - {typee}\n"