summaryrefslogtreecommitdiff
path: root/main.py
diff options
context:
space:
mode:
Diffstat (limited to 'main.py')
-rw-r--r--main.py17
1 files changed, 16 insertions, 1 deletions
diff --git a/main.py b/main.py
index fa9db38..4afaa78 100644
--- a/main.py
+++ b/main.py
@@ -639,7 +639,6 @@ class YAML(object):
self.constructor.add_constructor(tag, f_y)
return cls
-
def parse(self, stream):
# type: (StreamTextType, Any) -> Any
"""
@@ -920,6 +919,14 @@ def load(stream, Loader=None, version=None, preserve_quotes=None):
return loader._constructor.get_single_data()
finally:
loader._parser.dispose()
+ try:
+ loader._reader.reset_reader() # type: ignore
+ except AttributeError:
+ pass
+ try:
+ loader._scanner.reset_scanner() # type: ignore
+ except AttributeError:
+ pass
def load_all(stream, Loader=None, version=None, preserve_quotes=None):
@@ -937,6 +944,14 @@ def load_all(stream, Loader=None, version=None, preserve_quotes=None):
yield loader._constructor.get_data()
finally:
loader._parser.dispose()
+ try:
+ loader._reader.reset_reader() # type: ignore
+ except AttributeError:
+ pass
+ try:
+ loader._scanner.reset_scanner() # type: ignore
+ except AttributeError:
+ pass
def safe_load(stream, version=None):