Change image API

Author: Tinzzs

backend/controller/plant_controller.py

@@ -1,10 +1,11 @@
-from fastapi import APIRouter, Depends, HTTPException
+from fastapi import APIRouter, Depends, HTTPException, File, UploadFile
 from schemas.plant_schema import PlantSchema
 from schemas.user_schema import UserSchema
 from services.plant_service import get_service, PlantService
 from fastapi import  Depends
 from fastapi.responses import JSONResponse
 from config.authentication import get_current_user
+from pathlib import Path
 
 
 create_plant = APIRouter()
@@ -48,4 +49,36 @@ def get_plant_data(
     except HTTPException as he:
         raise he
     except Exception as e:
-        return JSONResponse(status_code=500, content={"status": "error", "error": f"Unexpected error: {str(e)}"})
+        JSONResponse(status_code=500, content={"status": "error", "error": f"Unexpected error: {str(e)}"})
+
+
+# Define allowed file types and max file size (in bytes)
+ALLOWED_FILE_TYPES = {"image/jpeg", "image/png"}
+MAX_FILE_SIZE = 5 * 1024 * 1024  # 5 MB
+
+@create_plant.put("/api/plant/data/{plant_id}", response_model=PlantSchema)
+async def update_plant_image(
+    plant_id: int,
+    file: UploadFile = File(...),
+    service: PlantService = Depends(get_service)
+):
+    # Validate file type
+    if file.content_type not in ALLOWED_FILE_TYPES:
+        raise HTTPException(status_code=400, detail="Invalid file type. Only JPEG and PNG are allowed.")
+
+    # Validate file size
+    file_content = await file.read()
+    if len(file_content) > MAX_FILE_SIZE:
+        raise HTTPException(status_code=400, detail="File size exceeds the maximum limit of 5 MB.")
+    
+    # Reset file read position
+    await file.seek(0)
+
+    # Update the plant entry with the new image filename and content
+    response = service.update_plant_image(plant_id, file.filename, file_content)
+
+    if "error" in response:
+        status_code = 400 if "Not Found" in response["error"] else 500
+        return JSONResponse(status_code=status_code, content={"status": "error", "error": response["error"]})
+
+    return response

backend/dal/plant_dal.py

@@ -2,6 +2,11 @@
 from schemas.plant_schema import PlantSchema
 import psycopg2 
 from psycopg2 import  DatabaseError, IntegrityError
+from pathlib import Path
+import os
+from fastapi.responses import FileResponse
+
+
 
 class PlantDAL:
     def __init__(self):
@@ -16,9 +21,9 @@ def create_plant(self, plant: PlantSchema):
 
             # Execute the query to insert the plant data
             self.cursor.execute("""
-                INSERT INTO plant (PlantID, PlantName, ScientificName, Threshold)
-                VALUES (%s, %s, %s, %s) RETURNING PlantID;
-            """, (plant.PlantID, plant.PlantName, plant.ScientificName, plant.Threshold))
+                INSERT INTO plant (PlantID, PlantName, ScientificName, Threshold, ImageFileName)
+                VALUES (%s, %s, %s, %s, %s) RETURNING PlantID;
+            """, (plant.PlantID, plant.PlantName, plant.ScientificName, plant.Threshold, plant.ImageFilename))
 
             # Commit the transaction
             self.conn.commit()
@@ -32,7 +37,8 @@ def create_plant(self, plant: PlantSchema):
                 "PlantID": plant_id,
                 "PlantName": plant.PlantName,
                 "ScientificName": plant.ScientificName,
-                "Threshhold": plant.Threshhold
+                "Threshhold": plant.Threshhold,
+                "ImageFilename": plant.ImageFilename
             }
 
         except IntegrityError as e:
@@ -81,7 +87,7 @@ def create_plant(self, plant: PlantSchema):
     def get_plants(self):
         try:
 
-            self.cursor.execute( "SELECT PlantID, PlantName, ScientificName, Threshold FROM plant;")
+            self.cursor.execute( "SELECT PlantID, PlantName, ScientificName, Threshold FROM plant;") # + ImageFilename maybe?
 
             plants= self.cursor.fetchall()
 
@@ -98,6 +104,7 @@ def get_plants(self):
                 "PlantName": plant[1],
                 "ScientificName": plant[2],
                 "Threshhold": plant[3]
+                #"ImageFilename" plant[4]
             }
                 for plant in plants
             ]
@@ -127,6 +134,72 @@ def get_plants(self):
                 "error": error_message
             }
 
+        finally:
+            # Ensure that the connection is released
+            release_connection(self.conn)
+
+
+    def update_plant_image(self, plant_id: int, new_image_filename: str, file_content: bytes):
+        try:
+            # Query for the current image filename
+            self.cursor.execute("SELECT ImageFilename FROM plant WHERE PlantID = %s;", (plant_id,))
+            result = self.cursor.fetchone()
+            if not result:
+                return {"status": "error", "error": "Plant not found"}
+
+            current_image_filename = result[0]
+
+            # Define the image directory
+            IMAGE_DIR = Path("/path/to/your/image/directory")
+
+            # Delete the old image file if it exists
+            if current_image_filename:
+                old_image_path = IMAGE_DIR / current_image_filename
+                if old_image_path.exists():
+                    os.remove(old_image_path)
+
+            # Save the new image file
+            new_image_path = IMAGE_DIR / new_image_filename
+            with open(new_image_path, "wb") as buffer:
+                buffer.write(file_content)
+
+            # Update the image filename in the database
+            self.cursor.execute("""
+                UPDATE plant
+                SET ImageFilename = %s
+                WHERE PlantID = %s;
+            """, (new_image_filename, plant_id))
+
+            # Commit the transaction
+            self.conn.commit()
+
+            # Return the updated image
+            return FileResponse(
+            path=str(new_image_path),            
+            filename=new_image_filename
+        )
+
+
+        except (psycopg2.Error, DatabaseError) as db_error:
+            # Handle database errors
+            self.conn.rollback()  # Rollback transaction on error
+            error_message = f"Database error: {db_error}"
+            print(f"Database error: {db_error}")
+            return {
+                "status": "error",
+                "error": error_message
+            }
+
+        except Exception as e:
+            # Catch any other unexpected errors
+            self.conn.rollback()  # Rollback transaction on error
+            error_message = f"Unexpected error: {e}"
+            print(f"Unexpected error: {e}")
+            return {
+                "status": "error",
+                "error": error_message
+            }
+
         finally:
             # Ensure that the connection is released
             release_connection(self.conn)

backend/models/models.py

@@ -7,6 +7,7 @@ class Plant(Base):
     PlantName = Column(String(50), nullable=False)
     ScientificName = Column(String(50), nullable=False)
     Threshhold = Column(Float, nullable=False)
+    ImageFilename = Column(String(255), nullable=True)  # New column for image filename
 
 class Sensor(Base):
     __tablename__ = "sensors"

backend/repository/plant_repository.py

@@ -9,4 +9,7 @@ def add_plant(self, plant: PlantSchema):
         return self.dal.create_plant(plant)
 
     def get_plants(self):
-        return self.dal.get_plants()
+        return self.dal.get_plants()
+
+    def update_plant_image(self, plant_id: int, new_image_filename: str, file_content: bytes):
+        return self.dal.update_plant_image(plant_id, new_image_filename)

backend/schemas/plant_schema.py

@@ -1,7 +1,9 @@
 from pydantic import BaseModel
+from typing import Optional
 
 class PlantSchema(BaseModel):
     PlantID: int
     PlantName: str
     ScientificName: str
-    Threshold: float
+    Threshold: float
+    ImageFilename: Optional[str] = None  # Optional field for image filename

backend/services/plant_service.py

@@ -13,6 +13,9 @@ def create_plant(self, plant: PlantSchema):
     def get_plants(self):
         return self.repository.get_plants()
 
+    def update_plant_image(self, plant_id: int, new_image_filename: str, file_content: bytes):
+        # Pass the file content to the repository
+        return self.repository.update_plant_image(plant_id, new_image_filename, file_content)
 
 def get_service():
     dal = PlantDAL()