Merge pull request #5 from JakeRoggenbuck/add-database-class

Add Database class in Rust

Author: JakeRoggenbuck

python/lstore/__init__.py

@@ -1,6 +1,7 @@
 from typing import Final
-from .lstore import hello_from_rust
+from .lstore import hello_from_rust, Database
 
+print(Database.ping())
 
 msg = hello_from_rust()
 assert "Rust" in msg

src/lib.rs

@@ -50,6 +50,20 @@ pub fn get_physical_block_size(drive: &str) -> i16 {
     read_block_size(drive, "physical")
 }
 
+#[pyclass]
+struct Database {
+    #[pyo3(get, set)]
+    page_size: usize,
+}
+
+#[pymethods]
+impl Database {
+    #[staticmethod]
+    fn ping() -> String {
+        return String::from("pong!");
+    }
+}
+
 /// Blazingly fast hello
 #[pyfunction]
 fn hello_from_rust() -> PyResult<String> {
@@ -59,6 +73,7 @@ fn hello_from_rust() -> PyResult<String> {
 /// A Python module implemented in Rust.
 #[pymodule]
 fn lstore(m: &Bound<'_, PyModule>) -> PyResult<()> {
+    m.add_class::<Database>()?;
     m.add_function(wrap_pyfunction!(hello_from_rust, m)?)?;
     Ok(())
 }