Retry crawl_url and work around Python's closure handling

Closures capture the scope and not the variables. Therefore, any
type-narrowing performed by mypy on captured variables is lost inside
the closure.
This commit is contained in:
I-Al-Istannen 2021-05-23 12:24:10 +02:00
parent e81005ae4b
commit 3d4b997d4a

View File

@ -109,6 +109,7 @@ def _wrap_io_in_warning(name: str) -> Callable[[AWrapped], AWrapped]:
# | # |
# crawl_course --+ # crawl_course --+
# | # |
# @_io_repeat | # retries internally (before the bar)
# +- crawl_url <-+ # +- crawl_url <-+
# | # |
# | # |
@ -176,13 +177,15 @@ class KitIliasWebCrawler(HttpCrawler):
await self._crawl_url(self._base_url) await self._crawl_url(self._base_url)
async def _crawl_url(self, url: str, expected_id: Optional[int] = None) -> None: async def _crawl_url(self, url: str, expected_id: Optional[int] = None) -> None:
cl = await self.crawl(PurePath(".")) maybe_cl = await self.crawl(PurePath("."))
if not cl: if not maybe_cl:
return return
cl = maybe_cl # Not mypy's fault, but explained here: https://github.com/python/mypy/issues/2608
@_iorepeat(3, "crawling url")
async def impl() -> None:
tasks = [] tasks = []
# TODO: Retry this when the crawl and download bar are reworked
async with cl: async with cl:
soup = await self._get_page(url) soup = await self._get_page(url)
@ -196,16 +199,23 @@ class KitIliasWebCrawler(HttpCrawler):
for child in page.get_child_elements(): for child in page.get_child_elements():
tasks.append(self._handle_ilias_element(PurePath("."), child)) tasks.append(self._handle_ilias_element(PurePath("."), child))
# The only point an I/O exception can be thrown is in `get_page`.
# If that happens, no task was spawned yet. Therefore, we only retry
# this method without having spawned a single task. Due to this we do
# not need to cancel anything or worry about this gather call or the forks
# further up.
await asyncio.gather(*tasks) await asyncio.gather(*tasks)
await impl()
async def _handle_ilias_page(self, url: str, parent: IliasPageElement, path: PurePath) -> None: async def _handle_ilias_page(self, url: str, parent: IliasPageElement, path: PurePath) -> None:
cl = await self.crawl(path) maybe_cl = await self.crawl(path)
if not cl: if not maybe_cl:
return return
cl = maybe_cl # Not mypy's fault, but explained here: https://github.com/python/mypy/issues/2608
@_iorepeat(3, "crawling folder") @_iorepeat(3, "crawling folder")
async def impl() -> None: async def impl() -> None:
assert cl # The function is only reached when cl is not None
tasks = [] tasks = []
async with cl: async with cl:
soup = await self._get_page(url) soup = await self._get_page(url)
@ -225,8 +235,8 @@ class KitIliasWebCrawler(HttpCrawler):
@anoncritical @anoncritical
# Shouldn't happen but this method must never raise an I/O error as that might interfere with # Shouldn't happen but this method must never raise an I/O error as that might interfere with
# handle_ilias_page # handle_ilias_page or crawl_url
@_wrap_io_in_warning("ilias element handling") @_wrap_io_in_warning("handling ilias element")
async def _handle_ilias_element(self, parent_path: PurePath, element: IliasPageElement) -> None: async def _handle_ilias_element(self, parent_path: PurePath, element: IliasPageElement) -> None:
element_path = PurePath(parent_path, element.name) element_path = PurePath(parent_path, element.name)
@ -250,13 +260,13 @@ class KitIliasWebCrawler(HttpCrawler):
raise CrawlWarning(f"Unknown element type: {element.type!r}") raise CrawlWarning(f"Unknown element type: {element.type!r}")
async def _download_link(self, element: IliasPageElement, element_path: PurePath) -> None: async def _download_link(self, element: IliasPageElement, element_path: PurePath) -> None:
dl = await self.download(element_path, mtime=element.mtime) maybe_dl = await self.download(element_path, mtime=element.mtime)
if not dl: if not maybe_dl:
return return
dl = maybe_dl # Not mypy's fault, but explained here: https://github.com/python/mypy/issues/2608
@_iorepeat(3, "link resolving") @_iorepeat(3, "resolving link")
async def impl() -> None: async def impl() -> None:
assert dl # This function is only reached when dl is not None
async with dl as (bar, sink): async with dl as (bar, sink):
export_url = element.url.replace("cmd=calldirectlink", "cmd=exportHTML") export_url = element.url.replace("cmd=calldirectlink", "cmd=exportHTML")
real_url = await self._resolve_link_target(export_url) real_url = await self._resolve_link_target(export_url)
@ -288,11 +298,12 @@ class KitIliasWebCrawler(HttpCrawler):
async def _download_video(self, element: IliasPageElement, element_path: PurePath) -> None: async def _download_video(self, element: IliasPageElement, element_path: PurePath) -> None:
# Videos will NOT be redownloaded - their content doesn't really change and they are chunky # Videos will NOT be redownloaded - their content doesn't really change and they are chunky
dl = await self.download(element_path, mtime=element.mtime, redownload=Redownload.NEVER) maybe_dl = await self.download(element_path, mtime=element.mtime, redownload=Redownload.NEVER)
if not dl: if not maybe_dl:
return return
dl = maybe_dl # Not mypy's fault, but explained here: https://github.com/python/mypy/issues/2608
@_iorepeat(3, "video download") @_iorepeat(3, "downloading video")
async def impl() -> None: async def impl() -> None:
assert dl # The function is only reached when dl is not None assert dl # The function is only reached when dl is not None
async with dl as (bar, sink): async with dl as (bar, sink):
@ -304,11 +315,12 @@ class KitIliasWebCrawler(HttpCrawler):
await impl() await impl()
async def _download_file(self, element: IliasPageElement, element_path: PurePath) -> None: async def _download_file(self, element: IliasPageElement, element_path: PurePath) -> None:
dl = await self.download(element_path, mtime=element.mtime) maybe_dl = await self.download(element_path, mtime=element.mtime)
if not dl: if not maybe_dl:
return return
dl = maybe_dl # Not mypy's fault, but explained here: https://github.com/python/mypy/issues/2608
@_iorepeat(3, "file download") @_iorepeat(3, "downloading file")
async def impl() -> None: async def impl() -> None:
assert dl # The function is only reached when dl is not None assert dl # The function is only reached when dl is not None
async with dl as (bar, sink): async with dl as (bar, sink):