Fix version CLI showing 0.0.0 (#2305)

Signed-off-by: Bernát Gábor <[email protected]>
koxudaxi · Feb 6, 2025 · 2eac729 · 2eac729
1 parent 94c21e6
commit 2eac729
Show file tree

Hide file tree

Showing 4 changed files with 108 additions and 104 deletions.
diff --git a/datamodel_code_generator/__main__.py b/datamodel_code_generator/__main__.py
@@ -399,9 +399,9 @@ def main(args: Optional[Sequence[str]] = None) -> Exit:
     arg_parser.parse_args(args, namespace=namespace)
 
     if namespace.version:
-        from datamodel_code_generator.version import version
+        from datamodel_code_generator import get_version
 
-        print(version)
+        print(get_version())
         exit(0)
 
     pyproject_config = _get_pyproject_toml_config(Path().resolve())

diff --git a/datamodel_code_generator/version.py b/datamodel_code_generator/version.py
diff --git a/tests/main/jsonschema/test_main_jsonschema.py b/tests/main/jsonschema/test_main_jsonschema.py
@@ -2458,7 +2458,7 @@ def test_version(capsys):
         main(['--version'])
     assert e.value.code == Exit.OK
     captured = capsys.readouterr()
-    assert captured.out == '0.0.0\n'
+    assert captured.out != '0.0.0\n'
     assert captured.err == ''