Omkar008 commited on
Commit
f126b71
·
verified ·
1 Parent(s): 78babf2

Create extract_and_store_supabase

Browse files
Files changed (1) hide show
  1. extract_and_store_supabase +19 -0
extract_and_store_supabase ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from supabase_models import Supabase_Client
2
+
3
+
4
+ def store_message_data(message: Message):
5
+ attachments = message.attachments
6
+ if attachments:
7
+ for attachment in attachments:
8
+ extension = attachment.filename.split(".")[-1]
9
+ file_name = f"{message.id}_{attachment.attachment_id}.{extension}"
10
+ print(f"file_name: {file_name}")
11
+ supabase = Supabase_Client().instance
12
+ base64_data = attachment.data
13
+ try:
14
+ decoded_data = base64.urlsafe_b64decode(base64_data)
15
+ supabase.storage.from_('receipt_radar').upload(file_name, decoded_data)
16
+ except binascii.Error as e:
17
+ print(f"Error decoding base64 data: {e}")
18
+ except Exception as e:
19
+ print(f"Error uploading file: {e}")