Jelajahi Sumber

fix: Replace `any` typing with Union[Iterable[Link], QuerySet] in `archive_links`

Cristian 5 tahun lalu
induk
melakukan
0a83392cbf
1 mengubah file dengan 2 tambahan dan 2 penghapusan
  1. 2 2
      archivebox/extractors/__init__.py

+ 2 - 2
archivebox/extractors/__init__.py

@@ -2,7 +2,7 @@ __package__ = 'archivebox.extractors'
 
 
 import os
 import os
 
 
-from typing import Optional, List, Iterable
+from typing import Optional, List, Iterable, Union, TYPE_CHECKING
 from datetime import datetime
 from datetime import datetime
 from django.db.models import QuerySet
 from django.db.models import QuerySet
 
 
@@ -130,7 +130,7 @@ def archive_link(link: Link, overwrite: bool=False, methods: Optional[Iterable[s
     return link
     return link
 
 
 @enforce_types
 @enforce_types
-def archive_links(all_links: any, overwrite: bool=False, methods: Optional[Iterable[str]]=None, out_dir: Optional[str]=None) -> List[Link]:
+def archive_links(all_links: Union[Iterable[Link], QuerySet], overwrite: bool=False, methods: Optional[Iterable[str]]=None, out_dir: Optional[str]=None) -> List[Link]:
 
 
     if type(all_links) is QuerySet:
     if type(all_links) is QuerySet:
         num_links: int = all_links.count()
         num_links: int = all_links.count()