feat: implement Python installation via uv

Author: karthiknadig

src/common/localize.ts

@@ -96,6 +96,10 @@ export namespace VenvManagerStrings {
     export const venvErrorNoBasePython = l10n.t('No base Python found');
     export const venvErrorNoPython3 = l10n.t('Did not find any base Python 3');
 
+    export const noEnvClickToCreate = l10n.t('No environment found, click to create');
+    export const noEnvFound = l10n.t('No python environments found.');
+    export const createEnvironment = l10n.t('Create Environment');
+
     export const venvName = l10n.t('Enter a name for the virtual environment');
     export const venvNameErrorEmpty = l10n.t('Name cannot be empty');
     export const venvNameErrorExists = l10n.t('A folder with the same name already exists');
@@ -206,3 +210,19 @@ export namespace ActivationStrings {
     );
     export const activatingEnvironment = l10n.t('Activating environment');
 }
+
+export namespace UvInstallStrings {
+    export const noPythonFound = l10n.t('No Python installation found');
+    export const installPythonPrompt = l10n.t('No Python found. Would you like to install Python using uv?');
+    export const installPythonAndUvPrompt = l10n.t(
+        'No Python found. Would you like to install uv and use it to install Python?',
+    );
+    export const installPython = l10n.t('Install Python');
+    export const installingUv = l10n.t('Installing uv...');
+    export const installingPython = l10n.t('Installing Python via uv...');
+    export const installComplete = l10n.t('Python installed successfully');
+    export const installFailed = l10n.t('Failed to install Python');
+    export const uvInstallFailed = l10n.t('Failed to install uv');
+    export const dontAskAgain = l10n.t("Don't ask again");
+    export const clickToInstallPython = l10n.t('No Python found, click to install');
+}

src/common/telemetry/constants.ts

@@ -10,6 +10,11 @@ export enum EventNames {
     VENV_USING_UV = 'VENV.USING_UV',
     VENV_CREATION = 'VENV.CREATION',
 
+    UV_PYTHON_INSTALL_PROMPTED = 'UV.PYTHON_INSTALL_PROMPTED',
+    UV_PYTHON_INSTALL_STARTED = 'UV.PYTHON_INSTALL_STARTED',
+    UV_PYTHON_INSTALL_COMPLETED = 'UV.PYTHON_INSTALL_COMPLETED',
+    UV_PYTHON_INSTALL_FAILED = 'UV.PYTHON_INSTALL_FAILED',
+
     PACKAGE_MANAGEMENT = 'PACKAGE_MANAGEMENT',
     ADD_PROJECT = 'ADD_PROJECT',
     /**
@@ -83,15 +88,49 @@ export interface IEventNamePropertyMapping {
     /* __GDPR__
         "venv.using_uv": {"owner": "eleanorjboyd" }
     */
-    [EventNames.VENV_USING_UV]: never | undefined /* __GDPR__
+    [EventNames.VENV_USING_UV]: never | undefined;
+
+    /* __GDPR__
         "venv.creation": {
             "creationType": { "classification": "SystemMetaData", "purpose": "FeatureInsight", "owner": "eleanorjboyd" }
         }
-    */;
+    */
     [EventNames.VENV_CREATION]: {
         creationType: 'quick' | 'custom';
     };
 
+    /* __GDPR__
+        "uv.python_install_prompted": {
+            "trigger": { "classification": "SystemMetaData", "purpose": "FeatureInsight", "owner": "karthiknadig" }
+        }
+    */
+    [EventNames.UV_PYTHON_INSTALL_PROMPTED]: {
+        trigger: 'activation' | 'createEnvironment';
+    };
+
+    /* __GDPR__
+        "uv.python_install_started": {
+            "uvAlreadyInstalled": { "classification": "SystemMetaData", "purpose": "FeatureInsight", "owner": "karthiknadig" }
+        }
+    */
+    [EventNames.UV_PYTHON_INSTALL_STARTED]: {
+        uvAlreadyInstalled: boolean;
+    };
+
+    /* __GDPR__
+        "uv.python_install_completed": {"owner": "karthiknadig" }
+    */
+    [EventNames.UV_PYTHON_INSTALL_COMPLETED]: never | undefined;
+
+    /* __GDPR__
+        "uv.python_install_failed": {
+            "stage": { "classification": "SystemMetaData", "purpose": "FeatureInsight", "owner": "karthiknadig" }
+        }
+    */
+    [EventNames.UV_PYTHON_INSTALL_FAILED]: {
+        stage: 'uvInstall' | 'pythonInstall';
+    };
+
     /* __GDPR__
         "package_management": {
             "managerId": { "classification": "SystemMetaData", "purpose": "FeatureInsight", "owner": "eleanorjboyd" },

src/features/views/treeViewItems.ts

@@ -1,6 +1,6 @@
 import { Command, MarkdownString, ThemeIcon, TreeItem, TreeItemCollapsibleState } from 'vscode';
 import { EnvironmentGroupInfo, IconPath, Package, PythonEnvironment, PythonProject } from '../../api';
-import { EnvViewStrings } from '../../common/localize';
+import { EnvViewStrings, UvInstallStrings, VenvManagerStrings } from '../../common/localize';
 import { InternalEnvironmentManager, InternalPackageManager } from '../../internal.api';
 import { isActivatableEnvironment } from '../common/activation';
 import { removable } from './utils';
@@ -115,20 +115,24 @@ export class NoPythonEnvTreeItem implements EnvTreeItem {
         private readonly tooltip?: string | MarkdownString,
         private readonly iconPath?: string | IconPath,
     ) {
-        const item = new TreeItem(
-            this.parent.manager.supportsCreate
-                ? 'No environment found, click to create'
-                : 'No python environments found.',
-            TreeItemCollapsibleState.None,
-        );
+        // Use special message for system manager (Python installation)
+        const isSystemManager = this.parent.manager.name === 'system';
+        let label: string;
+        if (this.parent.manager.supportsCreate) {
+            label = isSystemManager ? UvInstallStrings.clickToInstallPython : VenvManagerStrings.noEnvClickToCreate;
+        } else {
+            label = VenvManagerStrings.noEnvFound;
+        }
+
+        const item = new TreeItem(label, TreeItemCollapsibleState.None);
         item.contextValue = 'python-no-environment';
         item.description = this.description;
         item.tooltip = this.tooltip;
         item.iconPath = this.iconPath ?? new ThemeIcon('circle-slash');
         if (this.parent.manager.supportsCreate) {
             item.command = {
                 command: 'python-envs.create',
-                title: 'Create Environment',
+                title: isSystemManager ? UvInstallStrings.installPython : VenvManagerStrings.createEnvironment,
                 arguments: [this.parent],
             };
         }

src/managers/builtin/sysPythonManager.ts

@@ -1,6 +1,8 @@
 import * as path from 'path';
 import { EventEmitter, LogOutputChannel, MarkdownString, ProgressLocation, ThemeIcon, Uri, window } from 'vscode';
 import {
+    CreateEnvironmentOptions,
+    CreateEnvironmentScope,
     DidChangeEnvironmentEventArgs,
     DidChangeEnvironmentsEventArgs,
     EnvironmentChangeKind,
@@ -28,6 +30,7 @@ import {
     setSystemEnvForWorkspaces,
 } from './cache';
 import { refreshPythons, resolveSystemPythonEnvironmentPath } from './utils';
+import { installPythonWithUv, promptInstallPythonViaUv } from './uvPythonInstaller';
 
 export class SysPythonManager implements EnvironmentManager {
     private collection: PythonEnvironment[] = [];
@@ -70,6 +73,11 @@ export class SysPythonManager implements EnvironmentManager {
 
         await this.internalRefresh(false, SysManagerStrings.sysManagerDiscovering);
 
+        // If no Python environments were found, offer to install via uv
+        if (this.collection.length === 0) {
+            promptInstallPythonViaUv('activation', this.api, this.log);
+        }
+
         this._initialized.resolve();
     }
 
@@ -218,6 +226,25 @@ export class SysPythonManager implements EnvironmentManager {
         return resolved;
     }
 
+    /**
+     * Installs a global Python using uv.
+     * This method installs uv if not present, then uses it to install Python.
+     */
+    async create(
+        _scope: CreateEnvironmentScope,
+        _options?: CreateEnvironmentOptions,
+    ): Promise<PythonEnvironment | undefined> {
+        const success = await installPythonWithUv(this.api, this.log);
+
+        if (success) {
+            // Return the latest Python environment after installation
+            // The installPythonWithUv function already refreshes environments
+            return getLatest(this.collection);
+        }
+
+        return undefined;
+    }
+
     async clearCache(): Promise<void> {
         await clearSystemEnvCache();
     }