Migrate sct to declarative modules API

refs pyca#11158
alex · Jul 9, 2024 · 88f666f · 88f666f
1 parent 67a5215
commit 88f666f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/src/rust/src/lib.rs b/src/rust/src/lib.rs
@@ -107,6 +107,8 @@ mod _rust {
 
     #[pyo3::pymodule]
     mod x509 {
+        #[pymodule_export]
+        use crate::x509::sct::Sct;
         #[pymodule_export]
         use crate::x509::verify::{
             PolicyBuilder, PyClientVerifier, PyServerVerifier, PyStore, PyVerifiedClient,
@@ -119,7 +121,6 @@ mod _rust {
             crate::x509::common::add_to_module(x509_mod)?;
             crate::x509::crl::add_to_module(x509_mod)?;
             crate::x509::csr::add_to_module(x509_mod)?;
-            crate::x509::sct::add_to_module(x509_mod)?;
 
             Ok(())
         }

diff --git a/src/rust/src/x509/sct.rs b/src/rust/src/x509/sct.rs
@@ -5,7 +5,7 @@
 use std::collections::hash_map::DefaultHasher;
 use std::hash::{Hash, Hasher};
 
-use pyo3::types::{PyAnyMethods, PyDictMethods, PyListMethods, PyModuleMethods};
+use pyo3::types::{PyAnyMethods, PyDictMethods, PyListMethods};
 use pyo3::ToPyObject;
 
 use crate::error::CryptographyError;
@@ -259,12 +259,6 @@ pub(crate) fn parse_scts(
     Ok(py_scts.to_object(py))
 }
 
-pub(crate) fn add_to_module(module: &pyo3::Bound<'_, pyo3::types::PyModule>) -> pyo3::PyResult<()> {
-    module.add_class::<Sct>()?;
-
-    Ok(())
-}
-
 #[cfg(test)]
 mod tests {
     use super::*;