Mirror/app-MAIL-temp
1
0
Fork 0
mirror of https://github.com/simple-login/app.git synced 2024-11-13 07:31:12 +01:00
Code Issues Projects Releases Packages Wiki Activity

wrap shield around aiospamc.check to avoid the task being canceled

Browse source
This commit is contained in:
Son NK 2020-09-16 19:47:12 +02:00
parent 4b697fc897
commit 5f8fff5af3
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
email_handler.py
View file

@ -36,6 +36,7 @@ import email
import os
import time
import uuid
from asyncio import shield
from email import encoders
from email.message import Message
from email.mime.application import MIMEApplication
@ -1466,7 +1467,7 @@ async def get_spam_score(message: Message) -> float:
try:
# wait for at max 300s which is the default spamd timeout-child
response = await asyncio.wait_for(
aiospamc.check(sa_input, host=SPAMASSASSIN_HOST), timeout=300
shield(aiospamc.check(sa_input, host=SPAMASSASSIN_HOST)), timeout=300
)
return response.headers["Spam"].score
except asyncio.TimeoutError: