bump

Fraunhofer-AISEC · Oct 23, 2023 · b14fb61 · b14fb61
1 parent 566097e
commit b14fb61
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 9 deletions.
diff --git a/...uage-python/src/main/kotlin/de/fraunhofer/aisec/cpg/frontends/python/ExpressionHandler.kt b/...uage-python/src/main/kotlin/de/fraunhofer/aisec/cpg/frontends/python/ExpressionHandler.kt
@@ -49,7 +49,13 @@ class ExpressionHandler(frontend: PythonLanguageFrontend) :
     }
 
     private fun handleTuple(node: PythonAST.Tuple): Expression {
-        TODO()
+        val lst = mutableListOf<Expression>()
+        for (e in node.elts) {
+            lst += handle(e)
+        }
+        val ile = newInitializerListExpression(rawNode = node)
+        ile.initializers = lst.toList()
+        return ile
     }
 
     private fun handleIfExp(node: PythonAST.IfExp): Expression {
@@ -62,7 +68,18 @@ class ExpressionHandler(frontend: PythonLanguageFrontend) :
     }
 
     private fun handleDict(node: PythonAST.Dict): Expression {
-        TODO()
+        val lst = mutableListOf<Expression>()
+        for (i in node.values.indices) {
+            lst +=
+                newKeyValueExpression(
+                    key = node.keys[i]?.let { handle(it) },
+                    value = handle(node.values[i]),
+                    rawNode = node
+                )
+        }
+        val ile = newInitializerListExpression(rawNode = node)
+        ile.initializers = lst.toList()
+        return ile
     }
 
     private fun handleCompare(node: PythonAST.Compare): Expression {

diff --git a/cpg-language-python/src/main/kotlin/de/fraunhofer/aisec/cpg/frontends/python/PythonAST.kt b/cpg-language-python/src/main/kotlin/de/fraunhofer/aisec/cpg/frontends/python/PythonAST.kt
@@ -554,7 +554,7 @@ interface PythonAST {
      * ```
      */
     class Dict(pyObject: PyObject) : ExprBase(pyObject) {
-        val keys: List<ExprBase> by lazy { "keys" of pyObject }
+        val keys: List<ExprBase?> by lazy { "keys" of pyObject }
         val values: List<ExprBase> by lazy { "values" of pyObject }
     }
 

diff --git a/...age-python/src/test/kotlin/de/fraunhofer/aisec/cpg/frontends/python/PythonFrontendTest.kt b/...age-python/src/test/kotlin/de/fraunhofer/aisec/cpg/frontends/python/PythonFrontendTest.kt
@@ -908,9 +908,8 @@ class PythonFrontendTest : BaseTest() {
         assertLocalName("minor", ifCond.lhs as? Reference)
 
         // phr = {"user_id": user_id} | content
-        val phrDeclaration =
-            (ifThen.statements[0] as? DeclarationStatement)?.declarations?.get(0)
-                as? VariableDeclaration
+        val phrDeclaration = (ifThen.statements[0] as? AssignExpression)?.declarations?.get(0)
+
         assertNotNull(phrDeclaration)
         assertLocalName("phr", phrDeclaration)
         val phrInitializer = phrDeclaration.firstAssignment as? BinaryOperator
@@ -919,9 +918,7 @@ class PythonFrontendTest : BaseTest() {
         assertEquals(true, phrInitializer.lhs is InitializerListExpression)
 
         // z = {"user_id": user_id}
-        val elseStmt1 =
-            (ifElse.statements[0] as? DeclarationStatement)?.declarations?.get(0)
-                as? VariableDeclaration
+        val elseStmt1 = (ifElse.statements[0] as? AssignExpression)?.declarations?.get(0)
         assertNotNull(elseStmt1)
         assertLocalName("z", elseStmt1)