Add support for custom urls

Cargo.toml

@@ -1,7 +1,7 @@
 [package]
 name = "piston_rspy"
 description = "Python bindings for piston_rs."
-version = "0.3.1"
+version = "0.3.2"
 edition = "2021"
 authors = ["Jonxslays"]
 readme = "README.md"
@@ -20,4 +20,4 @@ crate-type = ["cdylib"]
 pyo3 = { version = "0.15", features = ["extension-module"] }
 pyo3-asyncio = { version = "0.15", features = ["tokio-runtime"] }
 tokio = { version = "1" }
-piston_rs = "^0.4.0"
+piston_rs = "^0.4.1"

piston_rspy/piston_rspy.pyi

@@ -434,6 +434,32 @@ class Client:
             key: `str`
                 The api key to use.
 
+        Returns:
+            `Client`: The new client.
+        """
+        ...
+    @staticmethod
+    def with_url(url: str) -> Client:
+        """Creates a new client with a custom url.
+
+        Args:
+            url: `str`
+                The url to use as the underlying Piston backend.
+
+        Returns:
+            `Client`: The new client.
+        """
+        ...
+    @staticmethod
+    def with_url_and_key(url: str, key: str) -> Client:
+        """Creates a new client with a custom url, and an api key.
+
+        Args:
+            url: `str`
+                The url to use as the underlying Piston backend.
+            key: `str`
+                The api key to use.
+
         Returns:
             `Client`: The new client.
         """

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "piston_rspy"
-version = "0.3.1"
+version = "0.3.2"
 description = "Python bindings for piston_rs."
 readme = "README.md"
 author = "Jonxslays"

src/client.rs

@@ -52,6 +52,55 @@ impl Client {
         Self { inner, headers }
     }
 
+    /// Creates a new Client with a custom url and an api key.
+    ///
+    /// ### Args:
+    ///
+    /// - url `str`:
+    /// The url to use as the underlying piston backend.
+    ///
+    /// - key `str`:
+    /// The api key to use.
+    ///
+    /// ### Returns:
+    ///
+    /// - `Client`: The new client.
+    #[staticmethod]
+    #[pyo3(text_signature = "(key: str, /) -> Client")]
+    fn with_url_and_key(url: String, key: String) -> Self {
+        let inner = Client_::with_url_and_key(&url, &key);
+        let headers = inner
+            .get_headers()
+            .iter()
+            .map(|(key, value)| (key.to_string(), value.to_str().unwrap().to_string()))
+            .collect();
+
+        Self { inner, headers }
+    }
+
+    /// Creates a new Client with a custom url.
+    ///
+    /// ### Args:
+    ///
+    /// - url `str`:
+    /// The url to use as the underlying piston backend.
+    ///
+    /// ### Returns:
+    ///
+    /// - `Client`: The new client.
+    #[staticmethod]
+    #[pyo3(text_signature = "(key: str, /) -> Client")]
+    fn with_url(url: String) -> Self {
+        let inner = Client_::with_url(&url);
+        let headers = inner
+            .get_headers()
+            .iter()
+            .map(|(key, value)| (key.to_string(), value.to_str().unwrap().to_string()))
+            .collect();
+
+        Self { inner, headers }
+    }
+
     /// Creates a new client, with an api key.
     ///
     /// ### Args: