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/', '')}"
             }
         )