add api conversion method to FerveoVariant

nucypher · Jul 11, 2023 · fbb97be · fbb97be
1 parent 7cbe65d
commit fbb97be
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/ferveo/src/bindings_python.rs b/ferveo/src/bindings_python.rs
@@ -268,7 +268,7 @@ pub fn decrypt_with_shared_secret(
 }
 
 #[pyclass(module = "ferveo")]
-struct FerveoVariant {}
+pub struct FerveoVariant {}
 
 #[pymethods]
 impl FerveoVariant {
@@ -283,6 +283,13 @@ impl FerveoVariant {
     }
 }
 
+impl FerveoVariant {
+    pub fn inner_form_string(variant: &str) -> PyResult<api::FerveoVariant> {
+        api::FerveoVariant::from_string(variant)
+            .map_err(|err| FerveoPythonError::FerveoError(err).into())
+    }
+}
+
 #[pyclass(module = "ferveo")]
 #[derive(derive_more::AsRef)]
 pub struct SharedSecret(api::SharedSecret);