Skip to content

Commit 2a8c256

Browse files
committed
fix(async): Fix code generator for edge cases at "smartquery" interface
1 parent d9f3f64 commit 2a8c256

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

CHANGELOG.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
# CHANGELOG
22

33
## unreleased
4+
- Async: Fixed code generator for edge cases at "smartquery" interface. Thanks, @JIAQIA.
45

56
## 4.3.2 (2025-03-04)
67
- Alerting: Allowed the datasource to be specified with managed alerting. Thanks, @dmyerscough.

script/generate_async.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,13 +101,14 @@ def process(source: Path, target: Path):
101101

102102
module_dump = re.sub(r"( {4}def )(?!_)", r" async def ", module_dump)
103103

104-
module_dump = module_dump.replace("self.client.", "await self.client.")
104+
module_dump = re.sub(r"self\.client\.(.+)\(", r"await self.client.\1(", module_dump)
105105

106106
for relative_import in [".base", "..client", "..knowledge", "..model"]:
107107
module_dump = module_dump.replace(f"from {relative_import}", f"from .{relative_import}")
108108

109109
module_dump = module_dump.replace("self.api.version", "await self.api.version")
110-
module_dump = module_dump.replace("= self.", "= await self.")
110+
module_dump = re.sub(r"= self\.(.+)\(", r"= await self.\1(", module_dump)
111+
module_dump = re.sub(r"send_request\(", r"await send_request(", module_dump)
111112

112113
module_processed.append(module_path)
113114
target_path = Path(str(module_path).replace(str(source), str(target)))

0 commit comments

Comments
 (0)