Remove stamina (#16)

Author: LsHanaha

pyproject.toml

@@ -19,7 +19,6 @@ dependencies = [
     "httpx",
     "aioboto3",
     "types-aioboto3[s3]",
-    "stamina",
     "pydantic",
     "pyvips",
     "pyvips-binary",

safe_s3_storage/kaspersky_scan_engine.py

@@ -1,20 +1,14 @@
 import base64
 import dataclasses
 import enum
-import logging
 import typing
 
 import httpx
 import pydantic
-import stamina
 
 from safe_s3_storage.exceptions import KasperskyScanEngineConnectionStatusError, KasperskyScanEngineThreatDetectedError
 
 
-kaspersky_logger: typing.Final = logging.getLogger(__name__)
-kaspersky_logger.setLevel(logging.ERROR)
-
-
 class KasperskyScanEngineRequest(pydantic.BaseModel):
     timeout: str
     object: str
@@ -53,9 +47,7 @@ async def scan_memory(self, *, file_name: str, file_content: bytes) -> None:
             timeout=str(self.timeout_ms), object=base64.b64encode(file_content).decode(), name=self.client_name
         ).model_dump(mode="json")
         try:
-            response: typing.Final = await stamina.retry(on=httpx.HTTPError, attempts=self.max_retries)(
-                self._send_scan_memory_request
-            )(payload)
+            response: typing.Final = await self._send_scan_memory_request(payload)
         except httpx.HTTPStatusError as exc:
             raise KasperskyScanEngineConnectionStatusError from exc
         validated_response: typing.Final = KasperskyScanEngineResponse.model_validate_json(response)

tests/conftest.py

@@ -2,19 +2,13 @@
 
 import faker
 import pytest
-import stamina
 
 
 @pytest.fixture(scope="session", autouse=True)
 def anyio_backend() -> str:
     return "asyncio"
 
 
-@pytest.fixture(scope="session", autouse=True)
-def deactivate_retries() -> None:
-    stamina.set_active(False)
-
-
 MIME_OCTET_STREAM: typing.Final = "application/octet-stream"