KeshavAnandCode/FTCWebsite
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add persistent file storage and improve error handling

Browse Source 
Refactor object storage initialization to use a dedicated function and add error handling for uploads, including a fallback to local storage.

Replit-Commit-Author: Agent
Replit-Commit-Session-Id: cd9a7d26-a4e5-4215-975c-c59f4ed1f06d
Replit-Commit-Checkpoint-Type: full_checkpoint
Replit-Commit-Event-Id: 127466ff-4d39-4e2e-bc28-80d552851c25
Replit-Commit-Screenshot-Url: https://storage.googleapis.com/screenshot-production-us-central1/d0a1d46d-d203-4308-bc6a-312ac7c0243b/cd9a7d26-a4e5-4215-975c-c59f4ed1f06d/05bPjFc
This commit is contained in:
abhiramtx
2025-12-13 21:34:03 +00:00
parent 34a7a59899
commit 6a8eb826a6
3 changed files with 40 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
object_storage.py
View File

@@ -4,35 +4,45 @@ Handles persistent file uploads using Google Cloud Storage (Replit Object Storag
"""
from google.cloud import storage
from google.auth import identity_pool
import os
import json
from uuid import uuid4
from werkzeug.utils import secure_filename
REPLIT_SIDECAR_ENDPOINT = "http://127.0.0.1:1106"
def get_replit_credentials():
"""Create proper credentials for Replit Object Storage"""
credentials_config = {
"audience": "replit",
"subject_token_type": "access_token",
"token_url": f"{REPLIT_SIDECAR_ENDPOINT}/token",
"type": "external_account",
"credential_source": {
"url": f"{REPLIT_SIDECAR_ENDPOINT}/credential",
"format": {
"type": "json",
"subject_token_field_name": "access_token",
},
},
"universe_domain": "googleapis.com",
}
credentials = identity_pool.Credentials.from_info(credentials_config)
return credentials
class ObjectStorageService:
def __init__(self):
"""Initialize the object storage client with Replit credentials"""
credentials = get_replit_credentials()
self.client = storage.Client(
credentials={
"audience": "replit",
"subject_token_type": "access_token",
"token_url": f"{REPLIT_SIDECAR_ENDPOINT}/token",
"type": "external_account",
"credential_source": {
"url": f"{REPLIT_SIDECAR_ENDPOINT}/credential",
"format": {
"type": "json",
"subject_token_field_name": "access_token",
},
},
"universe_domain": "googleapis.com",
},
credentials=credentials,
project="",
)
# Get bucket name from environment variable
self.bucket_name = os.environ.get('OBJECT_STORAGE_BUCKET')
if not self.bucket_name:
raise ValueError(
@@ -56,23 +66,19 @@ class ObjectStorageService:
if not file or not file.filename:
return None
# Generate unique filename
original_filename = secure_filename(file.filename)
file_extension = os.path.splitext(original_filename)[1]
unique_filename = f"{uuid4()}{file_extension}"
object_name = f"{folder}/{unique_filename}"
# Upload to bucket
blob = self.bucket.blob(object_name)
blob.upload_from_file(
file.stream,
content_type=file.content_type or 'application/octet-stream'
)
# Make the blob publicly accessible
blob.make_public()
# Return the public URL
return blob.public_url
def get_blob(self, url):
@@ -80,7 +86,6 @@ class ObjectStorageService:
if not url or not url.startswith('https://storage.googleapis.com/'):
return None
# Extract blob name from URL
parts = url.replace(f'https://storage.googleapis.com/{self.bucket_name}/', '')
return self.bucket.blob(parts)