From 0ce2ced1aadf28e38c518a8786eeab42fa829197 Mon Sep 17 00:00:00 2001 From: Rasmus Wriedt Larsen Date: Thu, 7 Apr 2022 16:16:10 +0200 Subject: [PATCH] Python: Model `pymongo.mongo_client.MongoClient` --- python/ql/src/experimental/semmle/python/frameworks/NoSQL.qll | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/ql/src/experimental/semmle/python/frameworks/NoSQL.qll b/python/ql/src/experimental/semmle/python/frameworks/NoSQL.qll index bdd067218b3..99681c8502d 100644 --- a/python/ql/src/experimental/semmle/python/frameworks/NoSQL.qll +++ b/python/ql/src/experimental/semmle/python/frameworks/NoSQL.qll @@ -15,6 +15,10 @@ private module NoSql { /** Gets a reference to `pymongo.MongoClient` */ private API::Node pyMongo() { result = API::moduleImport("pymongo").getMember("MongoClient").getReturn() + or + // see https://pymongo.readthedocs.io/en/stable/api/pymongo/mongo_client.html#pymongo.mongo_client.MongoClient + result = + API::moduleImport("pymongo").getMember("mongo_client").getMember("MongoClient").getReturn() } /** Gets a reference to `flask_pymongo.PyMongo` */