ruff format app

Author: monicasmith463

backend/app/api/routes/documents.py

@@ -8,11 +8,14 @@
 
 router = APIRouter(prefix="/documents", tags=["documents"])
 
+
 @router.post("/", response_model=DocumentPublic)
 def create_document(
-    *, session: SessionDep, current_user: CurrentUser,
-      background_tasks: BackgroundTasks, # noqa: ARG001
-      file: UploadFile = File(...),
+    *,
+    session: SessionDep,
+    current_user: CurrentUser,
+    background_tasks: BackgroundTasks,  # noqa: ARG001
+    file: UploadFile = File(...),
 ) -> Any:
     key = None
     try:
@@ -32,8 +35,9 @@ def create_document(
         s3_url=url,
     )
 
-    document = Document.model_validate(document_in, update={"owner_id": current_user.id})
-
+    document = Document.model_validate(
+        document_in, update={"owner_id": current_user.id}
+    )
 
     session.add(document)
     session.commit()

backend/app/crud.py

@@ -46,7 +46,9 @@ def authenticate(*, session: Session, email: str, password: str) -> User | None:
     return db_user
 
 
-def create_document(*, session: Session, document_in: DocumentCreate, owner_id: uuid.UUID) -> Document:
+def create_document(
+    *, session: Session, document_in: DocumentCreate, owner_id: uuid.UUID
+) -> Document:
     db_document = Document.model_validate(document_in, update={"owner_id": owner_id})
     session.add(db_document)
     session.commit()

backend/app/models.py

@@ -43,7 +43,9 @@ class UpdatePassword(SQLModel):
 class User(UserBase, table=True):
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
     hashed_password: str
-    documents: list["Document"] = Relationship(back_populates="owner", cascade_delete=True)
+    documents: list["Document"] = Relationship(
+        back_populates="owner", cascade_delete=True
+    )
 
 
 # Properties to return via API, id is always required
@@ -59,16 +61,18 @@ class UsersPublic(SQLModel):
 # Shared properties
 class DocumentBase(SQLModel):
     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
+    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
-
+    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):
     filename: str | None = Field(default=None, min_length=1, max_length=255)  # type: ignore
@@ -88,13 +92,15 @@ class Document(DocumentBase, table=True):
 #     id: uuid.UUID
 #     owner_id: uuid.UUID
 
+
 class DocumentPublic(DocumentBase):
     id: uuid.UUID
     owner_id: uuid.UUID
     filename: str
     content_type: str | None = None
     size: int | None = None
 
+
 class DocumentsPublic(SQLModel):
     data: list[DocumentPublic]
     count: int

backend/app/s3.py

@@ -12,6 +12,7 @@
     region_name=settings.AWS_REGION,
 )
 
+
 def upload_file_to_s3(file: UploadFile, user_id: str) -> str:
     safe_filename = file.filename or ""
     extension = safe_filename.split(".")[-1] if "." in safe_filename else ""

backend/app/tests/api/routes/test_documents.py

@@ -1,4 +1,3 @@
-
 import io
 from unittest.mock import patch
 
@@ -10,15 +9,13 @@
 def skip_test_create_document(
     client: TestClient, superuser_token_headers: dict[str, str]
 ) -> None:
-    '''Test creating a document with a file upload with the real S3 service.'''
+    """Test creating a document with a file upload with the real S3 service."""
     file_content = b"%PDF-1.4 test file content"
 
     response = client.post(
         f"{settings.API_V1_STR}/documents/",
         headers=superuser_token_headers,
-        files={
-            "file": ("example.pdf", io.BytesIO(file_content), "application/pdf")
-        },
+        files={"file": ("example.pdf", io.BytesIO(file_content), "application/pdf")},
     )
 
     assert response.status_code == 200
@@ -29,13 +26,16 @@ def skip_test_create_document(
     # assert "id" in content
     # assert "owner_id" in content
 
+
 def test_create_document(
     client: TestClient, superuser_token_headers: dict[str, str]
 ) -> None:
-    '''Test creating a document with a file upload using mocked S3.'''
+    """Test creating a document with a file upload using mocked S3."""
     file_content = b"%PDF-1.4 test file content"
 
-    with patch("app.api.routes.documents.upload_file_to_s3", return_value="document-slug"):
+    with patch(
+        "app.api.routes.documents.upload_file_to_s3", return_value="document-slug"
+    ):
         response = client.post(
             f"{settings.API_V1_STR}/documents/",
             headers=superuser_token_headers,