Merge pull request #18 from ivansaul/refactor

fix: add handling for quiz URLs
ivansaul · Nov 18, 2024 · 186fe6b · 186fe6b
2 parents aa3a53b + 6347ebb
commit 186fe6b
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/platzi/async_api.py b/src/platzi/async_api.py
@@ -180,6 +180,12 @@ async def download(self, url: str, **kwargs):
                     Logger.print(f"[{dst.name}]", "[DOWNLOADING]")
                     await self.save_page(unit.url, path=dst)
 
+                # download quiz
+                if unit.type == TypeUnit.QUIZ:
+                    dst = CHAP_DIR / f"{file_name}.mhtml"
+                    Logger.print(f"[{dst.name}]", "[DOWNLOADING]")
+                    await self.save_page(unit.url, path=dst)
+
             print("=" * 100)
 
     @try_except_request

diff --git a/src/platzi/collectors.py b/src/platzi/collectors.py
@@ -73,6 +73,14 @@ async def get_unit(context: BrowserContext, url: str) -> Unit:
     TITLE_SELECTOR = ".MaterialDesktopHeading_MaterialDesktopHeading-info__title__DaYr2"
     EXCEPTION = Exception("Could not collect unit data")
 
+    if "/quiz/" in url:
+        return Unit(
+            url=url,
+            title="Quiz",
+            type=TypeUnit.QUIZ,
+            slug=slugify("Quiz"),
+        )
+
     try:
         page = await context.new_page()
         await page.goto(url)