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:
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")
parser = argparse.ArgumentParser(
@ -41,13 +43,13 @@ def argparser() -> argparse.ArgumentParser:
Init
"""
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
"""
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
@ -61,7 +63,9 @@ def argparser() -> argparse.ArgumentParser:
Categorizing
"""
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_report = subparsers.add_parser("report", help="report help")
@ -95,9 +99,8 @@ def parse(args):
Args:
args (dict): argparse variables
db (DBManager): db connection manager
"""
db = DBManager(args.db)
db = DBManager(args.database)
for path in args.path:
if (dir := Path(path)).is_dir():
for file in dir.iterdir():