personal-finance/budget
2
1
Fork 0
Code Issues 11 Pull Requests Projects 6 Releases 2 Wiki Activity

Add long --database arg option

Browse Source
This commit is contained in:
Luís Murta 2021-07-03 00:51:55 +01:00
parent fb93f9d4fd
commit 1139dff249
Signed by: satprog
GPG Key ID: DDF2EFC6179009DC
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pfbudget/runnable.py
View File

@ -25,7 +25,9 @@ class DataFileMissing(Exception):
def argparser() -> argparse.ArgumentParser: def argparser() -> argparse.ArgumentParser:
help = argparse.ArgumentParser(add_help=False) help = argparse.ArgumentParser(add_help=False)
help.add_argument("--db", help="select current database", default=DEFAULT_DB) help.add_argument(
"-db", "--database", help="select current database", default=DEFAULT_DB
)
help.add_argument("-q", "--quiet", help="quiet") help.add_argument("-q", "--quiet", help="quiet")
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
@ -41,13 +43,13 @@ def argparser() -> argparse.ArgumentParser:
Init Init
""" """
p_init = subparsers.add_parser("init", parents=[help]) p_init = subparsers.add_parser("init", parents=[help])
p_init.set_defaults(func=lambda args: DBManager(args.db).init()) p_init.set_defaults(func=lambda args: DBManager(args.database).init())
""" """
Exporting Exporting
""" """
p_export = subparsers.add_parser("export", parents=[help]) p_export = subparsers.add_parser("export", parents=[help])
p_export.set_defaults(func=lambda args: DBManager(args.db).export()) p_export.set_defaults(func=lambda args: DBManager(args.database).export())
""" """
Parsing Parsing
@ -61,7 +63,9 @@ def argparser() -> argparse.ArgumentParser:
Categorizing Categorizing
""" """
p_categorize = subparsers.add_parser("categorize", parents=[help]) p_categorize = subparsers.add_parser("categorize", parents=[help])
p_categorize.set_defaults(func=lambda args: categorize_data(DBManager(args.db))) p_categorize.set_defaults(
func=lambda args: categorize_data(DBManager(args.database))
)
p_graph = subparsers.add_parser("graph", help="graph help") p_graph = subparsers.add_parser("graph", help="graph help")
p_report = subparsers.add_parser("report", help="report help") p_report = subparsers.add_parser("report", help="report help")
@ -95,9 +99,8 @@ def parse(args):
Args: Args:
args (dict): argparse variables args (dict): argparse variables
db (DBManager): db connection manager
""" """
db = DBManager(args.db) db = DBManager(args.database)
for path in args.path: for path in args.path:
if (dir := Path(path)).is_dir(): if (dir := Path(path)).is_dir():
for file in dir.iterdir(): for file in dir.iterdir():