Upload 155 files

src/apis/config/firebase_config.py

@@ -7,13 +7,13 @@ from dotenv import load_dotenv
 
 # Load environment variables
 load_dotenv()
-firebase_url_storageBucket = os.getenv("FIREBASE_URL_STORAGEBUCKET")
+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("FIREBASE_PRIVATE_KEY_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"),

src/apis/controllers/post_controller.py

@@ -8,6 +8,9 @@ from datetime import datetime
 from bson import ObjectId
 from src.utils.mongo import UserCRUD
 from asyncio import gather
+from src.utils.helper import call_external_api
+from src.utils.logger import get_date_time
+import base64
 
 
 def serialize_datetime(obj):
@@ -19,14 +22,28 @@ def serialize_datetime(obj):
 
 
 async def create_a_post_controller(
-    content: str, user_id: str, destination_id: str
+    content: str, user_id: str, destination_id: str, images: list
 ) -> Dict:
     try:
+        timestamp = get_date_time().replace(tzinfo=None).strftime("%Y-%m-%d %H:%M:%S")
+        encoded_timestamp = (
+            base64.urlsafe_b64encode(timestamp.encode()).decode().rstrip("=")
+        )
+
+        image_url = await call_external_api(
+            method="POST",
+            # url="http://localhost:8000/upload_image",
+            url="https://abao77-image-retrieval.hf.space/upload_image",
+            json={"base64_image": images[0], "image_name": f"_{encoded_timestamp}"},
+            
+        )
+        
         post = {
             "content": content,
             "user_id": user_id,
             "destination_id": destination_id,
             "comment_ids": [],
+            "picture": [image_url["public_url"]],
             "like": [],
         }
         await PostCRUD.create(post)
@@ -79,7 +96,7 @@ async def list_all_posts_controller():
                 "destination_id": post.get("destination_id"),
                 "comment_ids": post.get("comment_ids", []),
                 "like": post.get("like", []),
-                "picture": [],
+                "picture": post.get("picture", []),
                 "created_at": serialize_datetime(post.get("created_at")),
                 "updated_at": serialize_datetime(post.get("updated_at")),
                 "user_info": (

src/apis/models/post_models.py

@@ -40,6 +40,7 @@ class Post(BaseDocument):
     destination_id: str = Field("", description="Destination's id")
     comment_ids: list[str] = Field([], description="Comment's id")
     like: list[str] = Field([], description="User's id who like this post")
+    picture: list[str] = Field([], description="Picture's url")
     model_config = {
         "json_schema_extra": {
             "example": {
@@ -47,6 +48,8 @@ class Post(BaseDocument):
                 "user_id": "1234567890",
                 "destination_id": "1234567890",
                 "comment_ids": ["1234567890"],
+                "like": ["1234567890"],
+                "picture": ["https://example.com/picture.jpg"],
             }
         }
     }

src/apis/providers/firebase_provider.py

@@ -1,4 +1,7 @@
 from src.apis.config.firebase_config import firebase_bucket
+import base64
+import os
+import tempfile
 
 
 def upload_file_to_storage(file_path, file_name):
@@ -9,7 +12,6 @@ def upload_file_to_storage(file_path, file_name):
     return:
         str - The public URL of the uploaded file
     """
-    file_name = file_name
     blob = firebase_bucket.blob(file_name)
     blob.upload_from_filename(file_path)
     blob.make_public()
@@ -17,6 +19,7 @@ def upload_file_to_storage(file_path, file_name):
     return blob.public_url
 
 
+
 def delete_file_from_storage(file_name):
     """
     Delete a file from Firebase Storage
@@ -62,3 +65,33 @@ def download_file_from_storage(file_name, destination_path):
     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)
+
+        # Create a temporary file to save the decoded image
+        with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+            temp_file.write(image_data)
+            temp_file_path = temp_file.name
+
+        # Upload the temporary file to Firebase Storage
+        public_url = upload_file_to_storage(temp_file_path, file_name)
+
+        # Remove the temporary file
+        os.remove(temp_file_path)
+
+        return public_url
+    except Exception as e:
+        print("Error:", e)
+        return None

src/apis/routes/auth_route.py

@@ -35,3 +35,12 @@ async def get_user_info(user: user_dependency):
     if user is None:
         return JSONResponse(content={"message": "User not found"}, status_code=401)
     return JSONResponse(content={"user": user}, status_code=200)
+
+import time
+import asyncio
+@router.get("/test", status_code=status.HTTP_200_OK)
+async def test():
+    print("da vao")
+    await asyncio.sleep(5)  # Use asyncio.sleep for non-blocking sleep
+    print("alo")
+    return "test"

src/apis/routes/post_router.py

@@ -23,12 +23,14 @@ user_dependency = Annotated[User, Depends(get_current_user)]
 class BodyPost(BaseDocument):
     content: str = Field("", description="Post's content")
     destination_id: str = Field("", description="Destination's id")
+    images_base64: Optional[list] = Field([], description="Images base64")
 
     class Config:
         json_schema_extra = {
             "example": {
                 "content": "John Doe",
                 "destination_id": "1234567890",
+                "images_base64": ["base64_image1", "base64_image2"],
             }
         }
 
@@ -42,6 +44,7 @@ async def create_post(body: BodyPost, user: user_dependency):
         body.content,
         user_id,
         body.destination_id,
+        body.images_base64,
     )
     logger.info(f"RESULT: {result}")
     if result["status"] == "error":

src/utils/helper.py

@@ -6,8 +6,9 @@ from datetime import datetime, timezone
 import re
 from src.utils.logger import logger
 from pydantic import BaseModel, Field
-from typing import List, Optional, Union
-
+from typing import List, Union
+import httpx
+from fastapi import HTTPException
 
 def handle_validator_raise(func):
     """
@@ -252,3 +253,31 @@ def format_include_destinations(include_destinations: List[Union[Destination, st
     else:
         for index, destination in enumerate(include_destinations):
             formatted_string += f"#Destination {int(index) + 1}: {destination}\n"
+
+
+
+async def call_external_api(
+    method: str, 
+    url: str, 
+    headers: dict = None, 
+    params: dict = None, 
+    data: dict = None, 
+    json: dict = None, 
+    timeout: int = 10
+):
+    async with httpx.AsyncClient(timeout=timeout) as client:
+        try:
+            response = await client.request(
+                method=method,
+                url=url,
+                headers=headers,
+                params=params,
+                data=data,
+                json=json
+            )
+            response.raise_for_status()  # Raise an error for non-2xx/3xx responses
+            return response.json()
+        except httpx.HTTPStatusError as e:
+            raise HTTPException(status_code=e.response.status_code, detail=str(e))
+        except httpx.RequestError as e:
+            raise HTTPException(status_code=500, detail=f"Request failed: {str(e)}")