Skip to content

Commit

Permalink
fix: disable aiohttp client cookies
Browse files Browse the repository at this point in the history
  • Loading branch information
trim21 committed Aug 18, 2024
1 parent 97b4da2 commit 4db7642
Showing 1 changed file with 27 additions and 1 deletion.
28 changes: 27 additions & 1 deletion server/base.py
Original file line number Diff line number Diff line change
@@ -1,18 +1,44 @@
from http.cookies import BaseCookie
from typing import Any

import aiohttp
import asyncpg
import litestar
from aiohttp.abc import AbstractCookieJar, ClearCookiePredicate
from aiohttp.typedefs import LooseCookies
from litestar.exceptions import ClientException
from litestar.status_codes import HTTP_400_BAD_REQUEST
from loguru import logger
from typing_extensions import Never
from yarl import URL

from config import PG_DSN
from server.model import User


http_client = aiohttp.ClientSession()
class DisableCookiesJar(AbstractCookieJar):
"""disable cookies on aiohttp client"""

def clear(self, predicate: ClearCookiePredicate | None = None) -> None:
return

def clear_domain(self, domain: str) -> None:
return

def update_cookies(self, cookies: LooseCookies, response_url: URL = None) -> None:

Check failure on line 28 in server/base.py

View workflow job for this annotation

GitHub Actions / mypy

Incompatible default for argument "response_url" (default has type "None", argument has type "URL")
return

def filter_cookies(self, request_url: URL) -> BaseCookie[str]:
return BaseCookie()

def __len__(self):
return 0

def __iter__(self):
yield from ()


http_client = aiohttp.ClientSession(cookie_jar=DisableCookiesJar())
pg = asyncpg.create_pool(dsn=PG_DSN)


Expand Down

0 comments on commit 4db7642

Please sign in to comment.