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)