feat(auth-registry): add get_auth_registry() for Python (#1032)

Author: georgeh0

python/cocoindex/auth_registry.py

@@ -4,10 +4,9 @@
 
 from dataclasses import dataclass
 from typing import Generic, TypeVar
-import threading
 
 from . import _engine  # type: ignore
-from .convert import dump_engine_object
+from .convert import dump_engine_object, load_engine_object
 
 T = TypeVar("T")
 
@@ -38,3 +37,8 @@ def add_auth_entry(key: str, value: T) -> AuthEntryReference[T]:
 def ref_auth_entry(key: str) -> AuthEntryReference[T]:
     """Reference an auth entry by its key."""
     return AuthEntryReference(key)
+
+
+def get_auth_entry(cls: type[T], ref: TransientAuthEntryReference[T]) -> T:
+    """Get an auth entry by its key."""
+    return load_engine_object(cls, _engine.get_auth_entry(ref.key))

src/base/spec.rs

@@ -516,6 +516,14 @@ pub struct TransientFlowSpec {
     pub output_value: ValueMapping,
 }
 
+impl<T> AuthEntryReference<T> {
+    pub fn new(key: String) -> Self {
+        Self {
+            key,
+            _phantom: std::marker::PhantomData,
+        }
+    }
+}
 pub struct AuthEntryReference<T> {
     pub key: String,
     _phantom: std::marker::PhantomData<T>,
@@ -535,10 +543,7 @@ impl<T> fmt::Display for AuthEntryReference<T> {
 
 impl<T> Clone for AuthEntryReference<T> {
     fn clone(&self) -> Self {
-        Self {
-            key: self.key.clone(),
-            _phantom: std::marker::PhantomData,
-        }
+        Self::new(self.key.clone())
     }
 }
 
@@ -562,10 +567,7 @@ impl<'de, T> Deserialize<'de> for AuthEntryReference<T> {
         D: serde::Deserializer<'de>,
     {
         let untyped_ref = UntypedAuthEntryReference::<String>::deserialize(deserializer)?;
-        Ok(AuthEntryReference {
-            key: untyped_ref.key,
-            _phantom: std::marker::PhantomData,
-        })
+        Ok(AuthEntryReference::new(untyped_ref.key))
     }
 }
 

src/py/mod.rs

@@ -2,7 +2,7 @@ use crate::execution::evaluator::evaluate_transient_flow;
 use crate::prelude::*;
 
 use crate::base::schema::{FieldSchema, ValueType};
-use crate::base::spec::{NamedSpec, OutputMode, ReactiveOpSpec, SpecFormatter};
+use crate::base::spec::{AuthEntryReference, NamedSpec, OutputMode, ReactiveOpSpec, SpecFormatter};
 use crate::lib_context::{
     QueryHandlerContext, clear_lib_context, get_auth_registry, init_lib_context,
 };
@@ -630,6 +630,13 @@ fn add_transient_auth_entry(value: Pythonized<serde_json::Value>) -> PyResult<St
         .into_py_result()
 }
 
+#[pyfunction]
+fn get_auth_entry(key: String) -> PyResult<Pythonized<serde_json::Value>> {
+    let auth_ref = AuthEntryReference::new(key);
+    let json_value: serde_json::Value = get_auth_registry().get(&auth_ref).into_py_result()?;
+    Ok(Pythonized(json_value))
+}
+
 #[pyfunction]
 fn get_app_namespace(py: Python<'_>) -> PyResult<String> {
     let app_namespace = py
@@ -671,6 +678,7 @@ fn cocoindex_engine(m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_function(wrap_pyfunction!(remove_flow_context, m)?)?;
     m.add_function(wrap_pyfunction!(add_auth_entry, m)?)?;
     m.add_function(wrap_pyfunction!(add_transient_auth_entry, m)?)?;
+    m.add_function(wrap_pyfunction!(get_auth_entry, m)?)?;
     m.add_function(wrap_pyfunction!(get_app_namespace, m)?)?;
 
     m.add_class::<builder::flow_builder::FlowBuilder>()?;