Merge pull request #173 from rethinkdb/fix-171-restore-primary-key

Fixing import command no primary key issue
rethinkdb · Mar 2, 2020 · 1089a24 · 1089a24
2 parents 9aa68fe + 08c2739
commit 1089a24
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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 {})
             )
         )