python2ts 1.3.2 → 1.4.0

package/dist/{chunk-RNRRQS6K.js → chunk-2VKIPKNY.js}

@@ -262,6 +262,71 @@ function toJsName(pythonName) {
 }
 
 // src/transformer/index.ts
+var JS_RESERVED_KEYWORDS = /* @__PURE__ */ new Set([
+  // ECMAScript reserved words
+  "break",
+  "case",
+  "catch",
+  "continue",
+  "debugger",
+  "default",
+  "delete",
+  "do",
+  "else",
+  "finally",
+  "for",
+  "function",
+  "if",
+  "in",
+  "instanceof",
+  "new",
+  "return",
+  "switch",
+  "this",
+  "throw",
+  "try",
+  "typeof",
+  "var",
+  "void",
+  "while",
+  "with",
+  // ECMAScript 6+ reserved words
+  "class",
+  "const",
+  "enum",
+  "export",
+  "extends",
+  "import",
+  // Note: 'super' is intentionally NOT in this list as it's valid in JS class contexts
+  // Strict mode reserved words
+  "implements",
+  "interface",
+  "let",
+  "package",
+  "private",
+  "protected",
+  "public",
+  "static",
+  "yield",
+  // TypeScript reserved words
+  "abstract",
+  "as",
+  "async",
+  "await",
+  "declare",
+  "from",
+  "get",
+  "is",
+  "module",
+  "namespace",
+  "of",
+  "require",
+  "set",
+  "type"
+]);
+function escapeReservedKeyword(name) {
+  return JS_RESERVED_KEYWORDS.has(name) ? `_${name}` : name;
+}
 function createContext(source) {
   return {
     source,
@@ -521,13 +586,23 @@ function isDocstringNode(node, ctx) {
   const firstChild = children[0];
   if (firstChild?.name !== "String") return false;
   const text = getNodeText(firstChild, ctx.source);
-  return text.startsWith('"""') || text.startsWith("'''");
+  return isTripleQuotedString(text);
+}
+function isTripleQuotedString(text) {
+  let stripped = text;
+  if (/^[rRuU]/.test(text)) {
+    stripped = text.slice(1);
+  }
+  return stripped.startsWith('"""') || stripped.startsWith("'''");
 }
 function extractDocstringContent(node, ctx) {
   const children = getChildren(node);
   const stringNode = children[0];
   if (!stringNode) return "";
-  const text = getNodeText(stringNode, ctx.source);
+  let text = getNodeText(stringNode, ctx.source);
+  if (/^[rRuU]/.test(text)) {
+    text = text.slice(1);
+  }
   let content = text;
   if (content.startsWith('"""')) {
     content = content.slice(3, -3);
@@ -716,6 +791,8 @@ function transformNode(node, ctx) {
       return transformExpressionStatement(node, ctx);
     case "AssignStatement":
       return transformAssignStatement(node, ctx);
+    case "UpdateStatement":
+      return transformUpdateStatement(node, ctx);
     case "BinaryExpression":
       return transformBinaryExpression(node, ctx);
     case "UnaryExpression":
@@ -732,12 +809,14 @@ function transformNode(node, ctx) {
       return transformString(node, ctx);
     case "FormatString":
       return transformFormatString(node, ctx);
+    case "ContinuedString":
+      return transformContinuedString(node, ctx);
     case "Boolean":
       return transformBoolean(node, ctx);
     case "None":
       return "null";
     case "VariableName":
-      return getNodeText(node, ctx.source);
+      return escapeReservedKeyword(getNodeText(node, ctx.source));
     case "CallExpression":
       return transformCallExpression(node, ctx);
     case "MemberExpression":
@@ -802,6 +881,9 @@ function transformNode(node, ctx) {
       return transformAssertStatement(node, ctx);
     case "YieldStatement":
       return transformYieldStatement(node, ctx);
+    case "Ellipsis":
+      ctx.usesRuntime.add("Ellipsis");
+      return "Ellipsis";
     /* v8 ignore next 2 -- fallback for unknown AST nodes @preserve */
     default:
       return getNodeText(node, ctx.source);
@@ -947,6 +1029,23 @@ function extractVariableNames(nodes, source) {
   }
   return names;
 }
+function transformUpdateStatement(node, ctx) {
+  const children = getChildren(node);
+  const target = children.find(
+    (c) => c.name === "VariableName" || c.name === "MemberExpression" || c.name === "Subscript"
+  );
+  const op = children.find((c) => c.name === "UpdateOp");
+  const value = children.find(
+    (c) => c !== target && c.name !== "UpdateOp" && c.name !== "(" && c.name !== ")" && c.name !== "," && c.name !== ":"
+  );
+  if (!target || !op || !value) {
+    return getNodeText(node, ctx.source);
+  }
+  const targetCode = transformNode(target, ctx);
+  const opText = getNodeText(op, ctx.source);
+  const valueCode = transformNode(value, ctx);
+  return `${targetCode} ${opText} ${valueCode}`;
+}
 function isSliceExpression(node) {
   const children = getChildren(node);
   return children.some((c) => c.name === ":");
@@ -1239,6 +1338,51 @@ function transformFormatString(node, ctx) {
   result += "`";
   return result;
 }
+function transformContinuedString(node, ctx) {
+  const children = getChildren(node);
+  const hasFormatString = children.some((c) => c.name === "FormatString");
+  if (hasFormatString) {
+    const parts = children.filter((c) => c.name === "String" || c.name === "FormatString").map((c) => {
+      if (c.name === "FormatString") {
+        return transformFormatString(c, ctx);
+      } else {
+        const text = getNodeText(c, ctx.source);
+        let content;
+        if (/^[rR]['"]/.test(text)) {
+          content = text.slice(2, -1);
+        } else if (/^[rR]"""/.test(text) || /^[rR]'''/.test(text)) {
+          content = text.slice(4, -3);
+        } else if (text.startsWith('"""') || text.startsWith("'''")) {
+          content = text.slice(3, -3);
+        } else {
+          content = text.slice(1, -1);
+        }
+        return "`" + content.replace(/`/g, "\\`") + "`";
+      }
+    });
+    return parts.join(" + ");
+  } else {
+    const parts = children.filter((c) => c.name === "String").map((c) => {
+      const text = getNodeText(c, ctx.source);
+      let content;
+      if (/^[rR]['"]/.test(text)) {
+        content = text.slice(2, -1);
+      } else if (/^[rR]"""/.test(text) || /^[rR]'''/.test(text)) {
+        content = text.slice(4, -3);
+      } else if (text.startsWith('"""') || text.startsWith("'''")) {
+        content = text.slice(3, -3);
+      } else {
+        content = text.slice(1, -1);
+      }
+      return content;
+    });
+    const joined = parts.join("");
+    if (joined.includes('"') && !joined.includes("'")) {
+      return "'" + joined + "'";
+    }
+    return '"' + joined + '"';
+  }
+}
 function transformBoolean(node, ctx) {
   const text = getNodeText(node, ctx.source);
   return text === "True" ? "true" : "false";
@@ -1839,7 +1983,11 @@ function transformMethodCall(callee, args, ctx) {
     case "rjust":
       return `${objCode}.padStart(${args})`;
     // String split/join - join is special: "sep".join(arr) -> arr.join("sep")
+    // If the argument is a generator expression, convert to array with spread
     case "join":
+      if (args.includes("function*")) {
+        return `[...${args}].join(${objCode})`;
+      }
       return `(${args}).join(${objCode})`;
     case "split":
       return args ? `${objCode}.split(${args})` : `${objCode}.split(/\\s+/)`;
@@ -2040,15 +2188,23 @@ function transformMemberExpression(node, ctx) {
     if (text.includes(":")) {
       return transformSliceFromMember(obj, children, ctx);
     }
-    const indexElements = children.filter((c) => c.name !== "[" && c.name !== "]" && c !== obj);
-    const index = indexElements[0];
-    if (!index) return `${objCode}[]`;
-    const indexCode = transformNode(index, ctx);
-    if (isNegativeIndexLiteral(index, ctx)) {
-      ctx.usesRuntime.add("at");
-      return `at(${objCode}, ${indexCode})`;
+    const indexElements = children.filter(
+      (c) => c.name !== "[" && c.name !== "]" && c.name !== "," && c !== obj
+    );
+    if (indexElements.length === 0) return `${objCode}[]`;
+    if (indexElements.length === 1) {
+      const index = indexElements[0];
+      if (!index) return `${objCode}[]`;
+      const indexCode = transformNode(index, ctx);
+      if (isNegativeIndexLiteral(index, ctx)) {
+        ctx.usesRuntime.add("at");
+        return `at(${objCode}, ${indexCode})`;
+      }
+      return `${objCode}[${indexCode}]`;
     }
-    return `${objCode}[${indexCode}]`;
+    ctx.usesRuntime.add("tuple");
+    const indices = indexElements.map((el) => transformNode(el, ctx));
+    return `${objCode}[tuple(${indices.join(", ")})]`;
   } else {
     const prop = children[children.length - 1];
     if (!prop) return getNodeText(node, ctx.source);
@@ -3050,7 +3206,12 @@ function transformWithStatement(node, ctx) {
       body = child;
       break;
     }
-    if (child.name !== "as" && child.name !== ":" && child.name !== "VariableName") {
+    if (child.name === "as" || child.name === ":") {
+      i++;
+      continue;
+    }
+    const isContextManagerExpr = child.name !== "VariableName" || children[i + 1]?.name === "as" || children[i + 1]?.name === "Body" || children[i + 1]?.name === ",";
+    if (isContextManagerExpr) {
       const expr = child;
       let varName = null;
       const nextChild = children[i + 1];
@@ -3130,7 +3291,7 @@ function extractParamNames(node, source) {
     if (child.name === "*" || getNodeText(child, source) === "*") {
       const nextChild = children[i + 1];
       if (nextChild?.name === "VariableName") {
-        names.push(getNodeText(nextChild, source));
+        names.push(escapeReservedKeyword(getNodeText(nextChild, source)));
         i += 2;
         continue;
       }
@@ -3140,7 +3301,7 @@ function extractParamNames(node, source) {
     if (child.name === "**" || getNodeText(child, source) === "**") {
       const nextChild = children[i + 1];
       if (nextChild?.name === "VariableName") {
-        names.push(getNodeText(nextChild, source));
+        names.push(escapeReservedKeyword(getNodeText(nextChild, source)));
         i += 2;
         continue;
       }
@@ -3148,7 +3309,7 @@ function extractParamNames(node, source) {
       continue;
     }
     if (child.name === "VariableName") {
-      names.push(getNodeText(child, source));
+      names.push(escapeReservedKeyword(getNodeText(child, source)));
       i++;
       continue;
     }
@@ -3156,7 +3317,7 @@ function extractParamNames(node, source) {
       const paramChildren = getChildren(child);
       const name = paramChildren.find((c) => c.name === "VariableName");
       if (name) {
-        names.push(getNodeText(name, source));
+        names.push(escapeReservedKeyword(getNodeText(name, source)));
       }
       i++;
       continue;
@@ -4278,7 +4439,7 @@ function transformParamList(node, ctx) {
   const parseParam = (startIndex) => {
     const child = children[startIndex];
     if (child?.name !== "VariableName") return null;
-    const nameCode = getNodeText(child, ctx.source);
+    const nameCode = escapeReservedKeyword(getNodeText(child, ctx.source));
     let tsType = null;
     let defaultValue = null;
     let offset = 1;
@@ -4320,7 +4481,7 @@ function transformParamList(node, ctx) {
     if (child.name === "*" || getNodeText(child, ctx.source) === "*") {
       const nextChild = children[i + 1];
       if (nextChild?.name === "VariableName") {
-        const name = getNodeText(nextChild, ctx.source);
+        const name = escapeReservedKeyword(getNodeText(nextChild, ctx.source));
         const typeChild = children[i + 2];
         if (typeChild?.name === "TypeDef") {
           const tsType = extractTypeAnnotation(typeChild, ctx);
@@ -4340,7 +4501,7 @@ function transformParamList(node, ctx) {
     if (child.name === "**" || getNodeText(child, ctx.source) === "**") {
       const nextChild = children[i + 1];
       if (nextChild?.name === "VariableName") {
-        kwargsParam = getNodeText(nextChild, ctx.source);
+        kwargsParam = escapeReservedKeyword(getNodeText(nextChild, ctx.source));
         i += 2;
         continue;
       }
@@ -4365,7 +4526,7 @@ function transformParamList(node, ctx) {
       const typeDef = paramChildren.find((c) => c.name === "TypeDef");
       const defaultVal = paramChildren[paramChildren.length - 1];
       if (name) {
-        const nameCode = getNodeText(name, ctx.source);
+        const nameCode = escapeReservedKeyword(getNodeText(name, ctx.source));
         const tsType = extractTypeAnnotation(typeDef, ctx);
         let defaultValue = null;
         if (defaultVal && name !== defaultVal && defaultVal.name !== "TypeDef") {

package/dist/cli/index.js

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 import {
   transpileAsync
-} from "../chunk-RNRRQS6K.js";
+} from "../chunk-2VKIPKNY.js";
 
 // src/cli/index.ts
 import { parseArgs } from "util";

package/dist/index.js

@@ -12,7 +12,7 @@ import {
   transpile,
   transpileAsync,
   walkTree
-} from "./chunk-RNRRQS6K.js";
+} from "./chunk-2VKIPKNY.js";
 export {
   debugTree,
   formatCode,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "python2ts",
-  "version": "1.3.2",
+  "version": "1.4.0",
   "description": "AST-based Python to TypeScript transpiler. Convert Python code to clean, idiomatic TypeScript with full type preservation.",
   "homepage": "https://sebastian-software.github.io/python2ts/",
   "repository": {
@@ -53,7 +53,7 @@
     "eslint": "^9.39.2",
     "prettier": "^3.8.0",
     "typescript-eslint": "^8.53.1",
-    "pythonlib": "2.0.2"
+    "pythonlib": "2.0.3"
   },
   "devDependencies": {
     "tsup": "^8.5.1",