diff options
author | HombreLaser <buran@silosneeded.com> | 2024-03-12 23:46:02 -0600 |
---|---|---|
committer | HombreLaser <buran@silosneeded.com> | 2024-03-12 23:46:02 -0600 |
commit | a0fcdc83cbc2b440a5836313b2afd1a35698f5ef (patch) | |
tree | 60cf56be85de882b8999d990feaaf2cf4f683dba /src/arg_parser.py | |
parent | afd4e795e3a7e7b34e0d3127f11c05c0e7a2fede (diff) |
Añade pruebas
Diffstat (limited to 'src/arg_parser.py')
-rw-r--r-- | src/arg_parser.py | 18 |
1 files changed, 15 insertions, 3 deletions
diff --git a/src/arg_parser.py b/src/arg_parser.py index c2db3bd..a5fdb40 100644 --- a/src/arg_parser.py +++ b/src/arg_parser.py @@ -1,4 +1,5 @@ import argparse +import sys AVAILABLE_COMMANDS_HELP_STRING = """Available commands: @@ -6,13 +7,24 @@ AVAILABLE_COMMANDS_HELP_STRING = """Available commands: get-rows n: print first n rows insert "csv string": insert row at the end, the row must have exactly 3 elements. search "lookup": search "lookup" in the table. + delete-at n: delete row at nth position """ class ArgParser: - def __init__(self): + def __init__(self, args=sys.argv): self.parser = argparse.ArgumentParser() self.parser.add_argument('-f', required=True, help='CSV file location') self.parser.add_argument('-c', required=True, help=AVAILABLE_COMMANDS_HELP_STRING, nargs=2) - self.args = self.parser.parse_args() - print(self.args)
\ No newline at end of file + self.args = self.parser.parse_args(args) + + @property + def file(self): + return self.args.f + + @property + def command(self): + return self.args.c + + def print_help(self): + self.parser.print_help()
\ No newline at end of file |