def get(self):
try:
token = request.user.get_verification()
except Exception as e:
raise APIException(e.message)
data = {
'username': request.user.username,
'url': urljoin(config.SITE_URL, config.VERIFY_URL),
'token': token,
}
send_mail_template('verification', [request.user.email], data=data)
return 'send verification email'
def get_verification():
try:
token = request.user.get_verification()
except Exception as exc:
return send_error(exc.message)
data = {
'username': request.user.username,
'url': urljoin(config.SITE_URL, config.VERIFY_URL),
'token': token,
}
send_mail_template('verification', [request.user.email], data=data)
return send_success()
def post(self):
try:
user = User.get(User.email == self.data['email'])
except User.DoesNotExist:
raise APINotFound('hey email')
token = user.get_reset()
data = {
'username': user.username,
'url': urljoin(config.SITE_URL, config.RECOVER_URL),
'token': token,
}
send_mail_template('reset', [user.email], data=data)
return 'send recover email'
def post(self):
username = self.data['username']
password = self.data['password']
email = self.data['email']
user = User.add(username, password, email)
if email:
token = user.get_verification()
data = {
'username': username,
'url': urljoin(config.SITE_URL, config.VERIFY_URL),
'token': token,
}
send_mail_template('registration', [email], data=data)
return 'registration successful'
def reset(data):
try:
user = User.get(User.email == data['email'])
except User.DoesNotExist:
return send_error('email not found')
else:
token = user.get_reset()
data = {
'username': user.username,
'url': urljoin(config.SITE_URL, config.RECOVER_URL),
'token': token,
}
send_mail_template('reset', [user.email], data=data)
return send_success()
def test_send_mail_template(self, send_mail, render_template):
with app.test_request_context():
send_mail_template('registration', ['user1@mail'])
self.assertEqual(send_mail.call_count, 1)
self.assertEqual(render_template.call_count, 3)