Merge pull request #58 from mbande/master

Fix retuning key attname

Author: Photonios

psqlextra/compiler.py

@@ -47,7 +47,7 @@ def _form_returning(self):
         """Builds the RETURNING part of the query."""
 
         qn = self.connection.ops.quote_name
-        return ' RETURNING %s' % qn(self.query.model._meta.pk.name)
+        return ' RETURNING %s' % qn(self.query.model._meta.pk.attname)
 
 
 class PostgresInsertCompiler(SQLInsertCompiler):
@@ -105,7 +105,7 @@ def _rewrite_insert(self, sql, params, return_id=False):
             A tuple of the rewritten SQL query and new params.
         """
 
-        returning = self.qn(self.query.model._meta.pk.name) if return_id else '*'
+        returning = self.qn(self.query.model._meta.pk.attname) if return_id else '*'
 
         if self.query.conflict_action.value == 'UPDATE':
             return self._rewrite_insert_update(sql, params, returning)