Add setting to enable/disable datascience codelens (microsoft#5232)

* Add setting to enable/disable datascience codelens * Fix unit tests

Author: rchiodo

news/1 Enhancements/5211.md

@@ -0,0 +1 @@
+Add setting to just enable/disable the datascience codelens.

package.json

@@ -1453,6 +1453,12 @@
  "description": "Draw a highlight behind the currently active cell.",
  "scope": "resource"
  },
+ "python.dataScience.enableCellCodeLens": {
+ "type": "boolean",
+ "default": true,
+ "description": "Enables code lens for 'cells' in a python file.",
+ "scope": "resource"
+ },
  "python.disableInstallationCheck": {
  "type": "boolean",
  "default": false,

src/client/common/types.ts

@@ -306,6 +306,7 @@ export interface IDataScienceSettings {
  variableExplorerExclude?: string;
  liveShareConnectionTimeout?: number;
  decorateCells?: boolean;
+ enableCellCodeLens?: boolean;
 }
 
 export const IConfigurationService = Symbol('IConfigurationService');

src/client/datascience/editor-integration/codelensprovider.ts

@@ -24,7 +24,7 @@ export class DataScienceCodeLensProvider implements IDataScienceCodeLensProvider
  vscode.CodeLens[] {
  // Don't provide any code lenses if we have not enabled data science
  const settings = this.configuration.getSettings();
- if (!settings.datascience.enabled) {
+ if (!settings.datascience.enabled || !settings.datascience.enableCellCodeLens) {
  // Clear out any existing code watchers, providecodelenses is called on settings change
  // so we don't need to watch the settings change specifically here
  if (this.activeCodeWatchers.length > 0) {

src/test/datascience/editor-integration/codewatcher.unit.test.ts

@@ -68,7 +68,8 @@ suite('DataScience Code Watcher Unit Tests', () => {
  errorBackgroundColor: '#FFFFFF',
  sendSelectionToInteractiveWindow: false,
  codeRegularExpression: '^(#\\s*%%|#\\s*\\<codecell\\>|#\\s*In\\[\\d*?\\]|#\\s*In\\[ \\])',
- markdownRegularExpression: '^(#\\s*%%\\s*\\[markdown\\]|#\\s*\\<markdowncell\\>)'
+ markdownRegularExpression: '^(#\\s*%%\\s*\\[markdown\\]|#\\s*\\<markdowncell\\>)',
+ enableCellCodeLens: true
  };
 
  // Setup the service container to return code watchers