enhance: [2.5] Add force_drop parameter to drop_role method for role deletion

Signed-off-by: SimFG <[email protected]>

Author: SimFG

pymilvus/client/grpc_handler.py

@@ -1903,8 +1903,10 @@ def create_role(self, role_name: str, timeout: Optional[float] = None, **kwargs)
         check_status(resp)
 
     @retry_on_rpc_failure()
-    def drop_role(self, role_name: str, timeout: Optional[float] = None, **kwargs):
-        req = Prepare.drop_role_request(role_name)
+    def drop_role(
+        self, role_name: str, force_drop: bool = False, timeout: Optional[float] = None, **kwargs
+    ):
+        req = Prepare.drop_role_request(role_name, force_drop=force_drop)
         resp = self._stub.DropRole(req, wait_for_ready=True, timeout=timeout)
         check_status(resp)
 

pymilvus/client/prepare.py

@@ -1478,9 +1478,9 @@ def create_role_request(cls, role_name: str):
         return milvus_types.CreateRoleRequest(entity=milvus_types.RoleEntity(name=role_name))
 
     @classmethod
-    def drop_role_request(cls, role_name: str):
+    def drop_role_request(cls, role_name: str, force_drop: bool = False):
         check_pass_param(role_name=role_name)
-        return milvus_types.DropRoleRequest(role_name=role_name)
+        return milvus_types.DropRoleRequest(role_name=role_name, force_drop=force_drop)
 
     @classmethod
     def operate_user_role_request(cls, username: str, role_name: str, operate_user_role_type: Any):

pymilvus/milvus_client/milvus_client.py

@@ -1212,9 +1212,11 @@ def create_role(self, role_name: str, timeout: Optional[float] = None, **kwargs)
         conn = self._get_connection()
         conn.create_role(role_name, timeout=timeout, **kwargs)
 
-    def drop_role(self, role_name: str, timeout: Optional[float] = None, **kwargs):
+    def drop_role(
+        self, role_name: str, force_drop: bool = False, timeout: Optional[float] = None, **kwargs
+    ):
         conn = self._get_connection()
-        conn.drop_role(role_name, timeout=timeout, **kwargs)
+        conn.drop_role(role_name, force_drop=force_drop, timeout=timeout, **kwargs)
 
     def describe_role(self, role_name: str, timeout: Optional[float] = None, **kwargs) -> Dict:
         conn = self._get_connection()