Refactor DNA translation functions to use PyAny

Author: OrhunKok

src/python_api.rs

@@ -29,9 +29,10 @@ fn _check_table(table: u8) -> PyResult<()> {
 ///
 /// * `translate(b"CCNTACACK CATNCNAAT")` returns `b"PYTHXN"`
 #[pyfunction]
-fn _translate(py: Python<'_>, table: u8, dna: &PyBytes) -> PyResult<Py<PyAny>> {
+fn _translate(py: Python, table: u8, dna: &PyAny) -> PyResult<Py<PyAny>> {
     let table = TranslationTable::try_from(table)?;
-    let bytes = table.translate_dna_bytes::<NucleotideAmbiguous>(dna.as_bytes())?;
+    let dna_bytes = dna.downcast::<PyBytes>()?;
+    let bytes = table.translate_dna_bytes::<NucleotideAmbiguous>(dna_bytes.as_bytes())?;
     Ok(PyBytes::new(py, &bytes).into_py(py))
 }
 
@@ -42,9 +43,10 @@ fn _translate(py: Python<'_>, table: u8, dna: &PyBytes) -> PyResult<Py<PyAny>> {
 /// * `translate_strict(b"AAACCCTTTGGG")` returns `b"KPFG"`
 /// * `translate_strict(b"AAACCCTTTGGN")` is an error.
 #[pyfunction]
-fn _translate_strict(py: Python<'_>, table: u8, dna: &PyBytes) -> PyResult<Py<PyAny>> {
+fn _translate_strict(py: Python, table: u8, dna: &PyAny) -> PyResult<Py<PyAny>> {
     let table = TranslationTable::try_from(table)?;
-    let bytes = table.translate_dna_bytes::<Nucleotide>(dna.as_bytes())?;
+    let dna_bytes = dna.downcast::<PyBytes>()?;
+    let bytes = table.translate_dna_bytes::<Nucleotide>(dna_bytes.as_bytes())?;
     Ok(PyBytes::new(py, &bytes).into_py(py))
 }
 
@@ -54,8 +56,9 @@ fn _translate_strict(py: Python<'_>, table: u8, dna: &PyBytes) -> PyResult<Py<Py
 ///
 /// * `reverse_complement(b"AAAAABCCC")` returns `b"GGGVTTTTT"`
 #[pyfunction]
-fn _reverse_complement(py: Python<'_>, dna: &PyBytes) -> PyResult<Py<PyAny>> {
-    let bytes = reverse_complement_bytes::<NucleotideAmbiguous>(dna.as_bytes())?;
+fn _reverse_complement(py: Python, dna: &PyAny) -> PyResult<Py<PyAny>> {
+    let dna_bytes = dna.downcast::<PyBytes>()?;
+    let bytes = reverse_complement_bytes::<NucleotideAmbiguous>(dna_bytes.as_bytes())?;
     Ok(PyBytes::new(py, &bytes).into_py(py))
 }
 
@@ -66,13 +69,14 @@ fn _reverse_complement(py: Python<'_>, dna: &PyBytes) -> PyResult<Py<PyAny>> {
 /// * `reverse_complement_strict(b"AAAAAACCC")` returns `b"GGGTTTTTT"`
 /// * `reverse_complement_strict(b"AAAAAACCN")` is an error.
 #[pyfunction]
-fn _reverse_complement_strict(py: Python<'_>, dna: &PyBytes) -> PyResult<Py<PyAny>> {
-    let bytes = reverse_complement_bytes::<Nucleotide>(dna.as_bytes())?;
+fn _reverse_complement_strict(py: Python, dna: &PyAny) -> PyResult<Py<PyAny>> {
+    let dna_bytes = dna.downcast::<PyBytes>()?;
+    let bytes = reverse_complement_bytes::<Nucleotide>(dna_bytes.as_bytes())?;
     Ok(PyBytes::new(py, &bytes).into_py(py))
 }
 
 #[pymodule]
-fn quickdna(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
+fn quickdna(py: Python, m: &PyModule) -> PyResult<()> {
     m.add_function(wrap_pyfunction!(_check_table, m)?)?;
     m.add_function(wrap_pyfunction!(_translate, m)?)?;
     m.add_function(wrap_pyfunction!(_translate_strict, m)?)?;