Skip to content

Commit 1775a3e

Browse files
committed
Add option disabling results limit to API methods
1 parent b6f844c commit 1775a3e

File tree

1 file changed

+17
-6
lines changed

1 file changed

+17
-6
lines changed

easy_entrez/api.py

Lines changed: 17 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -142,13 +142,17 @@ def _request(self, query: EntrezQuery, custom_payload=None) -> EntrezResponse:
142142
@uses_query(SearchQuery)
143143
def search(
144144
self, term: Union[str, dict], max_results: int,
145-
database: EntrezDatabase = 'pubmed', min_date=None, max_date=None
145+
database: EntrezDatabase = 'pubmed', min_date=None, max_date=None,
146+
ignore_max_results_limit: bool = False
146147
):
147148
if isinstance(term, dict):
148149
term = _match_all(**term)
149150

150151
assert not min_date and not max_date # TODO
151-
query = SearchQuery(term=term, max_results=max_results, database=database)
152+
query = SearchQuery(
153+
term=term, max_results=max_results, database=database,
154+
ignore_max_results_limit=ignore_max_results_limit
155+
)
152156
return self._request(query=query)
153157

154158
def in_batches_of(self, size: int = 100, sleep_interval: int = 3):
@@ -161,20 +165,27 @@ def in_batches_of(self, size: int = 100, sleep_interval: int = 3):
161165
@uses_query(SummaryQuery)
162166
def summarize(
163167
self, ids: List[str], max_results: int,
164-
database: EntrezDatabase = 'pubmed'
168+
database: EntrezDatabase = 'pubmed', ignore_max_results_limit: bool = False
165169
):
166170
self._ensure_list_like(ids)
167-
query = SummaryQuery(ids=ids, max_results=max_results, database=database)
171+
query = SummaryQuery(
172+
ids=ids, max_results=max_results, database=database,
173+
ignore_max_results_limit=ignore_max_results_limit
174+
)
168175
return self._request(query=query)
169176

170177
@supports_batches
171178
@uses_query(FetchQuery)
172179
def fetch(
173180
self, ids: List[str], max_results: int,
174-
database: EntrezDatabase = 'pubmed', return_type: ReturnType = 'xml'
181+
database: EntrezDatabase = 'pubmed', return_type: ReturnType = 'xml',
182+
ignore_max_results_limit: bool = False
175183
):
176184
self._ensure_list_like(ids)
177-
query = FetchQuery(ids=ids, max_results=max_results, database=database, return_type=return_type)
185+
query = FetchQuery(
186+
ids=ids, max_results=max_results, database=database,
187+
return_type=return_type, ignore_max_results_limit=ignore_max_results_limit
188+
)
178189
return self._request(query=query)
179190

180191
@supports_batches

0 commit comments

Comments
 (0)