Python: More use of API::Node

tausbn · yoff · web-flow · commit 738d1bc3d46a · 2021-02-08T14:08:16.000+01:00
Co-authored-by: yoff &lt;lerchedahl@gmail.com&gt;
diff --git a/python/ql/src/semmle/python/frameworks/Flask.qll b/python/ql/src/semmle/python/frameworks/Flask.qll
@@ -33,7 +33,7 @@ private module FlaskModel {
     API::Node request() { result = flask_attr("request") }
 
     /** Gets a reference to the `flask.make_response` function. */
-    DataFlow::Node make_response() { result = flask_attr("make_response").getAUse() }
+    API::Node make_response() { result = flask_attr("make_response") }
 
     /**
      * Provides models for the `flask.Flask` class
@@ -407,7 +407,7 @@ private module FlaskModel {
     override CallNode node;
 
     FlaskMakeResponseCall() {
-      node.getFunction() = flask::make_response().asCfgNode()
+      node.getFunction() = flask::make_response().getAUse().asCfgNode()
       or
       node.getFunction() = flask::Flask::make_response_().getAUse().asCfgNode()
     }
