Pass params to docgenpython

2 files changed, 7 insertions, 1 deletions

diff --git a/src/Lang/Python/PythonDocGen.ts b/src/Lang/Python/PythonDocGen.ts
index 4b90f86..a784f3c 100644
--- a/src/Lang/Python/PythonDocGen.ts
+++ b/src/Lang/Python/PythonDocGen.ts
@@ -2,6 +2,12 @@ import { Range } from "vscode";
 import { IDocGen } from "../../Common/IDocGen";
 
 export class PythonDocGen implements IDocGen {
+    private params: string[];
+
+    public constructor(params: string[]) {
+        this.params = params;
+    }
+
     public GenerateDoc(rangeToReplace: Range) {
         throw new Error("Method not implemented.");
     }
diff --git a/src/Lang/Python/PythonParser.ts b/src/Lang/Python/PythonParser.ts
index 74f1bef..261f420 100644
--- a/src/Lang/Python/PythonParser.ts
+++ b/src/Lang/Python/PythonParser.ts
@@ -18,7 +18,7 @@ export default class CppParser implements ICodeParser {
         this.activeEditor = activeEditor;
         this.activeSelection = activeEditor.selection.active;
         const params = this.getParams(this.getLogicalLine());
-        return new PythonDocGen();
+        return new PythonDocGen(params);
     }
 
     private getLogicalLine(): string {