Fix mypy

Ostorlab · Dec 5, 2024 · c7b5eb5 · c7b5eb5
1 parent f015803
commit c7b5eb5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/agent/exploits/cve_2024_51378.py b/agent/exploits/cve_2024_51378.py
@@ -54,8 +54,8 @@ def _get_csrf_token(self, base_url: str) -> str:
         try:
             resp = self.session.get(base_url, timeout=DEFAULT_TIMEOUT.seconds)
             resp.raise_for_status()
-            return resp.cookies.get("csrftoken", "")
-        except requests_exceptions.RequestException as e:
+            return str(resp.cookies.get("csrftoken", ""))
+        except requests_exceptions.RequestException:
             return ""
 
     def _attempt_rce(self, base_url: str, csrf_token: str, endpoint: str) -> bool:
@@ -78,5 +78,5 @@ def _attempt_rce(self, base_url: str, csrf_token: str, endpoint: str) -> bool:
                 response.json().get("requestStatus") is not None
                 and "uid=" in response.text
             )
-        except requests_exceptions.RequestException as e:
+        except requests_exceptions.RequestException:
             return False