allow visitor to visit primitives at the root level

daggaz · Jan 19, 2025 · 27f0614 · 27f0614
1 parent cbc9111
commit 27f0614
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/json_stream/visitor.py b/src/json_stream/visitor.py
@@ -1,6 +1,7 @@
 from json_stream.base import StreamingJSONObject, StreamingJSONList, StreamingJSONBase
 from json_stream.iterators import ensure_file
 from json_stream.select_tokenizer import default_tokenizer
+from json_stream.tokenizer import TokenType
 
 
 def _visit(obj, visitor, path):
@@ -22,6 +23,9 @@ def _visit(obj, visitor, path):
 def visit(fp_or_iterator, visitor, tokenizer=default_tokenizer):
     fp = ensure_file(fp_or_iterator)
     token_stream = tokenizer(fp)
-    _, token = next(token_stream)
-    obj = StreamingJSONBase.factory(token, token_stream, persistent=False)
+    token_type, token = next(token_stream)
+    if token_type == TokenType.OPERATOR:
+        obj = StreamingJSONBase.factory(token, token_stream, persistent=False)
+    else:
+        obj = token
     _visit(obj, visitor, ())