update to add all new projects to projects view (#537)

fixes #517

Author: eleanorjboyd

src/extension.ts

@@ -119,7 +119,7 @@ export async function activate(context: ExtensionContext): Promise<PythonEnviron
         projectCreators.registerPythonProjectCreator(new ExistingProjects(projectManager)),
         projectCreators.registerPythonProjectCreator(new AutoFindProjects(projectManager)),
         projectCreators.registerPythonProjectCreator(new NewPackageProject(envManagers, projectManager)),
-        projectCreators.registerPythonProjectCreator(new NewScriptProject()),
+        projectCreators.registerPythonProjectCreator(new NewScriptProject(projectManager)),
     );
 
     setPythonApi(envManagers, projectManager, projectCreators, terminalManager, envVarManager);

src/features/creators/newPackageProject.ts

@@ -114,15 +114,8 @@ export class NewPackageProject implements PythonProjectCreator {
             await replaceInFilesAndNames(projectDestinationFolder, 'package_name', packageName);
 
             // 4. Create virtual environment if requested
-            let createdPackage: PythonProject | undefined;
             if (createVenv) {
-                createdPackage = {
-                    name: packageName,
-                    uri: Uri.file(projectDestinationFolder),
-                };
-
                 // add package to list of packages before creating the venv
-                this.projectManager.add(createdPackage);
                 await quickCreateNewVenv(this.envManagers, projectDestinationFolder);
             }
 
@@ -156,13 +149,12 @@ export class NewPackageProject implements PythonProjectCreator {
             };
             await manageLaunchJsonFile(destRoot, JSON.stringify(launchJsonConfig));
 
-            if (createdPackage) {
-                // return package if created (ie when venv is created)
-                return createdPackage;
-            } else {
-                // otherwise its not a package and just a folder
-                return Uri.file(projectDestinationFolder);
-            }
+            const createdPackage: PythonProject | undefined = {
+                name: packageName,
+                uri: Uri.file(projectDestinationFolder),
+            };
+            this.projectManager.add(createdPackage);
+            return createdPackage;
         }
         return undefined;
     }

src/features/creators/newScriptProject.ts

@@ -5,6 +5,7 @@ import { PythonProject, PythonProjectCreator, PythonProjectCreatorOptions } from
 import { NEW_PROJECT_TEMPLATES_FOLDER } from '../../common/constants';
 import { traceError } from '../../common/logging';
 import { showInputBoxWithButtons } from '../../common/window.apis';
+import { PythonProjectManager } from '../../internal.api';
 import { isCopilotInstalled, manageCopilotInstructionsFile, replaceInFilesAndNames } from './creationHelpers';
 
 export class NewScriptProject implements PythonProjectCreator {
@@ -13,7 +14,7 @@ export class NewScriptProject implements PythonProjectCreator {
     public readonly description = l10n.t('Creates a new script folder in your current workspace with PEP 723 support');
     public readonly tooltip = new MarkdownString(l10n.t('Create a new Python script'));
 
-    constructor() {}
+    constructor(private readonly projectManager: PythonProjectManager) {}
 
     async create(options?: PythonProjectCreatorOptions): Promise<PythonProject | Uri | undefined> {
         // quick create (needs name, will always create venv and copilot instructions)
@@ -113,7 +114,13 @@ export class NewScriptProject implements PythonProjectCreator {
                 ]);
             }
 
-            return Uri.file(scriptDestination);
+            // Add the created script to the project manager
+            const createdScript: PythonProject | undefined = {
+                name: scriptFileName,
+                uri: Uri.file(scriptDestination),
+            };
+            this.projectManager.add(createdScript);
+            return createdScript;
         }
         return undefined;
     }

src/features/views/treeViewItems.ts

@@ -1,9 +1,9 @@
-import { TreeItem, TreeItemCollapsibleState, MarkdownString, Command, ThemeIcon } from 'vscode';
+import { Command, MarkdownString, ThemeIcon, TreeItem, TreeItemCollapsibleState } from 'vscode';
+import { EnvironmentGroupInfo, IconPath, Package, PythonEnvironment, PythonProject } from '../../api';
+import { EnvViewStrings } from '../../common/localize';
 import { InternalEnvironmentManager, InternalPackageManager } from '../../internal.api';
-import { PythonEnvironment, IconPath, Package, PythonProject, EnvironmentGroupInfo } from '../../api';
-import { removable } from './utils';
 import { isActivatableEnvironment } from '../common/activation';
-import { EnvViewStrings } from '../../common/localize';
+import { removable } from './utils';
 
 export enum EnvTreeItemKind {
     manager = 'python-env-manager',