Add str.swapcase() to ast -> asr

yashsain · Sep 2, 2022 · 573a3ab · 573a3ab
1 parent 4ea45c5
commit 573a3ab
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/lpython/semantics/python_ast_to_asr.cpp b/src/lpython/semantics/python_ast_to_asr.cpp
@@ -4195,6 +4195,20 @@ class BodyVisitor : public CommonVisitor<BodyVisitor> {
                                     args.push_back(al, arg);
                                     tmp = make_call_helper(al, fn_div, current_scope, args, "_lpython_str_strip", x.base.base.loc);
                                     return;
+                                } else if (std::string(at->m_attr) == std::string("swapcase")) {
+                                    if(args.size() != 0) {
+                                        throw SemanticError("str.swapcase() takes no arguments",
+                                            x.base.base.loc);
+                                    }
+                                    ASR::symbol_t *fn_div = resolve_intrinsic_function(x.base.base.loc, "_lpython_str_swapcase");
+                                    Vec<ASR::call_arg_t> args;
+                                    args.reserve(al, 1);
+                                    ASR::call_arg_t arg;
+                                    arg.loc = x.base.base.loc;
+                                    arg.m_value = se;
+                                    args.push_back(al, arg);
+                                    tmp = make_call_helper(al, fn_div, current_scope, args, "_lpython_str_swapcase", x.base.base.loc);
+                                    return;
                                 } else if (std::string(at->m_attr) == std::string("startswith")) {
                                     if(args.size() != 1) {
                                         throw SemanticError("str.startwith() takes one argument",