From 7d36cc43d2915c41a65424c7f683fe4ce7a989fb Mon Sep 17 00:00:00 2001
From: Saurabh Kumar <developer.saurabh@outlook.com>
Date: Thu, 9 May 2024 15:09:41 +0530
Subject: [PATCH] Add `SetConstant` visitor

---
 src/libasr/codegen/asr_to_python.cpp | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/src/libasr/codegen/asr_to_python.cpp b/src/libasr/codegen/asr_to_python.cpp
index 204880be21..60d8ce3ad2 100644
--- a/src/libasr/codegen/asr_to_python.cpp
+++ b/src/libasr/codegen/asr_to_python.cpp
@@ -150,6 +150,9 @@ class ASRToLpythonVisitor : public ASR::BaseVisitor<ASRToLpythonVisitor>
             } case ASR::ttypeType::Logical : {
                 r = "bool";
                 break;
+            } case ASR::ttypeType::Set : {
+                r = ASRUtils::type_to_str_python(t);
+                break;
             } default : {
                 throw LCompilersException("The type `"
                     + ASRUtils::type_to_str_python(t) + "` is not handled yet");
@@ -619,6 +622,23 @@ class ASRToLpythonVisitor : public ASR::BaseVisitor<ASRToLpythonVisitor>
         s = r;
     }
 
+    void visit_SetConstant(const ASR::SetConstant_t &x) {
+        std::string r = "";
+        r += "{";
+        size_t i = 0;
+        while (i < x.n_elements - 1) {
+            visit_expr(*x.m_elements[i]);
+            r += s;
+            r += ", ";
+            i++;
+        }
+        visit_expr(*x.m_elements[i]);
+        r += s;
+        r += "}";
+
+        s = r;
+    }
+
 };
 
 Result<std::string> asr_to_python(Allocator& al, ASR::TranslationUnit_t &asr,