mirror of
https://gitcode.com/gh_mirrors/eas/EasyFace.git
synced 2025-12-30 04:52:28 +00:00
25 lines
668 B
Python
25 lines
668 B
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
import yaml
|
|
|
|
try:
|
|
from yaml import CDumper as Dumper
|
|
from yaml import CLoader as Loader
|
|
except ImportError:
|
|
from yaml import Loader, Dumper # type: ignore
|
|
|
|
from .base import FormatHandler # isort:skip
|
|
|
|
|
|
class YamlHandler(FormatHandler):
|
|
def load(self, file, **kwargs):
|
|
kwargs.setdefault('Loader', Loader)
|
|
return yaml.load(file, **kwargs)
|
|
|
|
def dump(self, obj, file, **kwargs):
|
|
kwargs.setdefault('Dumper', Dumper)
|
|
yaml.dump(obj, file, **kwargs)
|
|
|
|
def dumps(self, obj, **kwargs):
|
|
kwargs.setdefault('Dumper', Dumper)
|
|
return yaml.dump(obj, **kwargs)
|