fix: apply --max-document-size for directory imports too

Author: iantocristian

rethinkdb/_import.py

@@ -1085,10 +1085,6 @@ def parse_options(argv, prog=None):
             if options.custom_header is not None:
                 parser.error("--custom-header option is not valid for json files")
 
-            if options.max_document_size > 0:
-                global JSON_MAX_BUFFER_SIZE
-                JSON_MAX_BUFFER_SIZE = options.max_document_size
-
             options.file = os.path.abspath(options.file)
 
         else:
@@ -1099,6 +1095,11 @@ def parse_options(argv, prog=None):
 
     # --
 
+    # max_document_size - json
+    if options.max_document_size > 0:
+        global JSON_MAX_BUFFER_SIZE
+        JSON_MAX_BUFFER_SIZE = options.max_document_size
+
     # max_nesting_depth
     if options.max_nesting_depth > 0:
         global MAX_NESTING_DEPTH