通过中间表从多个对象中获取ManyToMany对象

时间:2009-04-11 02:28:11

标签: django django-models django-templates

是否有一种从返回多个对象的查询中获取ManyToMany对象的简单方法?我现在这样做的方式并不像我希望的那样性感。以下是我现在的观点:

contacts = Contact.objects.all()
# Use Custom Manager Method to Fetch Each Contacts Phone Numbers
contacts = PhoneNumber.objects.inject(contacts)

我的模特:

class PhoneNumber(models.Model):
    number = models.CharField()
    type = models.CharField()

    # My Custom Manager
    objects = PhoneNumberManager()

class Contact(models.Model):
    name = models.CharField()
    numbers = models.ManyToManyField(PhoneNumber, through='ContactPhoneNumbers')

class ContactPhoneNumbers(models.Model):
    number = models.ForeignKey(PhoneNumber)
    contact = models.ForeignKey(Contact)
    ext = models.CharField()

我的自定义管理器:

class PhoneNumberManager(models.Manager):
    def inject(self, contacts):
        contact_ids = ','.join([str(item.id) for item in contacts])
        cursor = connection.cursor()

        cursor.execute("""
            SELECT l.contact_id, l.ext, p.number, p.type
            FROM svcontact_contactphonenumbers l, svcontact_phonenumber p
            WHERE p.id = l.number_id AND l.contact_id IN(%s)
            """ % contact_ids)

        result = {}
        for row in cursor.fetchall():
            id = str(row[0])
            if not id in result:
                result[id] = []

            result[id].append({
                 'ext': row[1],
                 'number': row[2],
                 'type': row[3]
            })

        for contact in contacts:
            id = str(contact.id)
            if id in result:
                contact.phonenumbers = result[id]

        return contacts

1 个答案:

答案 0 :(得分:3)

你可以做几件事来发现性感: - )

Django没有任何OOTB方法将through表的属性注入到Contact实例中。带有额外数据的M2M表是一个SQL概念,因此Django不会尝试对抗关系,也不会猜测在命名空间冲突时应该发生什么等等。事实上,我甚至可以说你可能不希望在Contact对象上注入任意模型属性......如果你发现自己需要这样做,那么这可能是你应该修改模型定义的一个标志

相反,Django提供了无缝访问关系的便捷方式,包括查询和数据检索,同时保留了实体的完整性。在这种情况下,您会发现您的Contact对象提供了一个contactphonenumbers_set属性,您可以使用该属性来访问直通数据:

>>> c = Contact.objects.get(id=1)
>>> c.contactphonenumbers_set.all()
# Would produce a list of ContactPhoneNumbers objects for that contact

这意味着,在您的情况下,要迭代所有联系电话号码(例如),您将:

for contact in Contact.objects.all():
    for phone in contact.contactphonenumbers_set.all():
        print phone.number.number, phone.number.type, phone.ext

如果你真的,真的,真的想要出于某种原因进行注射,你会发现你可以使用上面的3行代码示例来做到这一点:只需将print语句更改为赋值语句。


另外,为了将来的参考,你可以编写没有SQL语句的inject函数。在Django中,直通表本身就是一个模型,因此您可以直接查询它:

def inject(self, contacts):
    contact_phone_numbers = ContactPhoneNumbers.objects.\
                            filter(contact__in=contacts)
    # And then do the result construction... 
    # - use contact_phone_number.number.phone to get the phone and ext
    # - use contact_phone_number.contact to get the contact instance