WIP

.gitignore

@@ -5,3 +5,4 @@
 .*.swo
 /dist/
 /target/
+__pycache__/

Cargo.toml

@@ -10,8 +10,11 @@ name = "etcd_client"
 crate-type = ["cdylib"]
 
 [dependencies]
+async-recursion = "1.0.5"
 etcd-client = "0.10"
-pyo3 = { version = "0.18", features = ["extension-module"] }
+pyo3 = { version = "0.18", features = ["extension-module", "multiple-pymethods"] }
 pyo3-asyncio = { version = "0.18", features = ["tokio-runtime"] }
+serde_json = "1.0.111"
 tokio = { version = "1", features = ["sync"] }
 tokio-stream = "0.1"
+url = "2.5.0"

Makefile

@@ -5,8 +5,8 @@ install:
 	maturin build
 	pip install .
 
-test:
-	python -m pytest
+etcd-clear:
+	etcdctl del "" --from-key=true
 
 fmt:
 	cargo fmt

etcd_client.pyi

@@ -30,6 +30,10 @@ class Communicator:
         """ """
     def put(self, key: str, value: str) -> None:
         """ """
+    def put_prefix(self, key: str, value: dict) -> None:
+        """ """
+    def replace(self, key: str, initial_value: str, new_value: str) -> bool:
+        """ """
     def watch(self, key: str) -> "Watch":
         """ """
     def watch_prefix(self, key: str) -> "Watch":

src/client.rs

@@ -0,0 +1,42 @@
+use etcd_client::Client as RustClient;
+use pyo3::prelude::*;
+use pyo3::types::PyTuple;
+use pyo3_asyncio::tokio::future_into_py;
+use std::sync::Arc;
+use tokio::sync::Mutex;
+
+use crate::communicator::Communicator;
+use crate::error::Error;
+
+#[pyclass]
+#[derive(Clone)]
+pub struct Client {
+    endpoints: Vec<String>,
+}
+
+#[pymethods]
+impl Client {
+    #[new]
+    fn new(endpoints: Vec<String>) -> Self {
+        Self { endpoints }
+    }
+
+    fn connect(&self) -> Self {
+        self.clone()
+    }
+
+    fn __aenter__<'a>(&'a self, py: Python<'a>) -> PyResult<&'a PyAny> {
+        let endpoints = self.endpoints.clone();
+        future_into_py(py, async move {
+            let result = RustClient::connect(endpoints, None).await;
+            result
+                .map(|client| Communicator(Arc::new(Mutex::new(client))))
+                .map_err(|e| Error(e).into())
+        })
+    }
+
+    #[pyo3(signature = (*_args))]
+    fn __aexit__<'a>(&'a self, py: Python<'a>, _args: &PyTuple) -> PyResult<&'a PyAny> {
+        future_into_py(py, async move { Ok(()) })
+    }
+}