diff options
Diffstat (limited to 'python/mozrelease/mozrelease/util.py')
-rw-r--r-- | python/mozrelease/mozrelease/util.py | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/python/mozrelease/mozrelease/util.py b/python/mozrelease/mozrelease/util.py new file mode 100644 index 0000000000..3858c40514 --- /dev/null +++ b/python/mozrelease/mozrelease/util.py @@ -0,0 +1,26 @@ +# -*- coding: utf-8 -*- +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this +# file, You can obtain one at http://mozilla.org/MPL/2.0/. + +from yaml.loader import SafeLoader + + +class UnicodeLoader(SafeLoader): + def construct_yaml_str(self, node): + return self.construct_scalar(node) + + +UnicodeLoader.add_constructor("tag:yaml.org,2002:str", UnicodeLoader.construct_yaml_str) + + +def load(stream): + """ + Parse the first YAML document in a stream + and produce the corresponding Python object. + """ + loader = UnicodeLoader(stream) + try: + return loader.get_single_data() + finally: + loader.dispose() |