✨ Update document model and add S3 upload functionality

Author: monicasmith463

.env

@@ -32,7 +32,7 @@ SMTP_SSL=False
 SMTP_PORT=587
 
 # Postgres
-POSTGRES_SERVER=localhost
+POSTGRES_SERVER=db
 POSTGRES_PORT=5432
 POSTGRES_DB=app
 POSTGRES_USER=postgres

backend/app/models.py

@@ -58,18 +58,20 @@ class UsersPublic(SQLModel):
 
 # Shared properties
 class DocumentBase(SQLModel):
-    title: str = Field(min_length=1, max_length=255)
-    description: str | None = Field(default=None, max_length=255)
+    filename: str = Field(min_length=1, max_length=255)
+    s3_url: str | None = Field(default=None, max_length=255)  # URL to the document in S3
+    content_type: str | None = Field(default=None, max_length=255)
+    size: int | None = Field(default=None, ge=0)  # Size in bytes
+
 
 
 # Properties to receive on document creation
 class DocumentCreate(DocumentBase):
     pass
 
-
 # Properties to receive on document update
 class DocumentUpdate(DocumentBase):
-    title: str | None = Field(default=None, min_length=1, max_length=255)  # type: ignore
+    filename: str | None = Field(default=None, min_length=1, max_length=255)  # type: ignore
 
 
 # Database model, database table inferred from class name
@@ -82,10 +84,17 @@ class Document(DocumentBase, table=True):
 
 
 # Properties to return via API, id is always required
+# class DocumentPublic(DocumentBase):
+#     id: uuid.UUID
+#     owner_id: uuid.UUID
+
 class DocumentPublic(DocumentBase):
     id: uuid.UUID
     owner_id: uuid.UUID
-
+    s3_url: str | None = None
+    filename: str | None = None
+    content_type: str | None = None
+    size: int | None = None
 
 class DocumentsPublic(SQLModel):
     data: list[DocumentPublic]

backend/app/s3.py

@@ -0,0 +1,18 @@
+import boto3
+import uuid
+from fastapi import UploadFile
+from app.core.config import settings
+
+s3 = boto3.client("s3")
+
+def upload_file_to_s3(file: UploadFile, user_id: str) -> str:
+    extension = file.filename.split(".")[-1]
+    key = f"documents/{user_id}/{uuid.uuid4()}.{extension}"
+
+    s3.upload_fileobj(file.file, settings.S3_BUCKET_NAME, key)
+
+    return key
+
+
+def generate_s3_url(key: str) -> str:
+    return f"https://{settings.S3_BUCKET_NAME}.s3.amazonaws.com/{key}"