From 08c2739e1edfd7a017f2f22bf8d662a6e8e5a150 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?G=C3=A1bor=20Boros?= Date: Sat, 22 Feb 2020 09:11:54 +0100 Subject: [PATCH] Fixing import command no primary key issue --- rethinkdb/_import.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) 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 {}) ) )