Skip to content
Prev Previous commit
Next Next commit
feat: Settings type
  • Loading branch information
HaeckelK committed Feb 14, 2022
commit 66dc3d4927cfd737a312577acbce2c70ace81967
11 changes: 10 additions & 1 deletion python-argparse-generator/src/index.ts
Original file line number Diff line number Diff line change
@@ -1,3 +1,11 @@
export type Settings = {
parserName: string;
}

export const defaultSettings = (): Settings => {
return {parserName: "parser"};
};

export type Argument = {
name: string;
type: string;
Expand Down Expand Up @@ -42,7 +50,8 @@ function argumentToText(argument: Argument, parserName: string) {

const argumentToMainParams = (argument: Argument) => `${argument.variableName}: ${argument.type}`;

export const argparseCode = (args: Argument[], parserName: string = 'parser') => {
export const argparseCode = (args: Argument[], settings: Settings = defaultSettings()) => {
const parserName: string = settings.parserName;
const mainParameters: string[] = args.map((arg) => argumentToMainParams(arg));
const argumentsText: string[] = args.map((arg) => argumentToText(arg, parserName));
const returnText: string[] = args.map((x) => {
Expand Down