add new field complete_title in fetching drama

Difference from `title`: Example: `Goblin (2016)` Title = `Goblin` Complete Title = `Goblin (2016)`
tbdsux · Apr 1, 2023 · 657d240 · 657d240 · vercel · Apr 1, 2023
1 parent e8fdb1c
commit 657d240
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/api/fetch.py b/api/fetch.py
@@ -20,7 +20,12 @@ def _get_main_container(self) -> None:
         # these are the most important drama infos / details
 
         # TITLE
-        self.info["title"] = container.find("h1", class_="film-title").find("a").text
+        # Example: Goblin (2016)
+        # Title = Goblin
+        # Complete Title = Goblin (2016)
+        film_title = container.find("h1", class_="film-title")
+        self.info["title"] = film_title.find("a").get_text().strip()
+        self.info["complete_title"] = film_title.get_text().strip()
 
         # RATING (could be either N/A or with number)
         self.info["rating"] = self._handle_rating(

diff --git a/tests/test_main.py b/tests/test_main.py
@@ -14,10 +14,13 @@ def test_index() -> None:
 def test_sample_drama() -> None:
     dramas = [
         {
-            "id": "65743-alien-girl-chai-xiao-qi-2",
-            "title": "My Girlfriend Is an Alien 2",
+            "id": "18452-goblin",
+            "expected": {"title": "Goblin", "complete_title": "Goblin (2016)"},
+        },
+        {
+            "id": "58953-mouse",
+            "expected": {"title": "Mouse", "complete_title": "Mouse (2021)"},
         },
-        {"id": "58953-mouse", "title": "Mouse"},
     ]
 
     for i in dramas:
@@ -27,7 +30,8 @@ def test_sample_drama() -> None:
 
         print(r.json()["data"])
 
-        assert r.json()["data"]["title"] == i["title"]
+        assert r.json()["data"]["title"] == i["expected"]["title"]
+        assert r.json()["data"]["complete_title"] == i["expected"]["complete_title"]
 
 
 def test_unknown_drama() -> None: