示例#1
0
def prepare_confirmation(user, token):
    """
    Prepare to confirmation.
    """
    tokens_dict = get_confirmation_dict()
    tokens_dict[token] = {'user_id': user.pk}
    
    if settings.USE_CELERY:
        send_confirmation_email.delay(user, token)
    else:
        send_confirmation_email(user, token)
示例#2
0
文件: demo.py 项目: tangerine26/ataxi
    def create_user_by_api(self, **user_data):
        register_url = reverse('user-register')
        response = self._client.post(register_url, user_data, format='json')
        try:
            content_data = json.loads(response.content)
        except ValueError:
            raise AssertionError(
                _(u'Invalid response content format, not a JSON'))

        user_id = content_data.get('id')
        if not user_id:
            raise AssertionError(_(u'Not a User ID in data'))

        confirm_token = content_data.get('confirm_token')
        if not confirm_token:
            raise AssertionError(_(u'Not a Confirm Token in data'))

        try:
            new_user = User.objects.get(pk=user_id)
        except User.DoesNotExist:
            raise AssertionError(_(u'User has not been created'))
        else:
            if new_user.is_active:
                 raise AssertionError(
                     _(u'New User is active after registration'))

        tokens_dict = get_confirmation_dict()
        user_info = tokens_dict.get(confirm_token)
        if not user_info:
            raise AssertionError(_(u'Confirmation token has not beed added'))
                     
        if not isinstance(user_info, dict) or 'user_id' not in user_info:
            raise AssertionError(_(u'Invalid confirmation token'))

        confirm_url = reverse('user-confirm', args=[confirm_token])
        response = self._client.get(confirm_url, format='json')

        assert response.status_code == status.HTTP_200_OK, \
            _(u'Invalid HTTp response code')

        try:
            new_user = User.objects.get(pk=user_id)
        except User.DoesNotExist:
            raise AssertionError(_(u'User has not been created'))
        else:
            if not new_user.is_active:
                 raise AssertionError(
                     _(u'New User is not active after confirmation'))
            return new_user
        return None
示例#3
0
    def test_user_register(self):
        """
        Test the new User registration.
        """
        register_url = reverse('user-register')
        response = self._client.post(register_url, NEW_USER_DATA, format='json')
        try:
            content_data = json.loads(response.content)
        except ValueError:
            raise AssertionError(
                _(u'Invalid response content format, not a JSON'))

        user_id = content_data.get('id')
        if not user_id:
            raise AssertionError(_(u'Not a User ID in data'))

        confirm_token = content_data.get('confirm_token')
        if not confirm_token:
            raise AssertionError(_(u'Not a Confirm Token in data'))

        try:
            new_user = User.objects.get(pk=user_id)
        except User.DoesNotExist:
            raise AssertionError(_(u'User has not been created'))
        else:
            if new_user.is_active:
                 raise AssertionError(
                     _(u'New User is active after registration'))

        tokens_dict = get_confirmation_dict()
        user_info = tokens_dict.get(confirm_token)
        if not user_info:
            raise AssertionError(_(u'Confirmation token has not beed added'))
                     
        if not isinstance(user_info, dict) or 'user_id' not in user_info:
            raise AssertionError(_(u'Invalid confirmation token'))
        
        confirm_url = reverse('user-confirm', args=[confirm_token])
        response = self._client.get(confirm_url, format='json')
        
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        try:
            new_user = User.objects.get(pk=user_id)
        except User.DoesNotExist:
            raise AssertionError(_(u'User has not been created'))
        else:
            if not new_user.is_active:
                 raise AssertionError(
                     _(u'New User is not active after confirmation'))
示例#4
0
def confirm(request, token):
    """
    Confirn the new User.
    """
    
    # Get default Redis List for confirmation tokens
    tokens_dict = get_confirmation_dict()
    
    # Get user info from list
    user_info = tokens_dict.get(token)
    result = False
    if user_info and isinstance(user_info, dict) and 'user_id' in user_info:
        # Retrieve the User by Id
        try:
            user = User.objects.get(pk=user_info['user_id'])
        except User.DoesNotExist:
            pass
        else:
            # Activate User
            user.is_active = True
            user.save()
            tokens_dict.discard(token)
            result = True
    return {'result_ok': result}