mypy: Reduce use of Any in zerver/lib/url_preview/ return types.

This commit is contained in:
neiljp (Neil Pilgrim) 2017-08-07 12:19:19 -07:00
parent b972b083d9
commit be856bad46
3 changed files with 9 additions and 9 deletions

View File

@ -1,9 +1,9 @@
from typing import Optional, Any, Text from typing import Optional, Text, Dict
from pyoembed import oEmbed, PyOembedException from pyoembed import oEmbed, PyOembedException
def get_oembed_data(url, maxwidth=640, maxheight=480): def get_oembed_data(url, maxwidth=640, maxheight=480):
# type: (Text, Optional[int], Optional[int]) -> Any # type: (Text, Optional[int], Optional[int]) -> Optional[Dict]
try: try:
data = oEmbed(url, maxwidth=maxwidth, maxheight=maxheight) data = oEmbed(url, maxwidth=maxwidth, maxheight=maxheight)
except PyOembedException: except PyOembedException:

View File

@ -1,17 +1,17 @@
from typing import Any, Dict from typing import Dict, Optional, Text
from zerver.lib.url_preview.parsers.base import BaseParser from zerver.lib.url_preview.parsers.base import BaseParser
class GenericParser(BaseParser): class GenericParser(BaseParser):
def extract_data(self): def extract_data(self):
# type: () -> Dict # type: () -> Dict[str, Optional[Text]]
return { return {
'title': self._get_title(), 'title': self._get_title(),
'description': self._get_description(), 'description': self._get_description(),
'image': self._get_image()} 'image': self._get_image()}
def _get_title(self): def _get_title(self):
# type: () -> Any # type: () -> Optional[Text]
soup = self._soup soup = self._soup
if (soup.title and soup.title.text != ''): if (soup.title and soup.title.text != ''):
return soup.title.text return soup.title.text
@ -20,7 +20,7 @@ class GenericParser(BaseParser):
return None return None
def _get_description(self): def _get_description(self):
# type: () -> Any # type: () -> Optional[Text]
soup = self._soup soup = self._soup
meta_description = soup.find('meta', attrs={'name': 'description'}) meta_description = soup.find('meta', attrs={'name': 'description'})
if (meta_description and meta_description['content'] != ''): if (meta_description and meta_description['content'] != ''):
@ -36,7 +36,7 @@ class GenericParser(BaseParser):
return None return None
def _get_image(self): def _get_image(self):
# type: () -> Any # type: () -> Optional[Text]
""" """
Finding a first image after the h1 header. Finding a first image after the h1 header.
Presumably it will be the main image. Presumably it will be the main image.

View File

@ -1,7 +1,7 @@
import re import re
import logging import logging
import traceback import traceback
from typing import Any, Optional, Text from typing import Any, Optional, Text, Dict
from typing.re import Match from typing.re import Match
import requests import requests
from zerver.lib.cache import cache_with_key, get_cache_with_key from zerver.lib.cache import cache_with_key, get_cache_with_key
@ -32,7 +32,7 @@ def cache_key_func(url):
@cache_with_key(cache_key_func, cache_name=CACHE_NAME, with_statsd_key="urlpreview_data") @cache_with_key(cache_key_func, cache_name=CACHE_NAME, with_statsd_key="urlpreview_data")
def get_link_embed_data(url, maxwidth=640, maxheight=480): def get_link_embed_data(url, maxwidth=640, maxheight=480):
# type: (Text, Optional[int], Optional[int]) -> Any # type: (Text, Optional[int], Optional[int]) -> Optional[Dict]
if not is_link(url): if not is_link(url):
return None return None
# Fetch information from URL. # Fetch information from URL.