summaryrefslogtreecommitdiffstats
path: root/fatcat_scholar/search.py
diff options
context:
space:
mode:
authorBryan Newbold <bnewbold@archive.org>2020-08-06 16:10:38 -0700
committerBryan Newbold <bnewbold@archive.org>2020-08-06 19:01:12 -0700
commit4a45f3aa1b5ed21dafec42ded23c300d51189e91 (patch)
treeba66592379cc2a9e50ba0b577a6daa35cf60bb83 /fatcat_scholar/search.py
parentdd96b7906e417dd74081e508aaabfebaeac5ece1 (diff)
downloadfatcat-scholar-4a45f3aa1b5ed21dafec42ded23c300d51189e91.tar.gz
fatcat-scholar-4a45f3aa1b5ed21dafec42ded23c300d51189e91.zip
make fmt
Diffstat (limited to 'fatcat_scholar/search.py')
-rw-r--r--fatcat_scholar/search.py28
1 files changed, 14 insertions, 14 deletions
diff --git a/fatcat_scholar/search.py b/fatcat_scholar/search.py
index 7fd26b3..6f11f30 100644
--- a/fatcat_scholar/search.py
+++ b/fatcat_scholar/search.py
@@ -89,7 +89,9 @@ def do_fulltext_search(
query: FulltextQuery, deep_page_limit: int = 2000
) -> FulltextHits:
- es_client = elasticsearch.Elasticsearch(settings.ELASTICSEARCH_BACKEND, timeout=25.0)
+ es_client = elasticsearch.Elasticsearch(
+ settings.ELASTICSEARCH_BACKEND, timeout=25.0
+ )
search = Search(using=es_client, index=settings.ELASTICSEARCH_FULLTEXT_INDEX)
# Convert raw DOIs to DOI queries
@@ -132,7 +134,8 @@ def do_fulltext_search(
tomorrow_date = str(date_today + datetime.timedelta(days=1))
year_ago_date = str(date_today - datetime.timedelta(days=365))
search = search.filter(
- Q("range", date=dict(gte=year_ago_date, lte=tomorrow_date)) | Q("term", year=this_year)
+ Q("range", date=dict(gte=year_ago_date, lte=tomorrow_date))
+ | Q("term", year=this_year)
)
elif query.filter_time == "since_2000":
this_year = datetime.date.today().year
@@ -152,7 +155,9 @@ def do_fulltext_search(
elif query.filter_availability == "everything":
pass
elif query.filter_availability == "fulltext" or query.filter_availability is None:
- search = search.filter("terms", **{"access.access_type": ["wayback", "ia_file", "ia_sim"]})
+ search = search.filter(
+ "terms", **{"access.access_type": ["wayback", "ia_file", "ia_sim"]}
+ )
elif query.filter_availability == "microfilm":
search = search.filter("term", **{"access.access_type": "ia_sim"})
else:
@@ -181,11 +186,7 @@ def do_fulltext_search(
allow_leading_wildcard=False,
lenient=True,
quote_field_suffix=".exact",
- fields=[
- "title^4",
- "biblio_all^3",
- "everything",
- ],
+ fields=["title^4", "biblio_all^3", "everything",],
)
has_fulltext = Q("terms", **{"access_type": ["ia_sim", "ia_file", "wayback"]})
poor_metadata = Q(
@@ -210,10 +211,7 @@ def do_fulltext_search(
search = search.sort("_doc")
else:
search = search.query(
- "boosting",
- positive=base_query,
- negative=poor_metadata,
- negative_boost=0.5,
+ "boosting", positive=base_query, negative=poor_metadata, negative_boost=0.5,
)
search = search.highlight(
@@ -279,7 +277,9 @@ def do_fulltext_search(
if isinstance(h.meta.inner_hits.more_pages.hits.total, int):
r["_collapsed_count"] = h.meta.inner_hits.more_pages.hits.total - 1
else:
- r["_collapsed_count"] = h.meta.inner_hits.more_pages.hits.total['value'] - 1
+ r["_collapsed_count"] = (
+ h.meta.inner_hits.more_pages.hits.total["value"] - 1
+ )
for k in h.meta.inner_hits.more_pages:
if k["key"] != r["key"]:
r["_collapsed"].append(k)
@@ -300,7 +300,7 @@ def do_fulltext_search(
if isinstance(resp.hits.total, int):
count_found = int(resp.hits.total)
else:
- count_found = int(resp.hits.total['value'])
+ count_found = int(resp.hits.total["value"])
count_returned = len(results)
# if we grouped to less than a page of hits, update returned count