jupyterlab_vscode_icons_extension 1.1.6 → 1.1.14

package/README.md

@@ -21,7 +21,7 @@ This extension brings 1414 beautiful file type icons from the vscode-icons proje
 - Zero configuration required - just install and enjoy
 - Lightweight integration using Iconify's JSON icon format
 - Compatible with Jupytext - properly displays Python and Markdown icons for .py and .md notebook files
-- No folders (we left those alone... for now)
+- Python package folder detection - folders declared in `pyproject.toml` or `setup.py` get a special Python package icon
 
 ## Requirements
 

package/lib/index.js

@@ -2,6 +2,7 @@ import { ISettingRegistry } from '@jupyterlab/settingregistry';
 import { IDefaultFileBrowser } from '@jupyterlab/filebrowser';
 import { LabIcon } from '@jupyterlab/ui-components';
 import { getIconSVG } from './icons';
+import { parsePyprojectToml, parseSetupPy } from './parsers';
 const PLUGIN_ID = 'jupyterlab_vscode_icons_extension:plugin';
 /**
  * Create a LabIcon from vscode-icons SVG data
@@ -369,10 +370,10 @@ const plugin = {
             const wordIcon = createLabIcon('file-type-word');
             const excelIcon = createLabIcon('file-type-excel');
             const powerpointIcon = createLabIcon('file-type-powerpoint');
-            // Custom Markdown icon (from markdown.svg - purple M with arrow, darker #7a2491)
+            // Custom Markdown icon (from markdown.svg - purple M with arrow, #8a2ea5)
             const markdownSvg = `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 309 327">
-        <path fill="#7a2491" opacity="1" stroke="none" d="m 138.68393,230.48651 c 36.58836,3.1e-4 72.68422,3.1e-4 108.78008,3.1e-4 0.13988,0.49669 0.0821,0.12537 -3.34406,3.81472 -27.34165,24.16766 -54.43119,49.41695 -81.72391,73.62893 -2.65146,2.35216 -4.5582,3.21609 -7.64686,0.37229 -26.89754,-24.76539 -75.191307,-68.40096 -80.889724,-74.12425 -0.744118,-0.74735 -1.274501,-1.57204 -2.95867,-3.69233 23.309236,0 45.299954,0 67.783144,3.3e-4 z"/>
-        <path fill="#7a2491" d="m 61.156397,14.443673 h 69.176263 q 14.81059,56.661581 23.29958,97.452667 l 5.96036,-27.150338 q 3.61233,-15.870486 7.76652,-30.954008 l 10.6564,-39.348321 H 248.6367 L 276.09047,189.5437 H 221.90541 L 207.27544,69.137838 173.50009,189.5437 H 136.47364 L 101.07273,68.875516 86.984609,189.5437 H 35.147571 Z"/>
+        <path fill="#8a2ea5" opacity="1" stroke="none" d="m 138.68393,230.48651 c 36.58836,3.1e-4 72.68422,3.1e-4 108.78008,3.1e-4 0.13988,0.49669 0.0821,0.12537 -3.34406,3.81472 -27.34165,24.16766 -54.43119,49.41695 -81.72391,73.62893 -2.65146,2.35216 -4.5582,3.21609 -7.64686,0.37229 -26.89754,-24.76539 -75.191307,-68.40096 -80.889724,-74.12425 -0.744118,-0.74735 -1.274501,-1.57204 -2.95867,-3.69233 23.309236,0 45.299954,0 67.783144,3.3e-4 z"/>
+        <path fill="#8a2ea5" d="m 61.156397,14.443673 h 69.176263 q 14.81059,56.661581 23.29958,97.452667 l 5.96036,-27.150338 q 3.61233,-15.870486 7.76652,-30.954008 l 10.6564,-39.348321 H 248.6367 L 276.09047,189.5437 H 221.90541 L 207.27544,69.137838 173.50009,189.5437 H 136.47364 L 101.07273,68.875516 86.984609,189.5437 H 35.147571 Z"/>
       </svg>`;
             // Custom PDF icon (document with folded corner and red PDF banner)
             const pdfSvg = `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 14 16">
@@ -406,9 +407,10 @@ const plugin = {
         <path fill="url(#py-b)" d="M55,0C29,0,29,10,29,12v13h27v4H19C11,29,0,34,0,55c0,20,8,27,16,27h9V69c0-6,3-16,16-16h26c4,0,15-2,15-14V14C82,11,82,0,55,0zM40,8c3,0,5,2,5,5s-2,5-5,5-5-2-5-5S37,8,40,8z"/>
         <path fill="url(#py-y)" d="M55,110c26,0,26-10,26-12V85H54v-4h37c8,0,18-5,18-26 0-23-11-27-16-27h-9v13c0,6-3,16-16,16H42c-4,0-15,2-15,14v24c0,3,0,14,28,14zM70,101c-3,0-5-2-5-5s2-5,5-5 5,2,5,5S73,101,70,101z"/>
       </svg>`;
-            // Custom README icon (info icon - centered, color #912bac from info2.svg)
+            // Custom README icon (info icon - purple background #912bac, gray "i" #bdbdbd)
             const readmeSvg = `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 100 100">
-        <path fill="#912bac" d="m 9.5247234,11.42511 v 76.943619 l 2.1166666,2.116667 h 77.809717 l 2.116667,-2.116667 v -76.94362 l -2.116667,-2.1166662 -77.809717,0 z m 36.2115596,9.141612 h 11.01003 l 2.116667,2.116667 0,7.619586 -2.116667,2.116667 h -11.01003 l -2.116667,-2.116667 v -7.619586 z m -6.168542,19.478681 h 23.084596 l 2.116667,2.116667 v 5.700733 l -2.116597,2.09951 -3.175139,-0.02574 -2.116597,2.09951 V 67.32187 l 2.116667,2.116667 h 4.211629 l 2.116667,2.116667 v 5.567219 L 63.688967,79.23909 H 38.509408 l -2.116667,-2.116667 v -6.096386 l 2.116667,-2.116667 h 4.968955 L 45.59503,66.792703 V 52.096294 l -2.116667,-2.116667 h -3.910622 l -2.116667,-2.116667 0,-5.70089 z"/>
+        <rect x="9.5" y="9.3" width="82" height="81.2" rx="4" fill="#912bac"/>
+        <path fill="#bdbdbd" d="m 45.736283,20.566722 h 11.01003 l 2.116667,2.116667 0,7.619586 -2.116667,2.116667 h -11.01003 l -2.116667,-2.116667 v -7.619586 z m -6.168542,19.478681 h 23.084596 l 2.116667,2.116667 v 5.700733 l -2.116597,2.09951 -3.175139,-0.02574 -2.116597,2.09951 V 67.32187 l 2.116667,2.116667 h 4.211629 l 2.116667,2.116667 v 5.567219 L 63.688967,79.23909 H 38.509408 l -2.116667,-2.116667 v -6.096386 l 2.116667,-2.116667 h 4.968955 L 45.59503,66.792703 V 52.096294 l -2.116667,-2.116667 h -3.910622 l -2.116667,-2.116667 0,-5.70089 z"/>
       </svg>`;
             // Get SVG content from VSCode icons
             const claudeSvg = claudeIcon.svgstr;
@@ -441,6 +443,8 @@ const plugin = {
             const pytestDataUri = `data:image/svg+xml;base64,${btoa(pytestSvg)}`;
             const pythonPackageSvg = '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 32 32"><defs><linearGradient id="ppa" x1="0%" y1="0%" x2="100%" y2="100%"><stop offset="0" stop-color="#387eb8"/><stop offset="1" stop-color="#366994"/></linearGradient><linearGradient id="ppb" x1="0%" y1="0%" x2="100%" y2="100%"><stop offset="0" stop-color="#ffe052"/><stop offset="1" stop-color="#ffc331"/></linearGradient></defs><g transform="scale(-1,1) translate(-32,0)"><path d="M27.4,5.5H18.1L16,9.7H4.3V26.5H29.5V5.5Zm0,4.2H19.2l1.1-2.1h7.1Z" fill="#58af7b"/><path d="M20.9,11c-5.1,0-4.8,2.2-4.8,2.2v2.3H21v.7H14.2S11,15.8,11,21s2.9,5,2.9,5h1.7V23.6a2.7,2.7,0,0,1,2.8-2.9h4.8a2.6,2.6,0,0,0,2.7-2.6V13.7S26.2,11,20.9,11Zm-2.7,1.5a.9.9,0,1,1-.8.9.9.9,0,0,1,.8-.9Z" fill="url(#ppa)"/><path d="M21.1,31c5.1,0,4.8-2.2,4.8-2.2V26.5H21v-.7h6.8S31,26.1,31,21s-2.9-5-2.9-5h-1.7v2.4a2.7,2.7,0,0,1-2.8,2.9H18.8a2.6,2.6,0,0,0-2.7,2.6v4.4S15.7,31,21,31Zm2.7-1.5a.9.9,0,1,1,.8-.9.9.9,0,0,1-.8.9Z" fill="url(#ppb)"/></g></svg>';
             const pythonPackageDataUri = `data:image/svg+xml;base64,${btoa(pythonPackageSvg)}`;
+            const venvSvg = '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 32 32"><g transform="scale(-1,1) translate(-32,0)"><path d="M27.4,5.5H18.1L16,9.7H4.3V26.5H29.5V5.5Zm0,4.2H19.2l1.1-2.1h7.1Z" fill="#9575cd"/><g transform="translate(22,22) scale(1.25)" fill="#bababa"><path d="M-1.2,-6 L1.2,-6 L1.5,-4.5 L2.8,-4 L4,-5 L5.5,-3.5 L4.5,-2.3 L5,-1 L6.5,-0.8 L6.5,1.2 L5,1.5 L4.5,2.8 L5.5,4 L4,5.5 L2.8,4.5 L1.5,5 L1.2,6.5 L-1.2,6.5 L-1.5,5 L-2.8,4.5 L-4,5.5 L-5.5,4 L-4.5,2.8 L-5,1.5 L-6.5,1.2 L-6.5,-0.8 L-5,-1 L-4.5,-2.3 L-5.5,-3.5 L-4,-5 L-2.8,-4 L-1.5,-4.5 Z"/><circle cx="0" cy="0" r="2.5" fill="#9575cd"/></g></g></svg>';
+            const venvDataUri = `data:image/svg+xml;base64,${btoa(venvSvg)}`;
             // Inject CSS that overrides icons for .py and .md files
             // Note: Jupytext marks .py and .md files as type="notebook", so we need to
             // use JavaScript to detect and mark these files for CSS targeting
@@ -644,6 +648,22 @@ const plugin = {
           background-repeat: no-repeat;
           background-position: center;
         }
+
+        /* Override venv folder icons (.venv, venv, .env, env) */
+        .jp-DirListing-item[data-venv-folder] .jp-DirListing-itemIcon svg,
+        .jp-DirListing-item[data-venv-folder] .jp-DirListing-itemIcon img {
+          display: none !important;
+        }
+        .jp-DirListing-item[data-venv-folder] .jp-DirListing-itemIcon::before {
+          content: '';
+          display: inline-block;
+          width: calc(var(--jp-ui-font-size1, 13px) * var(--jp-custom-icon-scale, 1.5));
+          height: calc(var(--jp-ui-font-size1, 13px) * var(--jp-custom-icon-scale, 1.5));
+          background-image: url('${venvDataUri}');
+          background-size: contain;
+          background-repeat: no-repeat;
+          background-position: center;
+        }
       `;
             // Add CSS to make JavaScript and .env icons less bright
             style.textContent += `
@@ -665,71 +685,60 @@ const plugin = {
           opacity: 55% !important;
         }
       `;
-            // Cache for Python package folder checks
-            const pythonPackageCache = new Map();
-            // Folders to exclude from Python package checking
-            const excludedFolderPatterns = [
-                /^\.git$/,
-                /^\.hg$/,
-                /^\.svn$/,
-                /^\.venv$/,
-                /^venv$/,
-                /^\.env$/,
-                /^env$/,
-                /^node_modules$/,
-                /^__pycache__$/,
-                /^\.ipynb_checkpoints$/,
-                /^\.pytest_cache$/,
-                /^\.mypy_cache$/,
-                /^\.ruff_cache$/,
-                /^\.tox$/,
-                /^\.nox$/,
-                /^\.coverage$/,
-                /^htmlcov$/,
-                /^dist$/,
-                /^build$/,
-                /^\.eggs$/,
-                /\.egg-info$/,
-                /\.dist-info$/
-            ];
-            const isExcludedFolder = (name) => {
-                return excludedFolderPatterns.some(pattern => pattern.test(name));
-            };
-            // Check if a folder is a Python package (contains __init__.py)
-            // Use the file browser's model to get the correct path
-            const checkPythonPackage = async (folderName) => {
-                var _a;
-                // Skip excluded folders
-                if (isExcludedFolder(folderName)) {
-                    return false;
-                }
-                // Get the current path from the file browser model
-                if (!defaultFileBrowser) {
-                    return false;
-                }
-                const currentPath = defaultFileBrowser.model.path;
-                const fullPath = currentPath
-                    ? `${currentPath}/${folderName}`
-                    : folderName;
-                if (pythonPackageCache.has(fullPath)) {
-                    return pythonPackageCache.get(fullPath);
+            // Cache for Python package detection (per directory)
+            let pythonPackagesCache = null;
+            // Detect Python packages by parsing pyproject.toml or setup.py
+            const detectPythonPackages = async () => {
+                const currentPath = (defaultFileBrowser === null || defaultFileBrowser === void 0 ? void 0 : defaultFileBrowser.model.path) || '';
+                // Return cached result if same directory
+                if ((pythonPackagesCache === null || pythonPackagesCache === void 0 ? void 0 : pythonPackagesCache.path) === currentPath) {
+                    return pythonPackagesCache.packages;
                 }
+                const packages = new Set();
                 try {
-                    // Use JupyterLab's contents manager with the correct path from file browser
-                    const contents = app.serviceManager.contents;
-                    const model = await contents.get(fullPath, { content: true });
-                    const hasInit = ((_a = model.content) === null || _a === void 0 ? void 0 : _a.some((item) => item.name === '__init__.py')) ||
-                        false;
-                    pythonPackageCache.set(fullPath, hasInit);
-                    return hasInit;
+                    const contents = await app.serviceManager.contents.get(currentPath, {
+                        content: true
+                    });
+                    const files = contents.content || [];
+                    // Check for pyproject.toml
+                    const hasPyproject = files.some((f) => f.name === 'pyproject.toml');
+                    if (hasPyproject) {
+                        const pyprojectPath = currentPath
+                            ? `${currentPath}/pyproject.toml`
+                            : 'pyproject.toml';
+                        const file = await app.serviceManager.contents.get(pyprojectPath);
+                        const content = file.content;
+                        const parsed = parsePyprojectToml(content);
+                        parsed.forEach(p => packages.add(p));
+                    }
+                    // Check for setup.py
+                    const hasSetupPy = files.some((f) => f.name === 'setup.py');
+                    if (hasSetupPy) {
+                        const setupPath = currentPath
+                            ? `${currentPath}/setup.py`
+                            : 'setup.py';
+                        const file = await app.serviceManager.contents.get(setupPath);
+                        const content = file.content;
+                        const parsed = parseSetupPy(content);
+                        parsed.forEach(p => packages.add(p));
+                    }
                 }
-                catch (_b) {
-                    pythonPackageCache.set(fullPath, false);
-                    return false;
+                catch (_a) {
+                    // Ignore errors - no packages detected
                 }
+                pythonPackagesCache = { path: currentPath, packages };
+                return packages;
             };
+            // Invalidate cache on directory change
+            if (defaultFileBrowser) {
+                defaultFileBrowser.model.pathChanged.connect(() => {
+                    pythonPackagesCache = null;
+                });
+            }
             // Add a MutationObserver to mark special files in the file browser
-            const markSpecialFiles = () => {
+            const markSpecialFiles = async () => {
+                // Get Python packages for current directory (cached)
+                const pythonPackages = await detectPythonPackages();
                 // Process ALL items - clear wrong attributes and set correct ones
                 const allItems = document.querySelectorAll('.jp-DirListing-item');
                 allItems.forEach(item => {
@@ -812,25 +821,39 @@ const plugin = {
                     const isDir = fileType === 'directory' ||
                         item.classList.contains('jp-DirListing-directory');
                     if (isDir) {
-                        // Check if this folder is a Python package (async)
-                        // Pass just the folder name - checkPythonPackage gets current path from file browser
-                        checkPythonPackage(name).then(isPythonPackage => {
-                            if (isPythonPackage) {
+                        // Check if folder is a venv folder (.venv, venv, .env, env)
+                        const venvNames = ['.venv', 'venv', '.env', 'env'];
+                        if (venvNames.includes(nameLower)) {
+                            item.setAttribute('data-venv-folder', 'true');
+                            item.removeAttribute('data-python-package');
+                        }
+                        else {
+                            item.removeAttribute('data-venv-folder');
+                            // Check if folder name matches a detected Python package
+                            if (pythonPackages.has(name)) {
                                 item.setAttribute('data-python-package', 'true');
                             }
                             else {
                                 item.removeAttribute('data-python-package');
                             }
-                        });
+                        }
                     }
                     else {
                         item.removeAttribute('data-python-package');
+                        item.removeAttribute('data-venv-folder');
                     }
                 });
             };
-            // Watch for changes in the file browser
+            // Debounce timeout for MutationObserver
+            let markSpecialFilesTimeout = null;
+            // Watch for changes in the file browser (debounced)
             const observer = new MutationObserver(() => {
-                markSpecialFiles();
+                if (markSpecialFilesTimeout) {
+                    clearTimeout(markSpecialFilesTimeout);
+                }
+                markSpecialFilesTimeout = setTimeout(() => {
+                    markSpecialFiles();
+                }, 100);
             });
             // Start observing when the file browser is ready
             setTimeout(() => {

package/lib/parsers.d.ts

@@ -0,0 +1,16 @@
+/**
+ * Python package configuration file parsers
+ * Extracts package names from pyproject.toml and setup.py files
+ */
+/**
+ * Parse pyproject.toml content to extract Python package names
+ * @param content - The content of pyproject.toml file
+ * @returns Set of package names found in the file
+ */
+export declare function parsePyprojectToml(content: string): Set<string>;
+/**
+ * Parse setup.py content to extract Python package names
+ * @param content - The content of setup.py file
+ * @returns Set of package names found in the file
+ */
+export declare function parseSetupPy(content: string): Set<string>;

package/lib/parsers.js

@@ -0,0 +1,47 @@
+/**
+ * Python package configuration file parsers
+ * Extracts package names from pyproject.toml and setup.py files
+ */
+/**
+ * Parse pyproject.toml content to extract Python package names
+ * @param content - The content of pyproject.toml file
+ * @returns Set of package names found in the file
+ */
+export function parsePyprojectToml(content) {
+    const packages = new Set();
+    // Parse [project] name = "package_name"
+    const nameMatch = content.match(/\[project\][^[]*name\s*=\s*["']([^"']+)["']/s);
+    if (nameMatch) {
+        packages.add(nameMatch[1]);
+        // Also add underscore variant (package-name -> package_name)
+        packages.add(nameMatch[1].replace(/-/g, '_'));
+    }
+    // Parse packages = ["pkg1", "pkg2"]
+    const packagesMatch = content.match(/packages\s*=\s*\[([^\]]+)\]/);
+    if (packagesMatch) {
+        const pkgList = packagesMatch[1].match(/["']([^"']+)["']/g);
+        pkgList === null || pkgList === void 0 ? void 0 : pkgList.forEach(p => packages.add(p.replace(/["']/g, '')));
+    }
+    return packages;
+}
+/**
+ * Parse setup.py content to extract Python package names
+ * @param content - The content of setup.py file
+ * @returns Set of package names found in the file
+ */
+export function parseSetupPy(content) {
+    const packages = new Set();
+    // Parse packages=["pkg1", "pkg2"]
+    const packagesMatch = content.match(/packages\s*=\s*\[([^\]]+)\]/);
+    if (packagesMatch) {
+        const pkgList = packagesMatch[1].match(/["']([^"']+)["']/g);
+        pkgList === null || pkgList === void 0 ? void 0 : pkgList.forEach(p => packages.add(p.replace(/["']/g, '')));
+    }
+    // Parse name="package_name"
+    const nameMatch = content.match(/name\s*=\s*["']([^"']+)["']/);
+    if (nameMatch) {
+        packages.add(nameMatch[1]);
+        packages.add(nameMatch[1].replace(/-/g, '_'));
+    }
+    return packages;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "jupyterlab_vscode_icons_extension",
-    "version": "1.1.6",
+    "version": "1.1.14",
     "description": "Jupyterlab extension with a shameless rip-off of the vscode-icons into our beloved environment",
     "keywords": [
         "jupyter",

package/src/__tests__/parsers.spec.ts

@@ -0,0 +1,342 @@
+/**
+ * Tests for Python package configuration file parsers
+ */
+
+import { parsePyprojectToml, parseSetupPy } from '../parsers';
+
+describe('parsePyprojectToml', () => {
+  describe('project name parsing', () => {
+    it('should parse project name with double quotes', () => {
+      const content = `
+[project]
+name = "my-package"
+version = "1.0.0"
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('my-package')).toBe(true);
+      expect(result.has('my_package')).toBe(true);
+    });
+
+    it('should parse project name with single quotes', () => {
+      const content = `
+[project]
+name = 'my-package'
+version = '1.0.0'
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('my-package')).toBe(true);
+      expect(result.has('my_package')).toBe(true);
+    });
+
+    it('should handle name without hyphens', () => {
+      const content = `
+[project]
+name = "mypackage"
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('mypackage')).toBe(true);
+      // underscore variant is same as original
+      expect(result.size).toBe(1);
+    });
+
+    it('should handle name with multiple hyphens', () => {
+      const content = `
+[project]
+name = "my-awesome-package"
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('my-awesome-package')).toBe(true);
+      expect(result.has('my_awesome_package')).toBe(true);
+    });
+
+    it('should handle whitespace around equals sign', () => {
+      const content = `
+[project]
+name="my-package"
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('my-package')).toBe(true);
+    });
+
+    it('should handle extra whitespace', () => {
+      const content = `
+[project]
+name   =   "my-package"
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('my-package')).toBe(true);
+    });
+  });
+
+  describe('packages array parsing', () => {
+    it('should parse packages array with double quotes', () => {
+      const content = `
+[tool.setuptools]
+packages = ["pkg1", "pkg2", "pkg3"]
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('pkg1')).toBe(true);
+      expect(result.has('pkg2')).toBe(true);
+      expect(result.has('pkg3')).toBe(true);
+    });
+
+    it('should parse packages array with single quotes', () => {
+      const content = `
+[tool.setuptools]
+packages = ['pkg1', 'pkg2']
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('pkg1')).toBe(true);
+      expect(result.has('pkg2')).toBe(true);
+    });
+
+    it('should parse packages array with mixed quotes', () => {
+      const content = `
+packages = ["pkg1", 'pkg2']
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('pkg1')).toBe(true);
+      expect(result.has('pkg2')).toBe(true);
+    });
+
+    it('should parse multiline packages array', () => {
+      const content = `
+packages = [
+    "pkg1",
+    "pkg2",
+    "pkg3"
+]
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('pkg1')).toBe(true);
+      expect(result.has('pkg2')).toBe(true);
+      expect(result.has('pkg3')).toBe(true);
+    });
+
+    it('should parse single package in array', () => {
+      const content = `
+packages = ["single_pkg"]
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('single_pkg')).toBe(true);
+    });
+  });
+
+  describe('combined parsing', () => {
+    it('should parse both project name and packages array', () => {
+      const content = `
+[project]
+name = "main-package"
+version = "1.0.0"
+
+[tool.setuptools]
+packages = ["subpkg1", "subpkg2"]
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('main-package')).toBe(true);
+      expect(result.has('main_package')).toBe(true);
+      expect(result.has('subpkg1')).toBe(true);
+      expect(result.has('subpkg2')).toBe(true);
+    });
+
+    it('should handle real-world pyproject.toml', () => {
+      const content = `
+[build-system]
+requires = ["hatchling>=1.5.0", "jupyterlab>=4.0.0,<5"]
+build-backend = "hatchling.build"
+
+[project]
+name = "jupyterlab_vscode_icons_extension"
+version = "1.1.9"
+description = "VSCode-style file icons for JupyterLab"
+readme = "README.md"
+license = { file = "LICENSE" }
+requires-python = ">=3.8"
+classifiers = [
+    "Framework :: Jupyter",
+    "Framework :: Jupyter :: JupyterLab",
+    "Framework :: Jupyter :: JupyterLab :: 4",
+    "License :: OSI Approved :: BSD License",
+    "Programming Language :: Python",
+    "Programming Language :: Python :: 3",
+]
+
+[tool.hatch.build.targets.wheel.shared-data]
+"jupyterlab_vscode_icons_extension/labextension" = "share/jupyter/labextensions/jupyterlab_vscode_icons_extension"
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('jupyterlab_vscode_icons_extension')).toBe(true);
+    });
+  });
+
+  describe('edge cases', () => {
+    it('should return empty set for empty content', () => {
+      const result = parsePyprojectToml('');
+      expect(result.size).toBe(0);
+    });
+
+    it('should return empty set for content without project section', () => {
+      const content = `
+[build-system]
+requires = ["hatchling"]
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.size).toBe(0);
+    });
+
+    it('should not match name outside [project] section', () => {
+      const content = `
+[other-section]
+name = "wrong-package"
+
+[project]
+version = "1.0.0"
+`;
+      const result = parsePyprojectToml(content);
+      expect(result.has('wrong-package')).toBe(false);
+    });
+  });
+});
+
+describe('parseSetupPy', () => {
+  describe('name parsing', () => {
+    it('should parse name with double quotes', () => {
+      const content = `
+from setuptools import setup
+
+setup(
+    name="my-package",
+    version="1.0.0",
+)
+`;
+      const result = parseSetupPy(content);
+      expect(result.has('my-package')).toBe(true);
+      expect(result.has('my_package')).toBe(true);
+    });
+
+    it('should parse name with single quotes', () => {
+      const content = `
+setup(
+    name='my-package',
+)
+`;
+      const result = parseSetupPy(content);
+      expect(result.has('my-package')).toBe(true);
+    });
+
+    it('should handle name without hyphens', () => {
+      const content = `
+setup(name="mypackage")
+`;
+      const result = parseSetupPy(content);
+      expect(result.has('mypackage')).toBe(true);
+    });
+  });
+
+  describe('packages array parsing', () => {
+    it('should parse packages array inline', () => {
+      const content = `
+setup(
+    packages=["pkg1", "pkg2", "pkg3"],
+)
+`;
+      const result = parseSetupPy(content);
+      expect(result.has('pkg1')).toBe(true);
+      expect(result.has('pkg2')).toBe(true);
+      expect(result.has('pkg3')).toBe(true);
+    });
+
+    it('should parse packages array multiline', () => {
+      const content = `
+setup(
+    packages=[
+        "pkg1",
+        "pkg2",
+    ],
+)
+`;
+      const result = parseSetupPy(content);
+      expect(result.has('pkg1')).toBe(true);
+      expect(result.has('pkg2')).toBe(true);
+    });
+
+    it('should parse packages with single quotes', () => {
+      const content = `
+packages=['pkg1', 'pkg2']
+`;
+      const result = parseSetupPy(content);
+      expect(result.has('pkg1')).toBe(true);
+      expect(result.has('pkg2')).toBe(true);
+    });
+  });
+
+  describe('combined parsing', () => {
+    it('should parse both name and packages', () => {
+      const content = `
+from setuptools import setup
+
+setup(
+    name="main-package",
+    version="1.0.0",
+    packages=["subpkg1", "subpkg2"],
+)
+`;
+      const result = parseSetupPy(content);
+      expect(result.has('main-package')).toBe(true);
+      expect(result.has('main_package')).toBe(true);
+      expect(result.has('subpkg1')).toBe(true);
+      expect(result.has('subpkg2')).toBe(true);
+    });
+
+    it('should handle real-world setup.py', () => {
+      const content = `
+#!/usr/bin/env python
+from setuptools import setup, find_packages
+
+setup(
+    name="my-awesome-lib",
+    version="2.0.0",
+    author="Developer",
+    author_email="dev@example.com",
+    description="An awesome library",
+    packages=find_packages(),
+    python_requires=">=3.8",
+    install_requires=[
+        "numpy>=1.0",
+        "pandas>=1.0",
+    ],
+)
+`;
+      const result = parseSetupPy(content);
+      expect(result.has('my-awesome-lib')).toBe(true);
+      expect(result.has('my_awesome_lib')).toBe(true);
+    });
+  });
+
+  describe('edge cases', () => {
+    it('should return empty set for empty content', () => {
+      const result = parseSetupPy('');
+      expect(result.size).toBe(0);
+    });
+
+    it('should return empty set for content with find_packages() only', () => {
+      const content = `
+setup(
+    packages=find_packages(),
+)
+`;
+      const result = parseSetupPy(content);
+      // find_packages() is dynamic, not a literal array
+      expect(result.size).toBe(0);
+    });
+
+    it('should handle whitespace variations', () => {
+      const content = `
+setup(name  =  "my-pkg",packages  =  ["pkg1"])
+`;
+      const result = parseSetupPy(content);
+      expect(result.has('my-pkg')).toBe(true);
+      expect(result.has('pkg1')).toBe(true);
+    });
+  });
+});

package/src/index.ts

@@ -6,6 +6,7 @@ import { ISettingRegistry } from '@jupyterlab/settingregistry';
 import { IDefaultFileBrowser } from '@jupyterlab/filebrowser';
 import { LabIcon } from '@jupyterlab/ui-components';
 import { getIconSVG } from './icons';
+import { parsePyprojectToml, parseSetupPy } from './parsers';
 
 const PLUGIN_ID = 'jupyterlab_vscode_icons_extension:plugin';
 
@@ -413,10 +414,10 @@ const plugin: JupyterFrontEndPlugin<void> = {
       const excelIcon = createLabIcon('file-type-excel');
       const powerpointIcon = createLabIcon('file-type-powerpoint');
 
-      // Custom Markdown icon (from markdown.svg - purple M with arrow, darker #7a2491)
+      // Custom Markdown icon (from markdown.svg - purple M with arrow, #8a2ea5)
       const markdownSvg = `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 309 327">
-        <path fill="#7a2491" opacity="1" stroke="none" d="m 138.68393,230.48651 c 36.58836,3.1e-4 72.68422,3.1e-4 108.78008,3.1e-4 0.13988,0.49669 0.0821,0.12537 -3.34406,3.81472 -27.34165,24.16766 -54.43119,49.41695 -81.72391,73.62893 -2.65146,2.35216 -4.5582,3.21609 -7.64686,0.37229 -26.89754,-24.76539 -75.191307,-68.40096 -80.889724,-74.12425 -0.744118,-0.74735 -1.274501,-1.57204 -2.95867,-3.69233 23.309236,0 45.299954,0 67.783144,3.3e-4 z"/>
-        <path fill="#7a2491" d="m 61.156397,14.443673 h 69.176263 q 14.81059,56.661581 23.29958,97.452667 l 5.96036,-27.150338 q 3.61233,-15.870486 7.76652,-30.954008 l 10.6564,-39.348321 H 248.6367 L 276.09047,189.5437 H 221.90541 L 207.27544,69.137838 173.50009,189.5437 H 136.47364 L 101.07273,68.875516 86.984609,189.5437 H 35.147571 Z"/>
+        <path fill="#8a2ea5" opacity="1" stroke="none" d="m 138.68393,230.48651 c 36.58836,3.1e-4 72.68422,3.1e-4 108.78008,3.1e-4 0.13988,0.49669 0.0821,0.12537 -3.34406,3.81472 -27.34165,24.16766 -54.43119,49.41695 -81.72391,73.62893 -2.65146,2.35216 -4.5582,3.21609 -7.64686,0.37229 -26.89754,-24.76539 -75.191307,-68.40096 -80.889724,-74.12425 -0.744118,-0.74735 -1.274501,-1.57204 -2.95867,-3.69233 23.309236,0 45.299954,0 67.783144,3.3e-4 z"/>
+        <path fill="#8a2ea5" d="m 61.156397,14.443673 h 69.176263 q 14.81059,56.661581 23.29958,97.452667 l 5.96036,-27.150338 q 3.61233,-15.870486 7.76652,-30.954008 l 10.6564,-39.348321 H 248.6367 L 276.09047,189.5437 H 221.90541 L 207.27544,69.137838 173.50009,189.5437 H 136.47364 L 101.07273,68.875516 86.984609,189.5437 H 35.147571 Z"/>
       </svg>`;
 
       // Custom PDF icon (document with folded corner and red PDF banner)
@@ -453,9 +454,10 @@ const plugin: JupyterFrontEndPlugin<void> = {
         <path fill="url(#py-y)" d="M55,110c26,0,26-10,26-12V85H54v-4h37c8,0,18-5,18-26 0-23-11-27-16-27h-9v13c0,6-3,16-16,16H42c-4,0-15,2-15,14v24c0,3,0,14,28,14zM70,101c-3,0-5-2-5-5s2-5,5-5 5,2,5,5S73,101,70,101z"/>
       </svg>`;
 
-      // Custom README icon (info icon - centered, color #912bac from info2.svg)
+      // Custom README icon (info icon - purple background #912bac, gray "i" #bdbdbd)
       const readmeSvg = `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 100 100">
-        <path fill="#912bac" d="m 9.5247234,11.42511 v 76.943619 l 2.1166666,2.116667 h 77.809717 l 2.116667,-2.116667 v -76.94362 l -2.116667,-2.1166662 -77.809717,0 z m 36.2115596,9.141612 h 11.01003 l 2.116667,2.116667 0,7.619586 -2.116667,2.116667 h -11.01003 l -2.116667,-2.116667 v -7.619586 z m -6.168542,19.478681 h 23.084596 l 2.116667,2.116667 v 5.700733 l -2.116597,2.09951 -3.175139,-0.02574 -2.116597,2.09951 V 67.32187 l 2.116667,2.116667 h 4.211629 l 2.116667,2.116667 v 5.567219 L 63.688967,79.23909 H 38.509408 l -2.116667,-2.116667 v -6.096386 l 2.116667,-2.116667 h 4.968955 L 45.59503,66.792703 V 52.096294 l -2.116667,-2.116667 h -3.910622 l -2.116667,-2.116667 0,-5.70089 z"/>
+        <rect x="9.5" y="9.3" width="82" height="81.2" rx="4" fill="#912bac"/>
+        <path fill="#bdbdbd" d="m 45.736283,20.566722 h 11.01003 l 2.116667,2.116667 0,7.619586 -2.116667,2.116667 h -11.01003 l -2.116667,-2.116667 v -7.619586 z m -6.168542,19.478681 h 23.084596 l 2.116667,2.116667 v 5.700733 l -2.116597,2.09951 -3.175139,-0.02574 -2.116597,2.09951 V 67.32187 l 2.116667,2.116667 h 4.211629 l 2.116667,2.116667 v 5.567219 L 63.688967,79.23909 H 38.509408 l -2.116667,-2.116667 v -6.096386 l 2.116667,-2.116667 h 4.968955 L 45.59503,66.792703 V 52.096294 l -2.116667,-2.116667 h -3.910622 l -2.116667,-2.116667 0,-5.70089 z"/>
       </svg>`;
 
       // Get SVG content from VSCode icons
@@ -493,6 +495,9 @@ const plugin: JupyterFrontEndPlugin<void> = {
       const pythonPackageSvg =
         '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 32 32"><defs><linearGradient id="ppa" x1="0%" y1="0%" x2="100%" y2="100%"><stop offset="0" stop-color="#387eb8"/><stop offset="1" stop-color="#366994"/></linearGradient><linearGradient id="ppb" x1="0%" y1="0%" x2="100%" y2="100%"><stop offset="0" stop-color="#ffe052"/><stop offset="1" stop-color="#ffc331"/></linearGradient></defs><g transform="scale(-1,1) translate(-32,0)"><path d="M27.4,5.5H18.1L16,9.7H4.3V26.5H29.5V5.5Zm0,4.2H19.2l1.1-2.1h7.1Z" fill="#58af7b"/><path d="M20.9,11c-5.1,0-4.8,2.2-4.8,2.2v2.3H21v.7H14.2S11,15.8,11,21s2.9,5,2.9,5h1.7V23.6a2.7,2.7,0,0,1,2.8-2.9h4.8a2.6,2.6,0,0,0,2.7-2.6V13.7S26.2,11,20.9,11Zm-2.7,1.5a.9.9,0,1,1-.8.9.9.9,0,0,1,.8-.9Z" fill="url(#ppa)"/><path d="M21.1,31c5.1,0,4.8-2.2,4.8-2.2V26.5H21v-.7h6.8S31,26.1,31,21s-2.9-5-2.9-5h-1.7v2.4a2.7,2.7,0,0,1-2.8,2.9H18.8a2.6,2.6,0,0,0-2.7,2.6v4.4S15.7,31,21,31Zm2.7-1.5a.9.9,0,1,1,.8-.9.9.9,0,0,1-.8.9Z" fill="url(#ppb)"/></g></svg>';
       const pythonPackageDataUri = `data:image/svg+xml;base64,${btoa(pythonPackageSvg)}`;
+      const venvSvg =
+        '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 32 32"><g transform="scale(-1,1) translate(-32,0)"><path d="M27.4,5.5H18.1L16,9.7H4.3V26.5H29.5V5.5Zm0,4.2H19.2l1.1-2.1h7.1Z" fill="#9575cd"/><g transform="translate(22,22) scale(1.25)" fill="#bababa"><path d="M-1.2,-6 L1.2,-6 L1.5,-4.5 L2.8,-4 L4,-5 L5.5,-3.5 L4.5,-2.3 L5,-1 L6.5,-0.8 L6.5,1.2 L5,1.5 L4.5,2.8 L5.5,4 L4,5.5 L2.8,4.5 L1.5,5 L1.2,6.5 L-1.2,6.5 L-1.5,5 L-2.8,4.5 L-4,5.5 L-5.5,4 L-4.5,2.8 L-5,1.5 L-6.5,1.2 L-6.5,-0.8 L-5,-1 L-4.5,-2.3 L-5.5,-3.5 L-4,-5 L-2.8,-4 L-1.5,-4.5 Z"/><circle cx="0" cy="0" r="2.5" fill="#9575cd"/></g></g></svg>';
+      const venvDataUri = `data:image/svg+xml;base64,${btoa(venvSvg)}`;
 
       // Inject CSS that overrides icons for .py and .md files
       // Note: Jupytext marks .py and .md files as type="notebook", so we need to
@@ -697,6 +702,22 @@ const plugin: JupyterFrontEndPlugin<void> = {
           background-repeat: no-repeat;
           background-position: center;
         }
+
+        /* Override venv folder icons (.venv, venv, .env, env) */
+        .jp-DirListing-item[data-venv-folder] .jp-DirListing-itemIcon svg,
+        .jp-DirListing-item[data-venv-folder] .jp-DirListing-itemIcon img {
+          display: none !important;
+        }
+        .jp-DirListing-item[data-venv-folder] .jp-DirListing-itemIcon::before {
+          content: '';
+          display: inline-block;
+          width: calc(var(--jp-ui-font-size1, 13px) * var(--jp-custom-icon-scale, 1.5));
+          height: calc(var(--jp-ui-font-size1, 13px) * var(--jp-custom-icon-scale, 1.5));
+          background-image: url('${venvDataUri}');
+          background-size: contain;
+          background-repeat: no-repeat;
+          background-position: center;
+        }
       `;
 
       // Add CSS to make JavaScript and .env icons less bright
@@ -720,78 +741,72 @@ const plugin: JupyterFrontEndPlugin<void> = {
         }
       `;
 
-      // Cache for Python package folder checks
-      const pythonPackageCache: Map<string, boolean> = new Map();
-
-      // Folders to exclude from Python package checking
-      const excludedFolderPatterns = [
-        /^\.git$/,
-        /^\.hg$/,
-        /^\.svn$/,
-        /^\.venv$/,
-        /^venv$/,
-        /^\.env$/,
-        /^env$/,
-        /^node_modules$/,
-        /^__pycache__$/,
-        /^\.ipynb_checkpoints$/,
-        /^\.pytest_cache$/,
-        /^\.mypy_cache$/,
-        /^\.ruff_cache$/,
-        /^\.tox$/,
-        /^\.nox$/,
-        /^\.coverage$/,
-        /^htmlcov$/,
-        /^dist$/,
-        /^build$/,
-        /^\.eggs$/,
-        /\.egg-info$/,
-        /\.dist-info$/
-      ];
-
-      const isExcludedFolder = (name: string): boolean => {
-        return excludedFolderPatterns.some(pattern => pattern.test(name));
-      };
+      // Cache for Python package detection (per directory)
+      let pythonPackagesCache: { path: string; packages: Set<string> } | null =
+        null;
 
-      // Check if a folder is a Python package (contains __init__.py)
-      // Use the file browser's model to get the correct path
-      const checkPythonPackage = async (
-        folderName: string
-      ): Promise<boolean> => {
-        // Skip excluded folders
-        if (isExcludedFolder(folderName)) {
-          return false;
-        }
+      // Detect Python packages by parsing pyproject.toml or setup.py
+      const detectPythonPackages = async (): Promise<Set<string>> => {
+        const currentPath = defaultFileBrowser?.model.path || '';
 
-        // Get the current path from the file browser model
-        if (!defaultFileBrowser) {
-          return false;
+        // Return cached result if same directory
+        if (pythonPackagesCache?.path === currentPath) {
+          return pythonPackagesCache.packages;
         }
-        const currentPath = defaultFileBrowser.model.path;
-        const fullPath = currentPath
-          ? `${currentPath}/${folderName}`
-          : folderName;
 
-        if (pythonPackageCache.has(fullPath)) {
-          return pythonPackageCache.get(fullPath)!;
-        }
+        const packages = new Set<string>();
+
         try {
-          // Use JupyterLab's contents manager with the correct path from file browser
-          const contents = app.serviceManager.contents;
-          const model = await contents.get(fullPath, { content: true });
-          const hasInit =
-            model.content?.some((item: any) => item.name === '__init__.py') ||
-            false;
-          pythonPackageCache.set(fullPath, hasInit);
-          return hasInit;
+          const contents = await app.serviceManager.contents.get(currentPath, {
+            content: true
+          });
+          const files = contents.content || [];
+
+          // Check for pyproject.toml
+          const hasPyproject = files.some(
+            (f: any) => f.name === 'pyproject.toml'
+          );
+          if (hasPyproject) {
+            const pyprojectPath = currentPath
+              ? `${currentPath}/pyproject.toml`
+              : 'pyproject.toml';
+            const file = await app.serviceManager.contents.get(pyprojectPath);
+            const content = file.content as string;
+            const parsed = parsePyprojectToml(content);
+            parsed.forEach(p => packages.add(p));
+          }
+
+          // Check for setup.py
+          const hasSetupPy = files.some((f: any) => f.name === 'setup.py');
+          if (hasSetupPy) {
+            const setupPath = currentPath
+              ? `${currentPath}/setup.py`
+              : 'setup.py';
+            const file = await app.serviceManager.contents.get(setupPath);
+            const content = file.content as string;
+            const parsed = parseSetupPy(content);
+            parsed.forEach(p => packages.add(p));
+          }
         } catch {
-          pythonPackageCache.set(fullPath, false);
-          return false;
+          // Ignore errors - no packages detected
         }
+
+        pythonPackagesCache = { path: currentPath, packages };
+        return packages;
       };
 
+      // Invalidate cache on directory change
+      if (defaultFileBrowser) {
+        defaultFileBrowser.model.pathChanged.connect(() => {
+          pythonPackagesCache = null;
+        });
+      }
+
       // Add a MutationObserver to mark special files in the file browser
-      const markSpecialFiles = () => {
+      const markSpecialFiles = async () => {
+        // Get Python packages for current directory (cached)
+        const pythonPackages = await detectPythonPackages();
+
         // Process ALL items - clear wrong attributes and set correct ones
         const allItems = document.querySelectorAll('.jp-DirListing-item');
         allItems.forEach(item => {
@@ -889,24 +904,38 @@ const plugin: JupyterFrontEndPlugin<void> = {
             fileType === 'directory' ||
             item.classList.contains('jp-DirListing-directory');
           if (isDir) {
-            // Check if this folder is a Python package (async)
-            // Pass just the folder name - checkPythonPackage gets current path from file browser
-            checkPythonPackage(name).then(isPythonPackage => {
-              if (isPythonPackage) {
+            // Check if folder is a venv folder (.venv, venv, .env, env)
+            const venvNames = ['.venv', 'venv', '.env', 'env'];
+            if (venvNames.includes(nameLower)) {
+              item.setAttribute('data-venv-folder', 'true');
+              item.removeAttribute('data-python-package');
+            } else {
+              item.removeAttribute('data-venv-folder');
+              // Check if folder name matches a detected Python package
+              if (pythonPackages.has(name)) {
                 item.setAttribute('data-python-package', 'true');
               } else {
                 item.removeAttribute('data-python-package');
               }
-            });
+            }
           } else {
             item.removeAttribute('data-python-package');
+            item.removeAttribute('data-venv-folder');
           }
         });
       };
 
-      // Watch for changes in the file browser
+      // Debounce timeout for MutationObserver
+      let markSpecialFilesTimeout: ReturnType<typeof setTimeout> | null = null;
+
+      // Watch for changes in the file browser (debounced)
       const observer = new MutationObserver(() => {
-        markSpecialFiles();
+        if (markSpecialFilesTimeout) {
+          clearTimeout(markSpecialFilesTimeout);
+        }
+        markSpecialFilesTimeout = setTimeout(() => {
+          markSpecialFiles();
+        }, 100);
       });
 
       // Start observing when the file browser is ready

package/src/parsers.ts

@@ -0,0 +1,57 @@
+/**
+ * Python package configuration file parsers
+ * Extracts package names from pyproject.toml and setup.py files
+ */
+
+/**
+ * Parse pyproject.toml content to extract Python package names
+ * @param content - The content of pyproject.toml file
+ * @returns Set of package names found in the file
+ */
+export function parsePyprojectToml(content: string): Set<string> {
+  const packages = new Set<string>();
+
+  // Parse [project] name = "package_name"
+  const nameMatch = content.match(
+    /\[project\][^[]*name\s*=\s*["']([^"']+)["']/s
+  );
+  if (nameMatch) {
+    packages.add(nameMatch[1]);
+    // Also add underscore variant (package-name -> package_name)
+    packages.add(nameMatch[1].replace(/-/g, '_'));
+  }
+
+  // Parse packages = ["pkg1", "pkg2"]
+  const packagesMatch = content.match(/packages\s*=\s*\[([^\]]+)\]/);
+  if (packagesMatch) {
+    const pkgList = packagesMatch[1].match(/["']([^"']+)["']/g);
+    pkgList?.forEach(p => packages.add(p.replace(/["']/g, '')));
+  }
+
+  return packages;
+}
+
+/**
+ * Parse setup.py content to extract Python package names
+ * @param content - The content of setup.py file
+ * @returns Set of package names found in the file
+ */
+export function parseSetupPy(content: string): Set<string> {
+  const packages = new Set<string>();
+
+  // Parse packages=["pkg1", "pkg2"]
+  const packagesMatch = content.match(/packages\s*=\s*\[([^\]]+)\]/);
+  if (packagesMatch) {
+    const pkgList = packagesMatch[1].match(/["']([^"']+)["']/g);
+    pkgList?.forEach(p => packages.add(p.replace(/["']/g, '')));
+  }
+
+  // Parse name="package_name"
+  const nameMatch = content.match(/name\s*=\s*["']([^"']+)["']/);
+  if (nameMatch) {
+    packages.add(nameMatch[1]);
+    packages.add(nameMatch[1].replace(/-/g, '_'));
+  }
+
+  return packages;
+}