diff options
Diffstat (limited to 'python/api_quotes')
| -rw-r--r-- | python/api_quotes/__main__.py | 28 |
1 files changed, 16 insertions, 12 deletions
diff --git a/python/api_quotes/__main__.py b/python/api_quotes/__main__.py index 4a6060f0..26b7e125 100644 --- a/python/api_quotes/__main__.py +++ b/python/api_quotes/__main__.py @@ -9,23 +9,14 @@ from .quotes import MarkitQuoteKind logger = logging.getLogger(__name__) -if __name__ == "__main__": - parser = argparse.ArgumentParser() - parser.add_argument( - "start_from", - type=datetime.date.fromisoformat, - default=datetime.date.today(), - nargs="?", - ) - args = parser.parse_args() + +def run(start_from): for asset_class in ( "ABS", "CD", "TRS", ): - after = ( - int((args.start_from + datetime.timedelta(days=1)).strftime("%s")) * 1000 - ) + after = int((start_from + datetime.timedelta(days=1)).strftime("%s")) * 1000 already_uploaded = MarkitQuoteKind[asset_class].already_uploaded() while True: try: @@ -64,3 +55,16 @@ if __name__ == "__main__": logger.error(f"Issue with {asset_class}: {after}") except AttributeError: MarkitAPI.update_api_key() + + +if __name__ == "__main__": + parser = argparse.ArgumentParser() + parser.add_argument( + "start_from", + type=datetime.date.fromisoformat, + default=datetime.date.today(), + nargs="?", + ) + args = parser.parse_args() + while True: + run(args.start_from) |