Python: Add starlette.requests.URL model

tausbn · RasmusWL · commit 4879a931ebe6 · 2023-12-08T11:27:51.000+01:00
diff --git a/python/ql/lib/semmle/python/frameworks/Starlette.qll b/python/ql/lib/semmle/python/frameworks/Starlette.qll
@@ -103,8 +103,10 @@ module Starlette {
    */
   module Url {
     /** Gets a reference to the `starlette.requests.URL` class. */
-    private API::Node classRef() {
+    API::Node classRef() {
       result = API::moduleImport("starlette").getMember("requests").getMember("URL")
+      or
+      result = ModelOutput::getATypeNode("starlette.requests.URL~Subclass").getASubclass*()
     }
 
     /**
diff --git a/python/ql/src/meta/ClassHierarchy/Find.ql b/python/ql/src/meta/ClassHierarchy/Find.ql
@@ -244,6 +244,12 @@ class StarletteWebsocket extends FindSubclassesSpec {
   override API::Node getAlreadyModeledClass() { result = Starlette::WebSocket::classRef() }
 }
 
+class StarletteUrl extends FindSubclassesSpec {
+  StarletteUrl() { this = "starlette.requests.URL~Subclass" }
+
+  override API::Node getAlreadyModeledClass() { result = Starlette::Url::classRef() }
+}
+
 class ClickhouseClient extends FindSubclassesSpec {
   ClickhouseClient() { this = "clickhouse_driver.client.Client~Subclass" }
 
