From ce49d20a5fe3726e1800bc495a25c7617212abf4 Mon Sep 17 00:00:00 2001
From: Emanuele Grasso <96300448+L0P0P@users.noreply.github.com>
Date: Sun, 14 Jul 2024 18:48:09 +0200
Subject: Reaching definition analysis (#17)

Co-authored-by: Santo Cariotti <santo@dcariotti.me>
Co-authored-by: geno <gabriele.genovese2@studio.unibo.it>
Co-authored-by: geno <gabrigeno@gmail.com>
---
 src/ast/nodes/ImportNode.java | 29 ++++++++++++++++++++++++++---
 1 file changed, 26 insertions(+), 3 deletions(-)

(limited to 'src/ast/nodes/ImportNode.java')

diff --git a/src/ast/nodes/ImportNode.java b/src/ast/nodes/ImportNode.java
index 97e3404..1f797f8 100644
--- a/src/ast/nodes/ImportNode.java
+++ b/src/ast/nodes/ImportNode.java
@@ -58,14 +58,14 @@ public class ImportNode implements Node {
     }
 
     @Override
-    public String toPrint(String prefix) {
+    public String printAST(String prefix) {
         String str = prefix + "Import\n";
 
         prefix += "  ";
         if (isFrom) {
-            str += prefix + "  From\n" + dottedName.toPrint(prefix + "  ");
+            str += prefix + "  From\n" + dottedName.printAST(prefix + "  ");
         } else {
-            str += dottedName.toPrint(prefix);
+            str += dottedName.printAST(prefix);
         }
 
         if (importAs) {
@@ -88,4 +88,27 @@ public class ImportNode implements Node {
         return str;
     }
 
+    @Override
+    public String toPrint(String prefix) {
+        String str = prefix;
+
+        if (isFrom) {
+            str += "from " + dottedName.toPrint("") + " import ";
+            if (importAll) {
+                str += "*";
+            } else {
+                str += names.get(0);
+                for (int i = 1; i < names.size(); ++i) {
+                    str += ", " + names.get(i);
+                }
+            }
+        } else {
+            str += "import " + dottedName.toPrint("");
+            if (importAs) {
+                str += " as " + names.get(0);
+            }
+        }
+        return str;
+    }
+
 }
-- 
cgit v1.2.3-18-g5258