Skip to content

Commit 64a021d

Browse files
committed
chore: Handle timeout
1 parent 5d2df80 commit 64a021d

1 file changed

Lines changed: 9 additions & 4 deletions

File tree

src/maily/locker_api.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,9 @@ def store_reply_record(mail, ses_response):
6464
try:
6565
requests.post(url=url, json=payload, headers=HEADERS, timeout=10)
6666
return True
67-
except (requests.exceptions.ConnectionError, requests.exceptions.ConnectTimeout, KeyError):
67+
except (requests.exceptions.ConnectionError, requests.exceptions.ConnectTimeout, requests.exceptions.Timeout,
68+
KeyError):
69+
time.sleep(10)
6870
retry += 1
6971
return False
7072

@@ -87,7 +89,8 @@ def check_user_premium(email):
8789
try:
8890
r = requests.get(url, headers=HEADERS, timeout=10).json()
8991
return r["is_premium"]
90-
except (requests.exceptions.ConnectionError, requests.exceptions.ConnectTimeout, KeyError):
92+
except (requests.exceptions.ConnectionError, requests.exceptions.ConnectTimeout, requests.exceptions.Timeout,
93+
KeyError):
9194
return False
9295

9396

@@ -103,7 +106,8 @@ def get_relay_address_plan(relay_address):
103106
try:
104107
r = requests.get(url, headers=HEADERS, timeout=10).json()
105108
return r
106-
except (requests.exceptions.ConnectionError, requests.exceptions.ConnectTimeout, KeyError):
109+
except (requests.exceptions.ConnectionError, requests.exceptions.ConnectTimeout, requests.exceptions.Timeout,
110+
KeyError):
107111
return {}
108112

109113

@@ -119,5 +123,6 @@ def send_statistic_relay_address(relay_address, statistic_type):
119123
if r.status_code >= 400:
120124
return False
121125
return True
122-
except (requests.exceptions.ConnectionError, requests.exceptions.ConnectTimeout, KeyError):
126+
except (requests.exceptions.ConnectionError, requests.exceptions.ConnectTimeout, requests.exceptions.Timeout,
127+
KeyError):
123128
return False

0 commit comments

Comments
 (0)