File tree Expand file tree Collapse file tree 2 files changed +5
-2
lines changed
src/datamodel_code_generator Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Original file line number Diff line number Diff line change 48
48
49
49
g_extend_yaml_scientifc_notation : bool = False
50
50
51
+
51
52
def get_safeloader_wrapper (safeloader : type [SafeLoader ]) -> type [SafeLoader ]:
52
53
return get_safeloader (safeloader , extend_yaml_scientifc_notation = g_extend_yaml_scientifc_notation )
53
54
Original file line number Diff line number Diff line change 16
16
from pathlib import Path
17
17
from typing import Literal
18
18
19
+ from yaml import CSafeLoader as SafeLoader
19
20
from yaml import SafeLoader
20
21
21
22
def load_toml (path : Path ) -> dict [str , Any ]: ...
22
23
23
24
else :
24
25
try :
25
- from yaml import CSafeLoader as SafeLoader
26
+ pass
26
27
except ImportError : # pragma: no cover
27
- from yaml import SafeLoader
28
+ pass
28
29
29
30
try :
30
31
from tomllib import load as load_tomllib
@@ -35,6 +36,7 @@ def load_toml(path: Path) -> dict[str, Any]:
35
36
with path .open ("rb" ) as f :
36
37
return load_tomllib (f )
37
38
39
+
38
40
def get_safeloader (safeloader : type [SafeLoader ], * , extend_yaml_scientifc_notation : bool ) -> type [SafeLoader ]:
39
41
safeloadertemp = copy .deepcopy (safeloader )
40
42
safeloadertemp .yaml_constructors = copy .deepcopy (safeloader .yaml_constructors )
You can’t perform that action at this time.
0 commit comments