Upload 5 files

src/firebase/firebase_config.py

@@ -0,0 +1,36 @@
+import firebase_admin
+from firebase_admin import credentials
+from firebase_admin import storage
+
+import os
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
+firebase_url_storageBucket = os.getenv("URL_STORAGEBUCKET")
+
+# Get credentials from environment variables
+credential_firebase = {
+    "type": os.getenv("TYPE"),
+    "project_id": os.getenv("PROJECT_ID"),
+    "private_key_id": os.getenv("PRIVATE_KEY_ID"),
+    "private_key": os.getenv("PRIVATE_KEY"),
+    "client_email": os.getenv("CLIENT_EMAIL"),
+    "client_id": os.getenv("CLIENT_ID"),
+    "auth_uri": os.getenv("AUTH_URI"),
+    "token_uri": os.getenv("TOKEN_URI"),
+    "auth_provider_x509_cert_url": os.getenv("AUTH_PROVIDER_X509_CERT_URL"),
+    "client_x509_cert_url": os.getenv("CLIENT_X509_CERT_URL"),
+    "universe_domain": os.getenv("UNIVERSE_DOMAIN"),
+}
+
+
+# Check if the app is not initialized yet
+if not firebase_admin._apps:
+    # Initialize the app with the credentials
+    cred = credentials.Certificate(credential_firebase)
+    firebase_admin.initialize_app(cred, {"storageBucket": firebase_url_storageBucket})
+
+# Initialize Firestore
+firebase_bucket = storage.bucket(app=firebase_admin.get_app())
+print("Storage connected")

src/firebase/firebase_provider.py

@@ -0,0 +1,96 @@
+from .firebase_config import firebase_bucket
+import base64
+import os
+import tempfile
+from PIL import Image
+import io
+
+def upload_file_to_storage(file_path, file_name):
+    """
+    Upload a file to Firebase Storage
+    param:
+        file_path: str - The path of the file on local machine to be uploaded
+    return:
+        str - The public URL of the uploaded file
+    """
+    blob = firebase_bucket.blob(file_name)
+    blob.upload_from_filename(file_path)
+    blob.make_public()
+
+    return blob.public_url
+
+
+
+def delete_file_from_storage(file_name):
+    """
+    Delete a file from Firebase Storage
+    param:
+        file_name: str - The name of the file to be deleted
+    return:
+        bool - True if the file is deleted successfully, False if the file is not found
+    """
+    try:
+        blob = firebase_bucket.blob(file_name)
+        blob.delete()
+        return True
+    except Exception as e:
+        print("Error:", e)
+        return False
+
+
+def list_all_files_in_storage():
+    """
+    View all files in Firebase Storage
+    return:
+        dict - Dictionary with keys are names and values are url of all files in Firebase Storage
+    """
+    blobs = firebase_bucket.list_blobs()
+    blob_dict = {blob.name: blob.public_url for blob in blobs}
+    return blob_dict
+
+
+def download_file_from_storage(file_name, destination_path):
+    """
+    Download a file from Firebase Storage
+    param:
+        file_name: str - The name of the file to be downloaded
+        destination_path: str - The path to save the downloaded file
+    return:
+        bool - True if the file is downloaded successfully, False if the file is not found
+    """
+    try:
+        blob = firebase_bucket.blob(file_name)
+        blob.download_to_filename(destination_path)
+        print("da tai xun thanh cong")
+        return True
+    except Exception as e:
+        print("Error:", e)
+        return False
+def upload_base64_image_to_storage(base64_image, file_name):
+    """
+    Upload a base64 image to Firebase Storage
+    param:
+        base64_image: str - The base64 encoded image
+        file_name: str - The name of the file to be uploaded
+    return:
+        str - The public URL of the uploaded file
+    """
+    try:
+        # Decode the base64 image
+        image_data = base64.b64decode(base64_image)
+        
+        # Convert the decoded image to a JPG file
+        image = Image.open(io.BytesIO(image_data))
+        temp_file_path = f"{tempfile.gettempdir()}/{file_name}.jpg"
+        image.save(temp_file_path, format="JPEG")
+        
+        # Upload the JPG file to Firebase Storage
+        public_url = upload_file_to_storage(temp_file_path, f"{file_name}.jpg")
+        
+        # Remove the temporary file
+        os.remove(temp_file_path)
+        
+        return public_url
+    except Exception as e:
+        print("Error:", e)
+        return None