diff --git a/rethinkdb/_import.py b/rethinkdb/_import.py index 107654e7..6696e95a 100755 --- a/rethinkdb/_import.py +++ b/rethinkdb/_import.py @@ -146,7 +146,11 @@ def __init__( self.write_hook = write_hook or [] # options - self.source_options = source_options or {} + self.source_options = source_options or { + "create_args": { + "primary_key": self.primary_key + } + } # name if hasattr(self._source, 'name') and self._source.name: @@ -249,7 +253,7 @@ def setup_table(self): ast.expr([self.table]).set_difference( query.db(self.db).table_list() ).for_each(query.db(self.db).table_create( - query.row, **self.source_options.create_args if 'create_args' in self.source_options else {}) + query.row, **self.source_options["create_args"] if 'create_args' in self.source_options else {}) ) )