Address some code style issues

Author: Maximilian Sander

src/datamodel_code_generator/__init__.py

@@ -30,7 +30,7 @@
     PythonVersionMin,
 )
 from datamodel_code_generator.parser import DefaultPutDict, LiteralType
-from datamodel_code_generator.util import SafeLoader, get_SafeLoader
+from datamodel_code_generator.util import SafeLoader, get_safeloader
 
 MIN_VERSION: Final[int] = 9
 MAX_VERSION: Final[int] = 13
@@ -46,12 +46,14 @@
 
 DEFAULT_BASE_CLASS: str = "pydantic.BaseModel"
 
-g_extend_yaml_scientifc_notation = False
-def get_SafeLoader_(SafeLoader):
-    return get_SafeLoader(SafeLoader, g_extend_yaml_scientifc_notation)
+g_extend_yaml_scientifc_notation: bool = False
+
+def get_safeloader_wrapper(safeloader: type[SafeLoader]) -> type[SafeLoader]:
+    return get_safeloader(safeloader, extend_yaml_scientifc_notation=g_extend_yaml_scientifc_notation)
+
 
 def load_yaml(stream: str | TextIO) -> Any:
-    return yaml.load(stream, Loader=get_SafeLoader_(SafeLoader))  # noqa: S506
+    return yaml.load(stream, Loader=get_safeloader_wrapper(SafeLoader))  # noqa: S506
 
 
 def load_yaml_from_path(path: Path, encoding: str) -> Any:

src/datamodel_code_generator/util.py

@@ -35,26 +35,30 @@ def load_toml(path: Path) -> dict[str, Any]:
         with path.open("rb") as f:
             return load_tomllib(f)
 
-def get_SafeLoader(SafeLoader, extend_yaml_scientifc_notation=False):
-    SafeLoaderTemp = copy.deepcopy(SafeLoader)
-    SafeLoaderTemp.yaml_constructors = copy.deepcopy(SafeLoader.yaml_constructors)
-    SafeLoaderTemp.add_constructor(
+def get_safeloader(safeloader: type[SafeLoader], *, extend_yaml_scientifc_notation: bool) -> type[SafeLoader]:
+    safeloadertemp = copy.deepcopy(safeloader)
+    safeloadertemp.yaml_constructors = copy.deepcopy(safeloader.yaml_constructors)
+    safeloadertemp.add_constructor(
         "tag:yaml.org,2002:timestamp",
-        SafeLoaderTemp.yaml_constructors["tag:yaml.org,2002:str"],
+        safeloadertemp.yaml_constructors["tag:yaml.org,2002:str"],
     )
     if extend_yaml_scientifc_notation is True:
-        SafeLoaderTemp.add_implicit_resolver(
-            u'tag:yaml.org,2002:float',
-            re.compile(u'''^(?:
+        safeloadertemp.add_implicit_resolver(
+            "tag:yaml.org,2002:float",
+            re.compile(
+                r"""^(?:
             [-+]?(?:[0-9][0-9_]*)\\.[0-9_]*(?:[eE][-+]?[0-9]+)?
             |[-+]?(?:[0-9][0-9_]*)(?:[eE][-+]?[0-9]+)
             |\\.[0-9_]+(?:[eE][-+][0-9]+)?
             |[-+]?[0-9][0-9_]*(?::[0-5]?[0-9])+\\.[0-9_]*
             |[-+]?\\.(?:inf|Inf|INF)
-            |\\.(?:nan|NaN|NAN))$''', re.X),
-            list(u'-+0123456789.')
+            |\\.(?:nan|NaN|NAN))$""",
+                re.VERBOSE,
+            ),
+            list("-+0123456789."),
         )
-    return SafeLoaderTemp
+    return safeloadertemp
+
 
 Model = TypeVar("Model", bound=_BaseModel)