Add type hints (#6191)

This commit is contained in:
Chan Sau Yee 2023-12-29 20:33:37 +09:00 committed by GitHub
parent 1902284942
commit 40e623b276
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 3 deletions

View File

@ -340,7 +340,9 @@ class FilesPipeline(MediaPipeline):
DEFAULT_FILES_URLS_FIELD = "file_urls"
DEFAULT_FILES_RESULT_FIELD = "files"
def __init__(self, store_uri, download_func=None, settings=None):
def __init__(
self, store_uri: Union[str, PathLike], download_func=None, settings=None
):
store_uri = _to_string(store_uri)
if not store_uri:
raise NotConfigured

View File

@ -8,7 +8,8 @@ import hashlib
import warnings
from contextlib import suppress
from io import BytesIO
from typing import Dict, Tuple
from os import PathLike
from typing import Dict, Tuple, Union
from itemadapter import ItemAdapter
@ -53,7 +54,9 @@ class ImagesPipeline(FilesPipeline):
DEFAULT_IMAGES_URLS_FIELD = "image_urls"
DEFAULT_IMAGES_RESULT_FIELD = "images"
def __init__(self, store_uri, download_func=None, settings=None):
def __init__(
self, store_uri: Union[str, PathLike], download_func=None, settings=None
):
try:
from PIL import Image