DonJayamanne · DonJayamanne · Jun 29, 2017 · May 19, 2017 · Jun 1, 2017
diff --git a/src/client/linters/baseLinter.ts b/src/client/linters/baseLinter.ts
@@ -23,7 +23,6 @@ export interface ILintMessage {
     code: string;
     message: string;
     type: string;
-    possibleWord?: string;
     severity?: LintMessageSeverity;
     provider: string;
 }
@@ -53,6 +52,7 @@ export abstract class BaseLinter {
     private installer: Installer;
     protected pythonSettings: settings.IPythonSettings;
     private _workspaceRootPath: string;
+    protected _columnOffset = 0;
     protected get workspaceRootPath(): string {
         return typeof this._workspaceRootPath === 'string' ? this._workspaceRootPath : vscode.workspace.rootPath;
     }
@@ -84,24 +84,11 @@ export abstract class BaseLinter {
                         match.line = Number(<any>match.line);
                         match.column = Number(<any>match.column);
 
-                        let possibleWord: string;
-                        if (!isNaN(match.column)) {
-                            let sourceLine = document.lineAt(match.line - 1).text;
-                            let sourceStart = sourceLine.substring(match.column - 1);
-
-                            // try to get the first word from the startig position
-                            let possibleProblemWords = sourceStart.match(/\w+/g);
-                            if (possibleProblemWords != null && possibleProblemWords.length > 0 && sourceStart.startsWith(possibleProblemWords[0])) {
-                                possibleWord = possibleProblemWords[0];
-                            }
-                        }
-
                         diagnostics.push({
                             code: match.code,
                             message: match.message,
-                            column: isNaN(match.column) || match.column === 0 ? 0 : match.column - 1,
+                            column: isNaN(match.column) || match.column === 0 ? 0 : match.column - this._columnOffset,
                             line: match.line,
-                            possibleWord: possibleWord,
                             type: match.type,
                             provider: this.Id
                         });

diff --git a/src/client/linters/flake8.ts b/src/client/linters/flake8.ts
@@ -6,6 +6,8 @@ import { Product } from '../common/installer';
 import { TextDocument, CancellationToken } from 'vscode';
 
 export class Linter extends baseLinter.BaseLinter {
+    _columnOffset = 1;
+
     constructor(outputChannel: OutputChannel, workspaceRootPath?: string) {
         super('flake8', Product.flake8, outputChannel, workspaceRootPath);
     }

diff --git a/src/client/linters/pep8Linter.ts b/src/client/linters/pep8Linter.ts
@@ -6,6 +6,8 @@ import { Product } from '../common/installer';
 import { TextDocument, CancellationToken } from 'vscode';
 
 export class Linter extends baseLinter.BaseLinter {
+    _columnOffset = 1;
+
     constructor(outputChannel: OutputChannel, workspaceRootPath?: string) {
         super('pep8', Product.pep8, outputChannel, workspaceRootPath);
     }

diff --git a/src/client/linters/prospector.ts b/src/client/linters/prospector.ts
@@ -54,22 +54,12 @@ export class Linter extends baseLinter.BaseLinter {
                 parsedData.messages.filter((value, index) => index <= this.pythonSettings.linting.maxNumberOfProblems).forEach(msg => {
 
                     let lineNumber = msg.location.line === null || isNaN(msg.location.line) ? 1 : msg.location.line;
-                    let sourceLine = document.lineAt(lineNumber - 1).text;
-                    let sourceStart = sourceLine.substring(msg.location.character);
-
-                    // try to get the first word from the starting position
-                    let possibleProblemWords = sourceStart.match(/\w+/g);
-                    let possibleWord: string;
-                    if (possibleProblemWords != null && possibleProblemWords.length > 0 && sourceStart.startsWith(possibleProblemWords[0])) {
-                        possibleWord = possibleProblemWords[0];
-                    }
 
                     diagnostics.push({
                         code: msg.code,
                         message: msg.message,
                         column: msg.location.character,
                         line: lineNumber,
-                        possibleWord: possibleWord,
                         type: msg.code,
                         provider: `${this.Id} - ${msg.source}`
                     });

diff --git a/src/client/linters/pylama.ts b/src/client/linters/pylama.ts
@@ -8,6 +8,8 @@ import { TextDocument, CancellationToken } from 'vscode';
 const REGEX = '(?<file>.py):(?<line>\\d+):(?<column>\\d+): \\[(?<type>\\w+)\\] (?<code>\\w\\d+):? (?<message>.*)\\r?(\\n|$)';
 
 export class Linter extends baseLinter.BaseLinter {
+    _columnOffset = 1;
+
     constructor(outputChannel: OutputChannel, workspaceRootPath?: string) {
         super('pylama', Product.pylama, outputChannel, workspaceRootPath);
     }

diff --git a/src/client/providers/lintProvider.ts b/src/client/providers/lintProvider.ts
@@ -22,14 +22,8 @@ lintSeverityToVSSeverity.set(linter.LintMessageSeverity.Information, vscode.Diag
 lintSeverityToVSSeverity.set(linter.LintMessageSeverity.Warning, vscode.DiagnosticSeverity.Warning);
 
 function createDiagnostics(message: linter.ILintMessage, document: vscode.TextDocument): vscode.Diagnostic {
-    let endCol = document.lineAt(message.line - 1).text.length;
-
-    // try to get the first word from the startig position
-    if (message.possibleWord === 'string' && message.possibleWord.length > 0) {
-        endCol = message.column + message.possibleWord.length;
-    }
-
-    let range = new vscode.Range(new vscode.Position(message.line - 1, message.column), new vscode.Position(message.line - 1, endCol));
+    let position = new vscode.Position(message.line - 1, message.column);
+    let range = new vscode.Range(position, position);
 
     let severity = lintSeverityToVSSeverity.get(message.severity);
     let diagnostic = new vscode.Diagnostic(range, message.code + ':' + message.message, severity);