Python: Add Django FileField model

tausbn · RasmusWL · commit 1f66659ff26a · 2023-12-08T11:27:52.000+01:00
diff --git a/python/ql/lib/semmle/python/frameworks/Django.qll b/python/ql/lib/semmle/python/frameworks/Django.qll
@@ -781,6 +781,9 @@ module PrivateDjango {
                     .getMember(className)
                     .getASubclass*()
             )
+            or
+            result =
+              ModelOutput::getATypeNode("django.db.models.FileField~Subclass").getASubclass*()
           }
         }
 
diff --git a/python/ql/src/meta/ClassHierarchy/Find.ql b/python/ql/src/meta/ClassHierarchy/Find.ql
@@ -437,6 +437,14 @@ class AiohttpView extends FindSubclassesSpec {
   override API::Node getAlreadyModeledClass() { result = AiohttpWebModel::View::subclassRef() }
 }
 
+class DjangoFileField extends FindSubclassesSpec {
+  DjangoFileField() { this = "django.db.models.FileField~Subclass" }
+
+  override API::Node getAlreadyModeledClass() {
+    result = PrivateDjango::DjangoImpl::DB::Models::FileField::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
`@@ -781,6 +781,9 @@ module PrivateDjango {`
`781`	`781`	`.getMember(className)`
`782`	`782`	`.getASubclass*()`
`783`	`783`	`)`
	`784`	`+ or`
	`785`	`+ result =`
	`786`	`+ ModelOutput::getATypeNode("django.db.models.FileField~Subclass").getASubclass*()`
`784`	`787`	`}`
`785`	`788`	`}`
`786`	`789`