pyi files, remove unused pyclass

Author: circuitsacul

regex_rs.pyi

@@ -0,0 +1,41 @@
+class Regex:
+    def __init__(
+        self,
+        pattern: str,
+        *,
+        case_insensive: bool | None = None,
+        dfa_size_limit: int | None = None,
+        dot_matches_new_line: bool | None = None,
+        ignore_whitespace: bool | None = None,
+        multi_line: bool | None = None,
+        nest_limit: int | None = None,
+        octal: bool | None = None,
+        size_limit: int | None = None,
+        swap_greed: bool | None = None,
+        unicode: bool | None = None,
+    ) -> None: ...
+    def is_match(self, text: str, start: int | None = None) -> bool: ...
+    def find(self, text: str, start: int | None) -> Match | None: ...
+    def find_iter(self, text: str) -> list[Match]: ...
+    def captures(self, text: str) -> Captures | None: ...
+    def split(self, text: str, limit: int | None = None) -> list[str]: ...
+    def replace(self, text: str, rep: str, limit: int | None = None) -> str: ...
+    def __repr__(self) -> str: ...
+    def __str__(self) -> str: ...
+
+
+class Captures:
+    def get(self, i: int) -> Match | None: ...
+    def name(self, name: str) -> Match | None: ...
+    def expand(self, replacement: str, dst: str) -> str: ...
+    def __getitem__(self, i: int) -> Match | None: ...
+    def __len__(self) -> int: ...
+    def __repr__(self) -> str: ...
+
+
+class Match:
+    matched_text: str
+    start: int
+    end: int
+    def __str__(self) -> str: ...
+    def __repr__(self) -> str: ...

src/capture_locations.rs

@@ -1,4 +1,3 @@
-mod capture_locations;
 mod captures;
 mod error;
 mod match_struct;
@@ -10,7 +9,6 @@ use pyo3::prelude::*;
 #[pymodule]
 fn regex_rs(_py: Python, m: &PyModule) -> PyResult<()> {
     m.add_class::<regex::Regex>()?;
-    m.add_class::<capture_locations::CaptureLocations>()?;
     m.add_class::<match_struct::Match>()?;
     m.add_class::<captures::Captures>()?;
     Ok(())

src/lib.rs

@@ -104,24 +104,22 @@ impl Regex {
         Captures::try_new(Arc::new(text), |text| self.0.captures(text).ok_or(())).ok()
     }
 
-    pub fn split(&self, text: &str) -> Vec<String> {
-        self.0.split(text).map(|v| v.to_owned()).collect()
-    }
-
-    pub fn splitn(&self, text: &str, limit: usize) -> Vec<String> {
-        self.0.splitn(text, limit).map(|v| v.to_owned()).collect()
-    }
-
-    pub fn replace(&self, text: &str, rep: &str) -> String {
-        self.0.replace(text, rep).into_owned()
-    }
-
-    pub fn replace_all(&self, text: &str, rep: &str) -> String {
-        self.0.replace_all(text, rep).into_owned()
+    #[pyo3(signature = (text, limit=None))]
+    pub fn split(&self, text: &str, limit: Option<usize>) -> Vec<String> {
+        if let Some(limit) = limit {
+            self.0.splitn(text, limit).map(|v| v.to_owned()).collect()
+        } else {
+            self.0.split(text).map(|v| v.to_owned()).collect()
+        }
     }
 
-    pub fn replacen(&self, text: &str, limit: usize, rep: &str) -> String {
-        self.0.replacen(text, limit, rep).into_owned()
+    #[pyo3(signature = (text, rep, limit=None))]
+    pub fn replace(&self, text: &str, rep: &str, limit: Option<usize>) -> String {
+        if let Some(limit) = limit {
+            self.0.replacen(text, limit, rep).into_owned()
+        } else {
+            self.0.replace_all(text, rep).into_owned()
+        }
     }
 
     // magic