Address #2954 for C++, C, Python, Node.js, and Java APIs (#2958)

mewim · web-flow · commit 85bdc9d2f8a8 · 2024-02-28T12:11:10.000+08:00
diff --git a/src_cpp/include/py_database.h b/src_cpp/include/py_database.h
@@ -16,6 +16,10 @@ class PyDatabase {
 
     static void initialize(py::handle& m);
 
+    static py::str getVersion();
+
+    static uint64_t getStorageVersion();
+
     explicit PyDatabase(const std::string& databasePath, uint64_t bufferPoolSize,
         uint64_t maxNumThreads, bool compression, bool readOnly, uint64_t maxDBSize);
 
diff --git a/src_cpp/py_database.cpp b/src_cpp/py_database.cpp
@@ -1,8 +1,10 @@
 #include "include/py_database.h"
-#include "pandas/pandas_scan.h"
 
 #include <memory>
 
+#include "main/version.h"
+#include "pandas/pandas_scan.h"
+
 using namespace kuzu::common;
 
 void PyDatabase::initialize(py::handle& m) {
@@ -26,7 +28,17 @@ void PyDatabase::initialize(py::handle& m) {
         .def("scan_node_table_as_float", &PyDatabase::scanNodeTable<float>, py::arg("table_name"),
             py::arg("prop_name"), py::arg("indices"), py::arg("np_array"), py::arg("num_threads"))
         .def("scan_node_table_as_bool", &PyDatabase::scanNodeTable<bool>, py::arg("table_name"),
-            py::arg("prop_name"), py::arg("indices"), py::arg("np_array"), py::arg("num_threads"));
+            py::arg("prop_name"), py::arg("indices"), py::arg("np_array"), py::arg("num_threads"))
+        .def_static("get_version", &PyDatabase::getVersion)
+        .def_static("get_storage_version", &PyDatabase::getStorageVersion);
+}
+
+py::str PyDatabase::getVersion() {
+    return py::str(Version::getVersion());
+}
+
+uint64_t PyDatabase::getStorageVersion() {
+    return Version::getStorageVersion();
 }
 
 PyDatabase::PyDatabase(const std::string& databasePath, uint64_t bufferPoolSize,
diff --git a/src_py/__init__.py b/src_py/__init__.py
@@ -51,6 +51,15 @@
 from .query_result import *
 from .types import *
 
+def __getattr__(name):
+    if name == "version":
+        return Database.get_version()
+    elif name == "storage_version":
+        return Database.get_storage_version()
+    else:
+        raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+    
+
 # Restore the original dlopen flags
 if sys.platform == "linux":
     sys.setdlopenflags(original_dlopen_flags)
diff --git a/src_py/database.py b/src_py/database.py
@@ -50,6 +50,28 @@ def __init__(self, database_path, buffer_pool_size=0, max_num_threads=0, compres
         self._database = None
         if not lazy_init:
             self.init_database()
+    
+    def get_version():
+        """
+        Get the version of the database.
+
+        Returns
+        -------
+        str
+            The version of the database.
+        """
+        return _kuzu.Database.get_version()
+    
+    def get_storage_version():
+        """
+        Get the storage version of the database.
+
+        Returns
+        -------
+        int
+            The storage version of the database.
+        """
+        return _kuzu.Database.get_storage_version()
 
     def __getstate__(self):
         state = {
diff --git a/test/test_version.py b/test/test_version.py
@@ -0,0 +1,4 @@
+def test_version():
+    import kuzu
+    assert kuzu.version != ""
+    assert kuzu.storage_version > 0
