|
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