Allow single file extensions

Change-Id: I0b9e04e9ba2d73f43fbf08515366bfa303f2557b
Reviewed-by: Eike Ziller <eike.ziller@qt.io>

1 files changed, 10 insertions, 3 deletions

diff --git a/plugins/pythonextensions/pythonextensionsplugin.cpp b/plugins/pythonextensions/pythonextensionsplugin.cpp
index b3730b4..893819c 100644
--- a/plugins/pythonextensions/pythonextensionsplugin.cpp
+++ b/plugins/pythonextensions/pythonextensionsplugin.cpp
@@ -141,9 +141,16 @@ static QVector<Extension> getExtensionList(const QDir &directory)
     QStringList entries = directory.entryList(QDir::AllDirs | QDir::NoDotAndDotDot);
     entries.removeAll("site-packages");
     entries.removeAll("__pycache__");
-    return Utils::transform<QVector>(entries, [](const QString &entry) {
-        return Extension({entry, false});
-    });
+    const QVector<Extension> packageExtensions
+        = Utils::transform<QVector>(entries, [](const QString &entry) {
+              return Extension({entry, false});
+          });
+    const QStringList fileEntries = directory.entryList({"*.py"}, QDir::Files);
+    const QVector<Extension> fileExtensions
+        = Utils::transform<QVector>(fileEntries, [](const QString &entry) {
+              return Extension({entry.left(entry.size() - 3), false});
+          });
+    return packageExtensions + fileExtensions;
 }
 
 QVector<Extension> PythonExtensionsPlugin::extensionList()