@darabonba/python-generator 2.0.9 → 2.0.10

package/lib/builtin.js

@@ -148,13 +148,22 @@ class URL extends Builtin {
 
 class Stream extends Builtin {
   constructor(generator) {
-    const methods = ['readAsBytes', 'readAsJSON', 'readAsString', 'readAsSSE'];
+    const methods = ['readAsBytes', 'readAsJSON', 'readAsString'];
     super(generator, `${CORE}Stream`, methods);
   }
 
+  readAsSSE(ast, level, async) {
+    this.generator.imports.push(_getImport(this.module));
+    this.generator.emit(`${this.module}.read_as_sse`);
+    if(async) {
+      this.generator.emit('_async');
+    }
+    this.generator.visitArgs(ast.args, level);
+  }
+
   write(ast, level, async) {
     if(async) {
-      this.emit('awiat ');
+      this.emit('await ');
     }
     this.generator.emit(`${_snakeCase(ast.left.id.lexeme)}.write${async ? '_async' : ''}(`);
     this.generator.visitExpr(ast.args[0], level);
@@ -431,7 +440,7 @@ class Func {
   sleep(ast, level, async) {
     const clientName = this.getClientName();
     if(async) {
-      this.generator.emit('awiat ');
+      this.generator.emit('await ');
     }
     this.generator.emit(`${clientName}.sleep`);
     if(async) {