# HG changeset patch
# User Dennis C. M. <dennis@denniscm.com>
# Date 1686242126 -3600
# Node ID d09dee7a86dae7577990ec3df873c7c13bab4224
# Parent  2350662483a3d02fa1cbf8ce7e6b6b6f9f57cc1c
fix KeyError bug

diff -r 2350662483a3 -r d09dee7a86da upload_document/app.py
--- 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/', '')}"
             }
         )