Spaces:
Sleeping
Sleeping
Create extract_and_store_supabase
Browse files- 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}")
|