8

如何配置 Tastypie 以将字段视为唯一字段?如果我尝试为标记为唯一的字段插入重复条目,我的期望是收到某种非 500 错误(可能是 409 冲突?)作为响应。


我浏览了文档,看起来对我来说应该很明显,但由于某种原因,我没有得到我期望看到的响应。

这是文档链接:

http://readthedocs.org/docs/django-tastypie/en/latest/fields.html?highlight=unique


示例代码如下:

网址.py

v1_api = Api(api_name='v1')
v1_api.register(CompanyResource())

urlpatterns = patterns('',
    (r'^api/', include(v1_api.urls)),
)

资源.py

class CompanyResource(ModelResource):

    CompanyName = fields.CharField(attribute='company_name')
    CompanyId = fields.CharField(attribute='company_id', unique=True)
    Contact = fields.CharField(attribute='contact')
    Email = fields.CharField(attribute='email')
    Phone = fields.CharField(attribute='phone')

    class Meta:
        queryset = Company.objects.all()
        authentication = BasicAuthentication()
        authorization = Authorization()
        allowed_methods = ['get', 'post']

模型.py

class Company(models.Model):

    company_name = models.TextField(default=None, blank=True, null=True)
    company_id = models.CharField(default='', unique=True, db_index=True, max_length=20)
    contact = models.TextField(default=None, blank=True, null=True)
    email = models.EmailField(default=None, blank=True, null=True)
    phone = models.TextField(default=None, blank=True, null=True)

我收到的错误如下(使用 curl 访问我的本地服务):

curl --dump-header - -H "Content-Type: application/json" -X POST --user user:password --data '{"CompanyName": "company", "CompanyId": "1234567890", "Contact": "John", "Email": "example@example.com", "Phone": "555-555-5555"}' http://localhost:8000/api/v1/company/
HTTP/1.0 500 INTERNAL SERVER ERROR
Date: Thu, 15 Sep 2011 18:25:20 GMT
Server: WSGIServer/0.1 Python/2.7.1
Content-Type: application/json; charset=utf-8

{"error_message": "(1062, \"Duplicate entry '1234567890' for key 'api_company_company_id_uniq'\")", 
...<snip>...
raise errorclass, errorvalue\n\nIntegrityError: (1062, \"Duplicate entry '1234567890' for key 'api_company_company_id_uniq'\")\n"}

当我unique=True, db_index=True,从公司模型中删除时,我没有收到完整性错误,而是创建了一个新的重复资源。同样,这不是预期的结果,因为我希望独特地执行一些验证并导致一些非 500 响应。

4

3 回答 3

8

这是我解决问题的方法:

根据验证文档,我能够实现一个自定义验证器,为我检查字段的唯一性。 http://django-tastypie.readthedocs.org/en/latest/validation.html

在 CompanyResource 中,我向元类添加了一个 CustomValidation。我将 CustomValidation 的实现放在了 validations.py 文件中。如果 isValid 返回错误,则 api 将返回 400,其中包含错误中的消息。

class CompanyResource(ModelResource):
    """
    CompanyIds should be unique
    """     
    CompanyName = fields.CharField(attribute='company_name')     
    CompanyId = fields.CharField(attribute='company_id', unique=True)     
    Contact = fields.CharField(attribute='contact')     
    Email = fields.CharField(attribute='email')     
    Phone = fields.CharField(attribute='phone')    

    class Meta:        
        queryset = Company.objects.all()        
        authentication = BasicAuthentication()        
        authorization = Authorization()        
        allowed_methods = ['get', 'post']                
        validation = CustomValidation()

验证.py

class CustomValidation(Validation):
    """
    The custom validation checks two things:
       1) that there is data
       2) that the CompanyId exists (unique check)
    """
    def is_valid(self, bundle, request=None):
        if not bundle.data:
            return {'__all__': 'Missing data, please include CompanyName, CompanyId, Contact, Email, and Phone.'}

        errors = {}                                    
        company_id=bundle.data.get('CompanyId', None)

        # manager method, returns true if the company exists, false otherwise
        if Company.objects.company_exists(company_id):
            errors['CompanyId']='Duplicate CompanyId, CompanyId %s already exists.' % company_id
        return errors
于 2011-09-21T10:14:20.613 回答
5

我今天遇到了同样的问题。这是我的处理方式:

在资源定义中覆盖 [request_method]_[request_type] 方法。例如,我覆盖 FooResource 中的 post_list:

def post_list(self, request, **kwargs):
    from django.db import IntegrityError
    try:
        return super(FooResource, self).post_list(request, **kwargs)
    except IntegrityError, e:
        if e.args[0] == 1062:
            return http.HttpConflict()

希望对你有效。

于 2012-01-09T05:57:31.960 回答
0

对于它的价值,我创建了一个稍微不同的解决方案,对我来说效果更好。它基于thoslin的回答。

我发现 e.args[0] == 1062 检查对我来说失败了。我很确定这是一个 MySQL 错误,我正在使用 Postgres。

我还在 obj_create 方法中实现了这一点,以便它将处理所有对象创建,而不仅仅是通过 post_list 完成的。

from tastypie.exceptions import ImmediateHttpResponse
from tastypie.http import HttpConflict
from django.db import IntegrityError

...

class MyBaseResource(ModelResource):
    def obj_create(self, bundle, **kwargs):
        try:
            return super(MyBaseResource, self).obj_create(bundle, **kwargs)
        except IntegrityError, e:
            if e.args[0] == 1062 or e.args[0].startswith('duplicate key'):
                raise ImmediateHttpResponse(HttpConflict())

...

class MyResource(MyBaseResource):
    [usual resource stuff]
于 2014-04-09T03:26:11.510 回答