add support for 'Constant' ast node type

yinwang0 · Apr 23, 2022 · d9deca0 · d9deca0
1 parent 8670ec2
commit d9deca0
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/src/main/java/org/yinwang/pysonar/Parser.java b/src/main/java/org/yinwang/pysonar/Parser.java
@@ -668,7 +668,26 @@ else if (s.length() >= 2 && s.startsWith("\"") && s.endsWith("\""))
             return new Yield(value, file, start, end, line, col);
         }
 
-        $.msg("\n[Please Report]: unexpected ast node: " + type);
+        // Python3's new node type 'Constant'
+        // Just convert to other types and call convert again
+        if (type.equals("Constant")) {
+            Object value = map.get("value");
+
+            if (map.containsKey("num_type")) {
+                map.put("pysonar_node_type", "Num");
+                map.put("n", map.get("value"));
+            }
+            else if (value instanceof String) {
+                map.put("pysonar_node_type", "Str");
+                map.put("s", value);
+            }
+            else if (value instanceof Boolean) {
+                map.put("pysonar_node_type", "NameConstant");
+            }
+            return convert(map);
+        }
+
+        $.msg("\n[Please report issue]: Unexpected ast node type: " + type + "\nnode: " + o + "\nfile: " + file);
         return new Unsupported(file, start, end, line, col);
     }