From a990e4568c7f341cc76f5ab572a3089bb1065da4 Mon Sep 17 00:00:00 2001 From: Anthon van der Neut Date: Wed, 6 Jul 2016 23:31:57 +0200 Subject: added optional preservation of quotes around scalars --- main.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'main.py') diff --git a/main.py b/main.py index e0041bc..797bdcd 100644 --- a/main.py +++ b/main.py @@ -65,12 +65,12 @@ def compose_all(stream, Loader=Loader): loader.dispose() -def load(stream, Loader=Loader, version=None): +def load(stream, Loader=Loader, version=None, preserve_quotes=None): """ Parse the first YAML document in a stream and produce the corresponding Python object. """ - loader = Loader(stream, version) + loader = Loader(stream, version, preserve_quotes=preserve_quotes) try: return loader.get_single_data() finally: @@ -108,22 +108,22 @@ def safe_load_all(stream, version=None): return load_all(stream, SafeLoader, version) -def round_trip_load(stream, version=None): +def round_trip_load(stream, version=None, preserve_quotes=None): """ Parse the first YAML document in a stream and produce the corresponding Python object. Resolve only basic YAML tags. """ - return load(stream, RoundTripLoader, version) + return load(stream, RoundTripLoader, version, preserve_quotes=preserve_quotes) -def round_trip_load_all(stream, version=None): +def round_trip_load_all(stream, version=None, preserve_quotes=None): """ Parse all YAML documents in a stream and produce corresponding Python objects. Resolve only basic YAML tags. """ - return load_all(stream, RoundTripLoader, version) + return load_all(stream, RoundTripLoader, version, preserve_quotes=preserve_quotes) def emit(events, stream=None, Dumper=Dumper, -- cgit v1.2.1