Fixed cookies= httpx warning, refs #2307

This commit is contained in:
Simon Willison 2024-03-15 15:19:18 -07:00
commit 54f5604caf

View file

@ -1935,50 +1935,58 @@ class DatasetteClient:
async def get(self, path, **kwargs): async def get(self, path, **kwargs):
async with httpx.AsyncClient( async with httpx.AsyncClient(
transport=httpx.ASGITransport(app=self.app) transport=httpx.ASGITransport(app=self.app),
cookies=kwargs.pop("cookies", None),
) as client: ) as client:
return await client.get(self._fix(path), **kwargs) return await client.get(self._fix(path), **kwargs)
async def options(self, path, **kwargs): async def options(self, path, **kwargs):
async with httpx.AsyncClient( async with httpx.AsyncClient(
transport=httpx.ASGITransport(app=self.app) transport=httpx.ASGITransport(app=self.app),
cookies=kwargs.pop("cookies", None),
) as client: ) as client:
return await client.options(self._fix(path), **kwargs) return await client.options(self._fix(path), **kwargs)
async def head(self, path, **kwargs): async def head(self, path, **kwargs):
async with httpx.AsyncClient( async with httpx.AsyncClient(
transport=httpx.ASGITransport(app=self.app) transport=httpx.ASGITransport(app=self.app),
cookies=kwargs.pop("cookies", None),
) as client: ) as client:
return await client.head(self._fix(path), **kwargs) return await client.head(self._fix(path), **kwargs)
async def post(self, path, **kwargs): async def post(self, path, **kwargs):
async with httpx.AsyncClient( async with httpx.AsyncClient(
transport=httpx.ASGITransport(app=self.app) transport=httpx.ASGITransport(app=self.app),
cookies=kwargs.pop("cookies", None),
) as client: ) as client:
return await client.post(self._fix(path), **kwargs) return await client.post(self._fix(path), **kwargs)
async def put(self, path, **kwargs): async def put(self, path, **kwargs):
async with httpx.AsyncClient( async with httpx.AsyncClient(
transport=httpx.ASGITransport(app=self.app) transport=httpx.ASGITransport(app=self.app),
cookies=kwargs.pop("cookies", None),
) as client: ) as client:
return await client.put(self._fix(path), **kwargs) return await client.put(self._fix(path), **kwargs)
async def patch(self, path, **kwargs): async def patch(self, path, **kwargs):
async with httpx.AsyncClient( async with httpx.AsyncClient(
transport=httpx.ASGITransport(app=self.app) transport=httpx.ASGITransport(app=self.app),
cookies=kwargs.pop("cookies", None),
) as client: ) as client:
return await client.patch(self._fix(path), **kwargs) return await client.patch(self._fix(path), **kwargs)
async def delete(self, path, **kwargs): async def delete(self, path, **kwargs):
async with httpx.AsyncClient( async with httpx.AsyncClient(
transport=httpx.ASGITransport(app=self.app) transport=httpx.ASGITransport(app=self.app),
cookies=kwargs.pop("cookies", None),
) as client: ) as client:
return await client.delete(self._fix(path), **kwargs) return await client.delete(self._fix(path), **kwargs)
async def request(self, method, path, **kwargs): async def request(self, method, path, **kwargs):
avoid_path_rewrites = kwargs.pop("avoid_path_rewrites", None) avoid_path_rewrites = kwargs.pop("avoid_path_rewrites", None)
async with httpx.AsyncClient( async with httpx.AsyncClient(
transport=httpx.ASGITransport(app=self.app) transport=httpx.ASGITransport(app=self.app),
cookies=kwargs.pop("cookies", None),
) as client: ) as client:
return await client.request( return await client.request(
method, self._fix(path, avoid_path_rewrites), **kwargs method, self._fix(path, avoid_path_rewrites), **kwargs