Simplified check that pk is fetched in RawQuerySet.__iter__().

This commit is contained in:
Sergey Fedoseev 2017-08-02 02:05:21 +05:00 committed by Tim Graham
parent 5d9034bc92
commit 7bd9639406
1 changed files with 2 additions and 9 deletions

View File

@ -1220,15 +1220,8 @@ class RawQuerySet:
try:
model_init_names, model_init_pos, annotation_fields = self.resolve_model_init_order()
# Find out which model's fields are not present in the query.
skip = set()
for field in self.model._meta.fields:
if field.attname not in model_init_names:
skip.add(field.attname)
if skip:
if self.model._meta.pk.attname in skip:
raise InvalidQuery('Raw query must include the primary key')
if self.model._meta.pk.attname not in model_init_names:
raise InvalidQuery('Raw query must include the primary key')
model_cls = self.model
fields = [self.model_fields.get(c) for c in self.columns]
converters = compiler.get_converters([