Merge pull request #1060 from planetlabs/1059-get-results-csv-not-throttled

subscriptions get_results_csv uses session.stream

Author: ischneider

planet/clients/subscriptions.py

@@ -395,7 +395,6 @@ async def get_results_csv(
         # during streaming. We may want to consider a retry decorator
         # for this entire method a la stamina:
         # https://github.com/hynek/stamina.
-        async with self._session._client.stream('GET', url,
-                                                params=params) as response:
+        async with self._session.stream('GET', url, params=params) as response:
             async for line in response.aiter_lines():
                 yield line

planet/http.py

@@ -396,8 +396,11 @@ async def _send(self, request, stream=False) -> httpx.Response:
 
     @asynccontextmanager
     async def stream(
-            self, method: str,
-            url: str) -> AsyncGenerator[models.StreamingResponse, None]:
+        self,
+        method: str,
+        url: str,
+        params: Optional[dict] = None
+    ) -> AsyncGenerator[models.StreamingResponse, None]:
         """Submit a request and get the response as a stream context manager.
 
         Parameters:
@@ -407,7 +410,9 @@ async def stream(
         Returns:
             Context manager providing the streaming response.
         """
-        request = self._client.build_request(method=method, url=url)
+        request = self._client.build_request(method=method,
+                                             url=url,
+                                             params=params)
         http_response = await self._retry(self._send, request, stream=True)
         response = models.StreamingResponse(http_response)
         try:

planet/models.py

@@ -72,6 +72,9 @@ async def aiter_bytes(self):
         async for c in self._http_response.aiter_bytes():
             yield c
 
+    def aiter_lines(self):
+        return self._http_response.aiter_lines()
+
     async def aclose(self):
         await self._http_response.aclose()