Merge pull request #38 from MoBagel/chore/mailgun-retry

add retry for mailgun request

awesome_sso/mail/mailgun.py

@@ -3,6 +3,8 @@
 
 import requests
 from pydantic import EmailStr, HttpUrl
+from requests.adapters import HTTPAdapter
+from urllib3.util.retry import Retry
 
 
 class MailGun:
@@ -25,7 +27,12 @@ def send_simple_message(
         cc: List[EmailStr] = [],
         bcc: List[EmailStr] = [],
     ):
-        return requests.post(
+        session = requests.Session()
+        retry = Retry(connect=3, backoff_factor=0.5)
+        adapter = HTTPAdapter(max_retries=retry)
+        session.mount("http://", adapter)
+        session.mount("https://", adapter)
+        return session.post(
             self.base_url + "/messages",
             auth=("api", self.api_key),
             files=[("attachment", attachment) for attachment in attachments],
@@ -51,7 +58,13 @@ def send_template(
         cc: List[EmailStr] = [],
         bcc: List[EmailStr] = [],
     ):
-        return requests.post(
+
+        session = requests.Session()
+        retry = Retry(connect=3, backoff_factor=0.5)
+        adapter = HTTPAdapter(max_retries=retry)
+        session.mount("http://", adapter)
+        session.mount("https://", adapter)
+        return session.post(
             self.base_url + "/messages",
             auth=("api", self.api_key),
             files=[("attachment", attachment) for attachment in attachments],