Mercurial > public > finance-parser
changeset 11:d09dee7a86da
fix KeyError bug
author | Dennis C. M. <dennis@denniscm.com> |
---|---|
date | Thu, 08 Jun 2023 17:35:26 +0100 |
parents | 2350662483a3 |
children | d4c4cd4760fa |
files | upload_document/app.py |
diffstat | 1 files changed, 17 insertions(+), 13 deletions(-) [+] |
line wrap: on
line diff
--- a/upload_document/app.py Thu Jun 08 17:16:36 2023 +0100 +++ b/upload_document/app.py Thu Jun 08 17:35:26 2023 +0100 @@ -50,27 +50,31 @@ account_value = '' with table.batch_writer() as batch: + try: + account_name = account['1'] - # pk -> item_type#company_ticker - # sk -> date#row_index + # pk -> item_type#company_ticker + # sk -> date#row_index - batch.put_item( - Item={ - 'pk': f"balance#{event_msg['companyTicker']}", - 'sk': f'{date}#{row_index}', - 'account_name': account['1'], - 'account_value': account_value, - 'column_types': column_types, - 'format': doc['format'] - } - ) + batch.put_item( + Item={ + 'pk': f"balance#{event_msg['companyTicker']}", + 'sk': f'{date}#{row_index}', + 'account_name': account_name, + 'account_value': account_value, + 'column_types': column_types, + 'format': doc['format'] + } + ) + except KeyError: + pass # pk -> item_type#company_ticker # sk -> date#filename table.put_item( Item={ - 'pk': f"file#balance#{event_msg['companyTicker']}", + 'pk': f"file#{event_msg['docType']}#{event_msg['companyTicker']}", 'sk': f"{date}#{event_msg['objectKey'].replace('processed/', '')}" } )