WIP

Cargo.toml

@@ -2,6 +2,8 @@
 name = "etcd-client-py"
 version = "0.1.1"
 edition = "2021"
+authors = ["Lablup Inc."]
+readme = "./README.md"
 
 [lib]
 name = "etcd_client"

Makefile

@@ -0,0 +1,15 @@
+build:
+	maturin build
+
+install:
+	maturin build
+	pip install .
+
+test:
+	python -m pytest
+
+fmt:
+	cargo fmt
+
+lint:
+	cargo clippy 

etcd_client.pyi

@@ -0,0 +1,53 @@
+"""
+Type hints for Native Rust Extension
+"""
+
+from typing import Any, Final, Optional
+
+class Client:
+    """ """
+
+    def __init__(self, endpoints: list[str]) -> None:
+        """ """
+    def connect(self) -> "Client":
+        """ """
+    def __aenter__(self) -> "Communicator":
+        """ """
+
+class Watch:
+    """ """
+
+class Communicator:
+    def get(self, key: str) -> str:
+        """ """
+    def get_prefix(self, key: str) -> dict:
+        """ """
+    def keys_prefix(self, key: str) -> list[str]:
+        """ """
+    def delete(self, key: str) -> None:
+        """ """
+    def delete_prefix(self, key: str) -> None:
+        """ """
+    def put(self, key: str, value: str) -> None:
+        """ """
+    def watch(self, key: str) -> "Watch":
+        """ """
+    def watch_prefix(self, key: str) -> "Watch":
+        """ """
+
+class Event:
+    """ """
+
+    def __init__(
+        key: str, value: str, event: "EventType", prev_value: Optional[str]
+    ) -> None: ...
+
+class EventType:
+    """ """
+
+    PUT: Final[Any]
+    """
+    """
+    DELETE: Final[Any]
+    """
+    """

manual-test.py

@@ -0,0 +1,3 @@
+from etcd_client import Client
+
+etcd = Client(["http://localhost:2379"])

pytest.ini

@@ -0,0 +1,2 @@
+[pytest]
+python_files = */tests/*.py

requirements.txt

@@ -0,0 +1,2 @@
+maturin==1.3.2
+pytest==7.3.1

src/lib.rs

@@ -1,12 +1,12 @@
 use std::collections::HashMap;
 use std::sync::Arc;
 
-use etcd_client::Client as RustClient;
-use etcd_client::Error as RustError;
-use etcd_client::Event as RustEvent;
-use etcd_client::EventType as RustEventType;
-use etcd_client::WatchStream as RustStream;
-use etcd_client::{DeleteOptions, GetOptions, WatchOptions};
+use ::etcd_client::Client as RustClient;
+use ::etcd_client::Error as RustError;
+use ::etcd_client::Event as RustEvent;
+use ::etcd_client::EventType as RustEventType;
+use ::etcd_client::WatchStream as RustStream;
+use ::etcd_client::{DeleteOptions, GetOptions, WatchOptions};
 use pyo3::create_exception;
 use pyo3::exceptions::{PyException, PyStopAsyncIteration};
 use pyo3::prelude::*;
@@ -28,7 +28,7 @@ impl From<Error> for PyErr {
 
 #[pyclass]
 #[derive(PartialEq, Eq, Clone)]
-struct EventType(RustEventType);
+pub struct EventType(RustEventType);
 
 #[pymethods]
 impl EventType {
@@ -41,7 +41,7 @@ impl EventType {
 
 #[pyclass]
 #[derive(PartialEq, Eq, Clone)]
-struct Event {
+pub struct Event {
     key: String,
     value: String,
     prev_value: Option<String>,
@@ -78,7 +78,7 @@ impl From<RustEvent> for Event {
 
 #[pyclass]
 #[derive(Clone)]
-struct Client {
+pub struct Client {
     endpoints: Vec<String>,
 }
 
@@ -114,7 +114,7 @@ impl Client {
 }
 
 #[pyclass]
-struct Communicator(Arc<Mutex<RustClient>>);
+pub struct Communicator(Arc<Mutex<RustClient>>);
 
 #[pymethods]
 impl Communicator {
@@ -266,7 +266,7 @@ impl Stream {
 
 #[pyclass]
 #[derive(Clone)]
-struct Watch {
+pub struct Watch {
     client: Arc<Mutex<RustClient>>,
     key: String,
     options: Option<WatchOptions>,
@@ -325,10 +325,11 @@ impl Watch {
 }
 
 #[pymodule]
-#[pyo3(name = "etcd_client")]
-fn init(_py: Python<'_>, module: &PyModule) -> PyResult<()> {
+fn etcd_client(_py: Python<'_>, module: &PyModule) -> PyResult<()> {
     module.add_class::<Client>()?;
     module.add_class::<Event>()?;
     module.add_class::<EventType>()?;
+    module.add_class::<Communicator>()?;
+    module.add_class::<Watch>()?;
     Ok(())
 }