diff --git a/python/ql/lib/semmle/python/frameworks/Django.qll b/python/ql/lib/semmle/python/frameworks/Django.qll index b83259267bb..92195db0721 100644 --- a/python/ql/lib/semmle/python/frameworks/Django.qll +++ b/python/ql/lib/semmle/python/frameworks/Django.qll @@ -86,6 +86,10 @@ module Django { } } + private class MaDSubclass extends ModeledSubclass { + MaDSubclass() { this = ModelOutput::getATypeNode("Django.Views.View~Subclass") } + } + /** Gets a reference to the `django.views.generic.View` class or any subclass. */ API::Node subclassRef() { result = any(ModeledSubclass subclass).getASubclass*() } } @@ -295,6 +299,10 @@ module Django { } } + private class MaDSubclass extends ModeledSubclass { + MaDSubclass() { this = ModelOutput::getATypeNode("Django.Forms.Field~Subclass") } + } + /** Gets a reference to the `django.forms.fields.Field` class or any subclass. */ API::Node subclassRef() { result = any(ModeledSubclass subclass).getASubclass*() } }