core/auth/email.py

46 lines
1.4 KiB
Python
Raw Normal View History

2021-08-25 08:31:51 +00:00
import requests
2021-08-25 13:39:24 +00:00
from starlette.responses import PlainTextResponse
from starlette.exceptions import HTTPException
2021-08-25 08:31:51 +00:00
from auth.authenticate import EmailAuthenticate
2021-08-25 17:12:01 +00:00
from settings import BACKEND_URL, MAILGUN_API_KEY, MAILGUN_DOMAIN
2021-08-25 08:31:51 +00:00
MAILGUN_API_URL = "https://api.mailgun.net/v3/%s/messages" % (MAILGUN_DOMAIN)
MAILGUN_FROM = "postmaster <postmaster@%s>" % (MAILGUN_DOMAIN)
2021-08-25 17:12:01 +00:00
AUTH_URL = "%s/email_authorize" % (BACKEND_URL)
2021-08-25 08:31:51 +00:00
2021-08-26 09:24:46 +00:00
async def send_confirm_email(user):
text = "<html><body>To confirm registration follow the <a href='%s'>link</link></body></html>"
await send_email(user, text)
2021-08-25 08:31:51 +00:00
async def send_auth_email(user):
2021-08-26 09:24:46 +00:00
text = "<html><body>To enter the site follow the <a href='%s'>link</link></body></html>"
await send_email(user, text)
async def send_email(user, text):
2021-08-25 08:31:51 +00:00
token = await EmailAuthenticate.get_email_token(user)
to = "%s <%s>" % (user.username, user.email)
2021-08-26 09:24:46 +00:00
auth_url_with_token = "%s?token=%s" % (AUTH_URL, token)
text = text % (auth_url_with_token)
2021-08-25 08:31:51 +00:00
response = requests.post(
MAILGUN_API_URL,
auth = ("api", MAILGUN_API_KEY),
data = {
"from": MAILGUN_FROM,
"to": to,
"subject": "authorize log in",
2021-08-26 09:24:46 +00:00
"html": text
2021-08-25 08:31:51 +00:00
}
)
response.raise_for_status()
2021-08-25 13:39:24 +00:00
async def email_authorize(request):
token = request.query_params.get('token')
if not token:
raise HTTPException(500, "invalid url")
auth_token, user = await EmailAuthenticate.authenticate(token)
return PlainTextResponse(auth_token)