diff options
| -rw-r--r-- | fatcat_scholar/schema.py | 4 | ||||
| -rw-r--r-- | fatcat_scholar/search.py | 9 | ||||
| -rw-r--r-- | fatcat_scholar/web.py | 4 | ||||
| -rw-r--r-- | pytype.cfg | 4 | ||||
| -rw-r--r-- | tests/test_scrub.py | 2 |
5 files changed, 16 insertions, 7 deletions
diff --git a/fatcat_scholar/schema.py b/fatcat_scholar/schema.py index 616857e..187fe58 100644 --- a/fatcat_scholar/schema.py +++ b/fatcat_scholar/schema.py @@ -10,8 +10,10 @@ from enum import Enum from typing import Optional, List, Any import ftfy -from pydantic import BaseModel from bs4 import BeautifulSoup +# pytype: disable=import-error +from pydantic import BaseModel +# pytype: enable=import-error from fatcat_openapi_client import ReleaseEntity, ReleaseContrib from fatcat_scholar.api_entities import entity_to_dict diff --git a/fatcat_scholar/search.py b/fatcat_scholar/search.py index 6402e2b..a529f6a 100644 --- a/fatcat_scholar/search.py +++ b/fatcat_scholar/search.py @@ -3,13 +3,16 @@ Helpers to make elasticsearch queries. """ import sys -from gettext import gettext import datetime +from gettext import gettext +from typing import List, Optional, Any + import elasticsearch -from pydantic import BaseModel from dynaconf import settings from elasticsearch_dsl import Search, Q -from typing import List, Optional, Any +# pytype: disable=import-error +from pydantic import BaseModel +# pytype: enable=import-error # i18n note: the use of gettext below doesn't actually do the translation here, # it just ensures that the strings are caught by babel for translation later diff --git a/fatcat_scholar/web.py b/fatcat_scholar/web.py index 4e912d4..8b55f75 100644 --- a/fatcat_scholar/web.py +++ b/fatcat_scholar/web.py @@ -70,14 +70,14 @@ async def search(query: FulltextQuery = Depends(FulltextQuery)) -> Any: web = APIRouter() -def locale_gettext(translations: List[Any]) -> Any: +def locale_gettext(translations: Any) -> Any: def gt(s): return translations.ugettext(s) return gt -def locale_ngettext(translations: List[Any]) -> Any: +def locale_ngettext(translations: Any) -> Any: def ngt(s, n): return translations.ungettext(s) diff --git a/pytype.cfg b/pytype.cfg new file mode 100644 index 0000000..d4916dc --- /dev/null +++ b/pytype.cfg @@ -0,0 +1,4 @@ +[pytype] +inputs = fatcat_scholar +pythonpath = . + diff --git a/tests/test_scrub.py b/tests/test_scrub.py index 2cd68cc..fc6ef22 100644 --- a/tests/test_scrub.py +++ b/tests/test_scrub.py @@ -1,4 +1,4 @@ -from fatcat_scholar.schema import * +from fatcat_scholar.schema import scrub_text def test_scrub() -> None: |