Implement copying python import path from opened file #25026
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,70 @@ | ||
| import * as path from 'path'; | ||
| import * as vscode from 'vscode'; | ||
| import { inject, injectable } from 'inversify'; | ||
|
|
||
| import { IClipboard, ICommandManager, IWorkspaceService } from '../../common/application/types'; | ||
| import { IExtensionSingleActivationService } from '../../activation/types'; | ||
| import { Commands } from '../../common/constants'; | ||
| import { getSysPath } from '../../common/utils/pythonUtils'; | ||
|
|
||
| @injectable() | ||
| export class CopyImportPathCommand implements IExtensionSingleActivationService { | ||
| public readonly supportedWorkspaceTypes = { untrustedWorkspace: true, virtualWorkspace: true }; | ||
|
|
||
| constructor( | ||
| @inject(ICommandManager) private readonly commands: ICommandManager, | ||
| @inject(IWorkspaceService) private readonly workspace: IWorkspaceService, | ||
| @inject(IClipboard) private readonly clipboard: IClipboard, | ||
| ) {} | ||
|
|
||
| async activate(): Promise<void> { | ||
| this.commands.registerCommand(Commands.CopyImportPath, this.execute, this); | ||
| } | ||
|
|
||
| private async execute(fileUri?: vscode.Uri): Promise<void> { | ||
| const uri = fileUri ?? vscode.window.activeTextEditor?.document.uri; | ||
| if (!uri || uri.scheme !== 'file' || !uri.fsPath.endsWith('.py')) { | ||
s-kai273 marked this conversation as resolved.
Show resolved
Hide resolved
|
||
| void vscode.window.showWarningMessage('No Python file selected for import-path copy.'); | ||
| return; | ||
| } | ||
|
|
||
| const importPath = this.resolveImportPath(uri.fsPath); | ||
| // await vscode.env.clipboard.writeText(importPath); | ||
| await this.clipboard.writeText(importPath); | ||
| void vscode.window.showInformationMessage(`Copied: ${importPath}`); | ||
| } | ||
|
|
||
| /** | ||
| * Resolves a Python import-style dotted path from an absolute file path. | ||
|
There was a problem hiding this comment. As described here, the strategy for resolving the Python import path is as follows:
I believe this approach is reasonable, but if you have any suggestions for improvement, I’d love to hear them! |
||
| * | ||
| * The resolution follows a 3-level fallback strategy: | ||
| * | ||
| * 1. If the file is located under any entry in `sys.path`, the path relative to that entry is used. | ||
| * 2. If the file is located under the current workspace folder, the path relative to the workspace root is used. | ||
| * 3. Otherwise, the import path falls back to the file name (without extension). | ||
| * | ||
| * @param absPath - The absolute path to a `.py` file. | ||
| * @returns The resolved import path in dotted notation (e.g., 'pkg.module'). | ||
| */ | ||
| private resolveImportPath(absPath: string): string { | ||
| // ---------- ① sys.path ---------- | ||
| for (const sysRoot of getSysPath()) { | ||
| if (sysRoot && absPath.startsWith(sysRoot)) { | ||
| return CopyImportPathCommand.toDotted(path.relative(sysRoot, absPath)); | ||
| } | ||
| } | ||
|
|
||
| // ---------- ② workspaceFolder ---------- | ||
| const ws = this.workspace.getWorkspaceFolder(vscode.Uri.file(absPath)); | ||
| if (ws && absPath.startsWith(ws.uri.fsPath)) { | ||
| return CopyImportPathCommand.toDotted(path.relative(ws.uri.fsPath, absPath)); | ||
| } | ||
|
|
||
| // ---------- ③ fallback ---------- | ||
| return path.basename(absPath, '.py'); | ||
| } | ||
|
|
||
| private static toDotted(relPath: string): string { | ||
| return relPath.replace(/\.py$/i, '').split(path.sep).filter(Boolean).join('.'); | ||
| } | ||
| } | ||
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,10 @@ | ||
| import { IServiceManager } from '../../ioc/types'; | ||
| import { IExtensionSingleActivationService } from '../../activation/types'; | ||
| import { CopyImportPathCommand } from './copyImportPathCommand'; | ||
|
|
||
| export function registerTypes(serviceManager: IServiceManager): void { | ||
| serviceManager.addSingleton<IExtensionSingleActivationService>( | ||
| IExtensionSingleActivationService, | ||
| CopyImportPathCommand, | ||
| ); | ||
| } |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,13 @@ | ||
| import { execFileSync } from 'child_process'; | ||
|
|
||
| export function getSysPath(pythonCmd = 'python3'): string[] { | ||
|
There was a problem hiding this comment. This isn't safe if there's a json.py file in the user's workspace (people sometimes open folders without checking). You might want to copy this code here: There was a problem hiding this comment. Thank you for your insightful comment. |
||
| try { | ||
| const out = execFileSync(pythonCmd, ['-c', 'import sys, json; print(json.dumps(sys.path))'], { | ||
| encoding: 'utf-8', | ||
| }); | ||
| return JSON.parse(out); | ||
| } catch (err) { | ||
| console.warn('[CopyImportPath] getSysPath failed:', err); | ||
| return []; | ||
| } | ||
| } | ||
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,92 @@ | ||
| // Copyright (c) Microsoft Corporation. All rights reserved. | ||
| // Licensed under the MIT License. | ||
|
|
||
| 'use strict'; | ||
|
|
||
| import { anything, instance, mock, verify, when } from 'ts-mockito'; | ||
| import { expect } from 'chai'; | ||
| import * as vscode from 'vscode'; | ||
| import { CopyImportPathCommand } from '../../../client/application/importPath/copyImportPathCommand'; | ||
| import { IClipboard, ICommandManager, IWorkspaceService } from '../../../client/common/application/types'; | ||
| import * as pythonUtils from '../../../client/common/utils/pythonUtils'; | ||
| import { ClipboardService } from '../../../client/common/application/clipboard'; | ||
| import { CommandManager } from '../../../client/common/application/commandManager'; | ||
| import { WorkspaceService } from '../../../client/common/application/workspace'; | ||
|
|
||
| suite('Copy Import Path Command', () => { | ||
| let command: CopyImportPathCommand; | ||
| let commandManager: ICommandManager; | ||
| let workspaceService: IWorkspaceService; | ||
| let clipboard: IClipboard; | ||
| let originalGetSysPath: () => string[]; | ||
|
|
||
| let clipboardText = ''; | ||
|
|
||
| setup(() => { | ||
| commandManager = mock(CommandManager); | ||
| workspaceService = mock(WorkspaceService); | ||
| clipboard = mock(ClipboardService); | ||
| command = new CopyImportPathCommand(instance(commandManager), instance(workspaceService), instance(clipboard)); | ||
| originalGetSysPath = pythonUtils.getSysPath; | ||
|
|
||
| clipboardText = ''; | ||
| when(clipboard.writeText(anything())).thenCall(async (text: string) => { | ||
| clipboardText = text; | ||
| }); | ||
| }); | ||
|
|
||
| teardown(() => { | ||
| ((pythonUtils as unknown) as { getSysPath: () => string[] }).getSysPath = originalGetSysPath; | ||
| }); | ||
|
|
||
| test('Confirm command handler is added', async () => { | ||
| await command.activate(); | ||
| verify(commandManager.registerCommand('python.copyImportPath', anything(), anything())).once(); | ||
| }); | ||
|
|
||
| test('execute() – sys.path match takes precedence', async () => { | ||
| const absPath = '/home/user/project/src/pkg/module.py'; | ||
| const uri = vscode.Uri.file(absPath); | ||
| ((pythonUtils as unknown) as { getSysPath: () => string[] }).getSysPath = () => ['/home/user/project/src']; | ||
|
|
||
| when(workspaceService.getWorkspaceFolder(anything())).thenReturn(undefined); | ||
| ((vscode.window as unknown) as { activeTextEditor: { document: { uri: vscode.Uri } } }).activeTextEditor = { | ||
| document: { uri }, | ||
| }; | ||
|
|
||
| await ((command as unknown) as { execute(u: vscode.Uri): Promise<void> }).execute(uri); | ||
| expect(clipboardText).to.equal('pkg.module'); | ||
| }); | ||
|
|
||
| test('execute() – workspaceFolder used when no sys.path match', async () => { | ||
| const absPath = '/home/user/project/tools/util.py'; | ||
| const uri = vscode.Uri.file(absPath); | ||
| ((pythonUtils as unknown) as { getSysPath: () => string[] }).getSysPath = () => []; | ||
|
|
||
| const wsFolder = { | ||
| uri: vscode.Uri.file('/home/user/project'), | ||
| name: 'project', | ||
| index: 0, | ||
| } as vscode.WorkspaceFolder; | ||
| when(workspaceService.getWorkspaceFolder(anything())).thenReturn(wsFolder); | ||
|
|
||
| ((vscode.window as unknown) as { activeTextEditor: { document: { uri: vscode.Uri } } }).activeTextEditor = { | ||
| document: { uri }, | ||
| }; | ||
| await ((command as unknown) as { execute(u: vscode.Uri): Promise<void> }).execute(uri); | ||
| expect(clipboardText).to.equal('tools.util'); | ||
| }); | ||
|
|
||
| test('execute() – falls back to filename when no matches', async () => { | ||
| const absPath = '/tmp/standalone.py'; | ||
| const uri = vscode.Uri.file(absPath); | ||
| ((pythonUtils as unknown) as { getSysPath: () => string[] }).getSysPath = () => []; | ||
| when(workspaceService.getWorkspaceFolder(anything())).thenReturn(undefined); | ||
|
|
||
| ((vscode.window as unknown) as { activeTextEditor: { document: { uri: vscode.Uri } } }).activeTextEditor = { | ||
| document: { uri }, | ||
| }; | ||
| await ((command as unknown) as { execute(u: vscode.Uri): Promise<void> }).execute(uri); | ||
| expect(clipboardText).to.equal('standalone'); | ||
| }); | ||
| }); |
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I've set
Ctrl+Alt+Shift+Ias the default keybinding.If you find it inconvenient or have a better suggestion, please let me know!