Python: extend aiohttp model

tausbn · RasmusWL · commit aa3f1cf3e1fc · 2023-12-08T11:27:52.000+01:00
diff --git a/python/ql/lib/semmle/python/frameworks/Aiohttp.qll b/python/ql/lib/semmle/python/frameworks/Aiohttp.qll
@@ -32,6 +32,8 @@ module AiohttpWebModel {
     /** Gets a reference to the `aiohttp.web.View` class or any subclass. */
     API::Node subclassRef() {
       result = API::moduleImport("aiohttp").getMember("web").getMember("View").getASubclass*()
+      or
+      result = ModelOutput::getATypeNode("aiohttp.web.View~Subclass").getASubclass*()
     }
   }
 
diff --git a/python/ql/src/meta/ClassHierarchy/Find.ql b/python/ql/src/meta/ClassHierarchy/Find.ql
@@ -431,6 +431,12 @@ class AiochClient extends FindSubclassesSpec {
   override API::Node getAlreadyModeledClass() { result = Aioch::Client::subclassRef() }
 }
 
+class AiohttpView extends FindSubclassesSpec {
+  AiohttpView() { this = "aiohttp.web.View~Subclass" }
+
+  override API::Node getAlreadyModeledClass() { result = AiohttpWebModel::View::subclassRef() }
+}
+
 bindingset[fullyQualified]
 predicate fullyQualifiedToYamlFormat(string fullyQualified, string type2, string path) {
   exists(int firstDot | firstDot = fullyQualified.indexOf(".", 0, 0) |

Original file line number	Diff line number	Diff line change
`@@ -32,6 +32,8 @@ module AiohttpWebModel {`
`32`	`32`	/** Gets a reference to the `aiohttp.web.View` class or any subclass. */
`33`	`33`	`API::Node subclassRef() {`
`34`	`34`	`result = API::moduleImport("aiohttp").getMember("web").getMember("View").getASubclass*()`
	`35`	`+ or`
	`36`	`+ result = ModelOutput::getATypeNode("aiohttp.web.View~Subclass").getASubclass*()`
`35`	`37`	`}`
`36`	`38`	`}`
`37`	`39`