Python: добавлена поддержка == в if

transpile_python/service.go

@@ -183,6 +183,18 @@ func handleBinaryExpr(expr ast.Expr) string {
 	code += handleExpr(be.Y)
 	return code
 }
+func handleBinaryExpr_FromIfStmt(expr ast.Expr) string {
+	be := expr.(*ast.BinaryExpr)
+	code := handleExpr(be.X)
+	switch be.Op {
+	case token.EQL:
+		code += "=="
+	default:
+		code += " " + be.Op.String() + " "
+	}
+	code += handleExpr(be.Y)
+	return code
+}
 
 func handleCallExpr(expr *ast.CallExpr) string {
 	// spew.Dump(expr)
@@ -262,6 +274,34 @@ func handleExpr(expr ast.Expr) string {
 	}
 	return code
 }
+func handleExpr_FromIfStmt(expr ast.Expr) string {
+	code := ""
+	// spew.Dump(expr)
+	switch e := expr.(type) {
+	case *ast.BasicLit:
+		code += handleBasicLit(e)
+	case *ast.CompositeLit:
+		code += handleCompositeLit(e)
+	case *ast.UnaryExpr:
+		code += handleUnaryExpr(e)
+	case *ast.BinaryExpr:
+		code += handleBinaryExpr_FromIfStmt(e)
+	case *ast.CallExpr:
+		code += handleCallExpr(e)
+	case *ast.Ident:
+		code += handleIdent(e)
+	case *ast.ParenExpr:
+		code += handleParenExpr(e)
+	case *ast.SelectorExpr:
+		code += handleSelectorExpr(e)
+	case *ast.IndexExpr:
+		code += handleIndexExpr(e)
+	default:
+		spew.Dump(e)
+		code += "Unknown in handleExpr"
+	}
+	return code
+}
 
 func handleParenExpr(stmt *ast.ParenExpr) string {
 	code := ""
@@ -449,7 +489,7 @@ func handleIdent(expr ast.Expr) string {
 }
 
 func handleIfStmt(stmt *ast.IfStmt) string {
-	cond := handleExpr(stmt.Cond)
+	cond := handleExpr_FromIfStmt(stmt.Cond)
 	increaseInd()
 	body := handleBlockStmt(stmt.Body)
 	decreaseInd()

transpile_python/service_test.go

@@ -503,7 +503,35 @@ main()
 			Compare(source, expected)
 		})
 
-		It("Condition", func() {
+		It("Condition ==", func() {
+			source := `package test
+				import . "skidl"
+
+				func main() {
+					if a == 2 {
+						x = 1
+						y = 5
+					} else {
+						x = 6
+						y = 9
+					}
+				}
+				`
+			expected := `from skidl import *
+
+def main():
+	if a==2:
+		x = 1
+		y = 5
+	else:
+		x = 6
+		y = 9
+main()
+`
+			Compare(source, expected)
+		})
+
+		It("Condition >", func() {
 			source := `package test
 				import . "skidl"