Use pk instead of id

Author: ei-grad

elasticutils/contrib/django/tasks.py

@@ -54,15 +54,15 @@ def update_in_index(sender, instance, **kw):
     for id_list in chunked(ids, chunk_size):
         documents = []
 
-        for obj in model.objects.filter(id__in=id_list):
+        for obj in model.objects.filter(pk__in=id_list):
             try:
-                documents.append(mapping_type.extract_document(obj.id, obj))
+                documents.append(mapping_type.extract_document(obj.pk, obj))
             except Exception as exc:
                 log.exception('Unable to extract document {0}: {1}'.format(
                               obj, repr(exc)))
 
         if documents:
-            mapping_type.bulk_index(documents, id_field='id', es=es, index=index)
+            mapping_type.bulk_index(documents, id_field=model._meta.pk.name, es=es, index=index)
 
 
 @task