| 
5 | 5 | from typing import Any, Collection, Dict, List  | 
6 | 6 | 
 
  | 
7 | 7 | import simple_salesforce  | 
 | 8 | +from simple_salesforce import Salesforce  # type: ignore[attr-defined]  | 
8 | 9 | 
 
  | 
9 | 10 | from zulip_bots.bots.salesforce.utils import commands, default_query, link_query, object_types  | 
10 | 11 | from zulip_bots.lib import BotHandler  | 
@@ -73,9 +74,7 @@ def format_result(  | 
73 | 74 |     return output  | 
74 | 75 | 
 
  | 
75 | 76 | 
 
  | 
76 |  | -def query_salesforce(  | 
77 |  | -    arg: str, salesforce: simple_salesforce.Salesforce, command: Dict[str, Any]  | 
78 |  | -) -> str:  | 
 | 77 | +def query_salesforce(arg: str, salesforce: Salesforce, command: Dict[str, Any]) -> str:  | 
79 | 78 |     arg = arg.strip()  | 
80 | 79 |     qarg = arg.split(" -", 1)[0]  | 
81 | 80 |     split_args: List[str] = []  | 
@@ -164,7 +163,7 @@ def get_salesforce_response(self, content: str) -> str:  | 
164 | 163 |     def initialize(self, bot_handler: BotHandler) -> None:  | 
165 | 164 |         self.config_info = bot_handler.get_config_info("salesforce")  | 
166 | 165 |         try:  | 
167 |  | -            self.sf = simple_salesforce.Salesforce(  | 
 | 166 | +            self.sf = Salesforce(  | 
168 | 167 |                 username=self.config_info["username"],  | 
169 | 168 |                 password=self.config_info["password"],  | 
170 | 169 |                 security_token=self.config_info["security_token"],  | 
 | 
0 commit comments