?HAMA

Edit File Size: 2.14 KB

/opt/cloudlinux/venv/lib/python3.11/site-packages/aiohttp_session/cookie_storage.py

Text Base64

import json
import base64

from cryptography import fernet
from cryptography.fernet import InvalidToken

from . import AbstractStorage, Session
from .log import log

class EncryptedCookieStorage(AbstractStorage):
    """Encrypted JSON storage.
    """

def __init__(self, secret_key, *, cookie_name="AIOHTTP_SESSION",
                 domain=None, max_age=None, path='/',
                 secure=None, httponly=True,
                 encoder=json.dumps, decoder=json.loads):
        super().__init__(cookie_name=cookie_name, domain=domain,
                         max_age=max_age, path=path, secure=secure,
                         httponly=httponly,
                         encoder=encoder, decoder=decoder)

if isinstance(secret_key, str):
            pass
        elif isinstance(secret_key, (bytes, bytearray)):
            secret_key = base64.urlsafe_b64encode(secret_key)
        self._fernet = fernet.Fernet(secret_key)

async def load_session(self, request):
        cookie = self.load_cookie(request)
        if cookie is None:
            return Session(None, data=None, new=True, max_age=self.max_age)
        else:
            try:
                data = self._decoder(
                    self._fernet.decrypt(
                        cookie.encode('utf-8'),
                        ttl=self.max_age
                    ).decode('utf-8')
                )
                return Session(None, data=data,
                               new=False, max_age=self.max_age)
            except InvalidToken:
                log.warning("Cannot decrypt cookie value, "
                            "create a new fresh session")
                return Session(None, data=None, new=True, max_age=self.max_age)

async def save_session(self, request, response, session):
        if session.empty:
            return self.save_cookie(response, '',
                                    max_age=session.max_age)

cookie_data = self._encoder(
            self._get_session_data(session)
        ).encode('utf-8')
        self.save_cookie(
            response,
            self._fernet.encrypt(cookie_data).decode('utf-8'),
            max_age=session.max_age
        )

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2026-02-06 08:01:08	Edit Download recursive recursive
cookie_storage.py	2.14 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
log.py	54 B	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
memcached_storage.py	2.80 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
nacl_storage.py	2.20 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
redis_storage.py	3.14 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
__init__.py	8.14 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive

Directory Contents