From 36226fc1be554c0a3fffbf7b50a8ba552f74c356 Mon Sep 17 00:00:00 2001
From: Daniil Fajnberg <mail@daniil.fajnberg.de>
Date: Sun, 28 Nov 2021 17:19:12 +0100
Subject: [PATCH] fixed and refactored cli

---
 src/mwfin/__main__.py | 41 +++++++++++++++++++++++++++--------------
 1 file changed, 27 insertions(+), 14 deletions(-)

diff --git a/src/mwfin/__main__.py b/src/mwfin/__main__.py
index 9492793..c2086ed 100644
--- a/src/mwfin/__main__.py
+++ b/src/mwfin/__main__.py
@@ -13,6 +13,30 @@ from .constants import END_DATE
 
 JSON_EXT, CSV_EXT = '.json', '.csv'
 
+TICKER_SYMBOL = 'ticker_symbol'
+QUARTERLY = 'quarterly'
+TO_FILE = 'to_file'
+
+
+def parse_cli() -> dict:
+    parser = ArgumentParser(description="Scrape company financials")
+    parser.add_argument(
+        TICKER_SYMBOL,
+        type=str,
+        help="Stock ticker symbol of the company to be scraped the financials of"
+    )
+    parser.add_argument(
+        '-Q', f'--{QUARTERLY}',
+        action='store_true',
+        help="If set, the financial data for the last quarters is returned; otherwise yearly data is returned."
+    )
+    parser.add_argument(
+        '-f', f'--{TO_FILE.replace("_", "-")}',
+        type=Path,
+        help="Writes results to the specified destination file. If omitted results are printed to stdout."
+    )
+    return vars(parser.parse_args())
+
 
 def write_to_csv(data: Dict[str, ResultDict], file_obj) -> None:
     writer = csv.writer(file_obj)
@@ -24,24 +48,13 @@ def write_to_csv(data: Dict[str, ResultDict], file_obj) -> None:
 
 
 async def main() -> None:
-    parser = ArgumentParser(description="Scrape company financials")
-    parser.add_argument(
-        'symbol',
-        type=str,
-        help="Stock ticker symbol of the company to be scraped the financials of"
-    )
-    parser.add_argument(
-        '-f', '--to-file',
-        type=Path,
-        help="Writes results to the specified destination file. If omitted results are printed to stdout."
-    )
-    args = parser.parse_args()
+    args = parse_cli()
     session = ClientSession()
     try:
-        data = await get_company_financials(args.symbol, False, session)
+        data = await get_company_financials(args[TICKER_SYMBOL], quarterly=args[QUARTERLY], session=session)
     finally:
         await session.close()
-    path: Path = args.to_file
+    path: Path = args[TO_FILE]
     if path is None:
         print(json.dumps(data, indent=2))
         return