v 3.0.3

pysteamauth/base/__init__.py

@@ -0,0 +1,8 @@
+from .request import BaseRequestStrategy
+from .storage import BaseCookieStorage
+
+
+__all__ = [
+    'BaseRequestStrategy',
+    'BaseCookieStorage',
+]

pysteamauth/base/request.py

@@ -0,0 +1,65 @@
+from typing import (
+    Any,
+    Mapping,
+    Optional,
+)
+
+import aiohttp
+from aiohttp import (
+    ClientResponse,
+    ClientSession,
+    ClientTimeout,
+)
+
+from pysteamauth.abstract import RequestStrategyAbstract
+from pysteamauth.errors import check_steam_error
+
+DEFAULT_REQUEST_TIMEOUT = ClientTimeout(total=30)
+
+
+class BaseRequestStrategy(RequestStrategyAbstract):
+
+    def __init__(self):
+        self._session: Optional[ClientSession] = None
+
+    def __del__(self):
+        if self._session:
+            self._session.connector.close()
+
+    def _create_session(self) -> ClientSession:
+        """
+        Create aiohttp session.
+        Aiohttp session saves and stores cookies.
+        It writes cookies from responses after each request that specified
+        in Set-Cookie header.
+
+        :return: aiohttp.ClientSession object.
+        """
+        return ClientSession(
+            connector=aiohttp.TCPConnector(ssl=False),
+            timeout=DEFAULT_REQUEST_TIMEOUT,
+        )
+
+    async def request(self, url: str, method: str, **kwargs: Any) -> ClientResponse:
+        if self._session is None:
+            self._session = self._create_session()
+        response = await self._session.request(method, url, **kwargs)
+        error = response.headers.get('X-eresult')
+        if error:
+            check_steam_error(int(error))
+        return response
+
+    def cookies(self, domain: str = 'steamcommunity.com') -> Mapping[str, str]:
+        if self._session is None:
+            raise RuntimeError('Session is not initialized')
+        cookies = {}
+        for cookie in self._session.cookie_jar:
+            if cookie['domain'] == domain:
+                cookies[cookie.key] = cookie.value
+        return cookies
+
+    async def text(self, url: str, method: str, **kwargs: Any) -> str:
+        return await (await self.request(url, method, **kwargs)).text()
+
+    async def bytes(self, url: str, method: str, **kwargs: Any) -> bytes:
+        return await (await self.request(url, method, **kwargs)).read()

pysteamauth/base/storage.py

@@ -0,0 +1,21 @@
+from typing import (
+    Dict,
+    Mapping,
+)
+
+from pysteamauth.abstract import CookieStorageAbstract
+
+
+class BaseCookieStorage(CookieStorageAbstract):
+
+    def __init__(self):
+        self.cookies: Dict[str, Mapping[str, Mapping[str, str]]] = {}
+
+    async def set(self, login: str, cookies: Mapping[str, Mapping[str, str]]) -> None:
+        self.cookies[login] = cookies
+
+    async def get(self, login: str, domain: str) -> Mapping[str, str]:
+        cookies = self.cookies.get(login)
+        if not cookies:
+            return {}
+        return cookies.get(domain, {})