From 7d4c19806722a9762433a071221ef4514bf15a51 Mon Sep 17 00:00:00 2001
From: Palash Tyagi <23239946+Magnus167@users.noreply.github.com>
Date: Wed, 16 Apr 2025 01:36:56 +0100
Subject: [PATCH] feat: update create_blacklist_from_qdf function signature to
 include group_by_cid and blacklist_name parameters

---
 src/_py/utils.rs | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/src/_py/utils.rs b/src/_py/utils.rs
index b944951..5584578 100644
--- a/src/_py/utils.rs
+++ b/src/_py/utils.rs
@@ -1,7 +1,7 @@
-use pyo3::types::IntoPyDict;
 use pyo3::{prelude::*, types::PyDict};
 use pyo3_polars::{PyDataFrame, PySeries};
 
+
 /// Python wrapper for [`crate::utils::qdf`] module.
 #[allow(deprecated)]
 #[pymodule]
@@ -37,10 +37,20 @@ pub fn get_bdates_series_default_opt(
 }
 
 #[allow(deprecated)]
-#[pyfunction(signature = (df, metric = None))]
-pub fn create_blacklist_from_qdf(df: PyDataFrame, metric: Option<String>) -> PyResult<PyObject> {
-    let result = crate::utils::qdf::blacklist::create_blacklist_from_qdf(&df.into(), metric)
-        .map_err(|e| PyErr::new::<pyo3::exceptions::PyValueError, _>(format!("{}", e)))?;
+#[pyfunction(signature = (df, group_by_cid=None, blacklist_name=None, metric=None))]
+pub fn create_blacklist_from_qdf(
+    df: PyDataFrame,
+    group_by_cid: Option<bool>,
+    blacklist_name: Option<String>,
+    metric: Option<String>,
+) -> PyResult<PyObject> {
+    let result = crate::utils::qdf::blacklist::create_blacklist_from_qdf(
+        &df.into(),
+        group_by_cid,
+        blacklist_name,
+        metric,
+    )
+    .map_err(|e| PyErr::new::<pyo3::exceptions::PyValueError, _>(format!("{}", e)))?;
     Python::with_gil(|py| {
         let dict = PyDict::new(py);
         // for (key, (start_date, end_date)) in result {