2012-01-17 44 views
18

我的代码如下还挺工作,它创建的用户对象并保存,但它不保存密码:如何以编程方式创建或注册用户使用django-tastypie API?

class CreateUserResource(ModelResource): 
    class Meta: 
     allowed_methods = ['post'] 
     object_class = User 
     authentication = Authentication() 
     authorization = Authorization() 
     include_resource_uri = False 
     fields = ['username'] 

    def obj_create(self, bundle, request=None, **kwargs): 
     try: 
      bundle = super(CreateUserResource, self).obj_create(bundle, request, **kwargs) 
     except IntegrityError: 
      raise BadRequest('That username already exists') 
     return bundle 

如果我加上“密码”到元区,然后它不保存原始密码,但不散列它。我究竟做错了什么?


因此,这是对我工作:

def obj_create(self, bundle, request=None, **kwargs): 
    username, password = bundle.data['username'], bundle.data['password'] 
    try: 
     bundle.obj = User.objects.create_user(username, '', password) 
    except IntegrityError: 
     raise BadRequest('That username already exists') 
    return bundle 
+1

你怎么做验证,而不首先创建用户? – Burak 2012-08-01 10:12:40

+0

您应该调用您的类UserResource而不是CreateUserResource来尊重REST原则。它仅用于创建的事实已经在allowed_methods属性中进行了说明,并且可以添加到类名称上方的文档字符串中。 – 2014-03-06 00:19:29

+0

@DavidW。如果您需要使用不同的身份验证方法来创建和列出,那么您会如何做? – antonagestam 2014-05-12 12:43:57

回答

21

创建用户时,你需要或者使用方法set_password user.set_password(bundle.data.get('password'))或使用用户对象的create_user方法。

user = User.objects.create_user(bundle.data.get('username'), bundle.data.get('email'), bundle.data.get('password')) 

因此,像这样会为你工作:

def obj_create(self, bundle, request=None, **kwargs): 
    try: 
     bundle = super(CreateUserResource, self).obj_create(bundle, request, **kwargs) 
     bundle.obj.set_password(bundle.data.get('password')) 
     bundle.obj.save() 
    except IntegrityError: 
     raise BadRequest('That username already exists') 
    return bundle 
+0

我在这个http://www.psjinx.com/programming/2013/06/07/so-you-want-to上写了一篇博客文章-create-users-using-djangotastypie/ – pankaj28843 2013-06-07 12:31:08

4

我是在同样的情况,发现这两种解决方案帮助,但不是完整的。在他们中,我都可以用空的用户名创建用户。我不想有这样的泄漏,所以这就是我所做的。

首先,我创建了一个验证的形式:

from django import forms 
from django.forms import ModelForm 
from django.contrib.auth.models import User 

class UserForm(forms.ModelForm): 
def __init__(self, *args, **kwargs): 
    super(UserForm, self).__init__(*args, **kwargs) 

    self.fields['username'].error_messages = {'required': "Please enter username"} 
    self.fields['username'].max_length = 30 
    self.fields['password'].error_messages = {'required': 'Please enter password'} 
    self.fields['password'].max_length = 30 

    self.fields['email'].required = False 

def clean_username(self): 
    username = self.cleaned_data['username'] 
    if len(username) < 4: 
     raise forms.ValidationError("Username has to be longer than 4 characters") 
    return username 

def clean_password(self): 
    password = self.cleaned_data['password'] 
    if len(password) < 5: 
     raise forms.ValidationError("Password has to be longer than 5 characters") 
    return password 

class Meta: 
    model = User 
    fields = ('username', 'email', 'password') 

然后,UserResource

validation = FormValidation(form_class=UserForm) 

def obj_create(self, bundle, request=None, **kwargs): 
    bundle = super(UserResource, self).obj_create(bundle, request, **kwargs) 
    bundle.obj.set_password(bundle.data.get('password')) 
    bundle.obj.save() 

    return bundle 

我希望我的解决方案可以帮助其他开发中。快乐的编码!

2

我正在尝试使用类似的代码与django-tastypie == 0.9.12,并进入错误queryset和obj_create参数的数量。使用下面的代码为我工作:

from django.contrib.auth.models import User 
from django.db import IntegrityError 

from tastypie.resources import ModelResource 
from tastypie.authorization import Authorization 
from tastypie.authentication import Authentication 
from tastypie import fields 
from tastypie.exceptions import BadRequest 

class UserSignUpResource(ModelResource): 
    class Meta: 
     object_class = User 
     resource_name = 'register' 
     fields = ['username', 'first_name', 'last_name', 'email'] 
     allowed_methods = ['post'] 
     include_resource_uri = False 
     authentication = Authentication() 
     authorization = Authorization() 
     queryset = User.objects.all() 

    def obj_create(self, bundle, request=None, **kwargs): 
     try: 
      bundle = super(UserSignUpResource, self).obj_create(bundle) 
      bundle.obj.set_password(bundle.data.get('password')) 
      bundle.obj.save() 
     except IntegrityError: 
      raise BadRequest('Username already exists') 

     return bundle 

一些测试代码是:

from django.contrib.auth.models import User 
from django.contrib.auth.hashers import check_password 

from tastypie.test import ResourceTestCase 

class UserSignUpResourceTest(ResourceTestCase): 
    def test_post_list(self): 
     post_arguments = { 
      "email": "[email protected]", 
      "first_name": "John", 
      "last_name": "Doe", 
      "username": "test-user", 
      "password": "idiotic-pwd" 
     } 
     # Check how many are there 
     self.assertEqual(User.objects.count(), 0) 
     self.assertHttpCreated(self.api_client.post('/api/register/', 
     format='json', data=post_arguments)) 
     # Check how many are there. Should be one more 
     self.assertEqual(User.objects.count(), 1) 
     # Check attributes got saved correctly 
     user = User.objects.get(username='test-user') 
     for atr in post_arguments: 
      if atr == 'password': 
       check_password(post_arguments[atr], getattr(user, atr)) 
      else: 
       self.assertEqual(post_arguments[atr], getattr(user, atr)) 
+0

非常感谢你这篇文章,它对于获取为.9.12创建的用户注册资源非常有帮助 – 2015-03-31 00:20:48

相关问题