Skip to content

Commit ee1577b

Browse files
committed
IDEV-2020: Override self.data() method to isolate handling of FeedsResults data.
1 parent 6a8a76a commit ee1577b

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

domaintools/results.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -151,13 +151,20 @@ class FeedsResults(Results):
151151
def response(self) -> Generator:
152152
status_code = None
153153
while status_code != 200:
154-
resp_data = super().response()
154+
resp_data = self.data()
155155
status_code = self.status
156156
yield resp_data
157157

158-
self._data = None
159-
self._response = None
158+
self._data = None # clear the data here
160159
if not self.kwargs.get("sessionID"):
161160
# we'll only do iterative request for queries that has sessionID.
162161
# Otherwise, we will have an infinite request if sessionID was not provided but the required data asked is more than the maximum (1 hour of data)
163162
break
163+
164+
def data(self):
165+
results = self._get_results()
166+
self.setStatus(results.status_code, results)
167+
self._data = results.text
168+
self.check_limit_exceeded()
169+
170+
return self._data

0 commit comments

Comments
 (0)