run ruff lint fixes

Author: monicasmith463

backend/app/api/routes/documents.py

@@ -1,33 +1,34 @@
 from typing import Any
+
+from fastapi import APIRouter, BackgroundTasks, File, HTTPException, UploadFile
+
 from app.api.deps import CurrentUser, SessionDep
 from app.models import Document, DocumentCreate, DocumentPublic
-from fastapi import APIRouter, BackgroundTasks, File, UploadFile, HTTPException
-from app.s3 import upload_file_to_s3, generate_s3_url
+from app.s3 import generate_s3_url, upload_file_to_s3
 
 router = APIRouter(prefix="/documents", tags=["documents"])
-    
+
 @router.post("/", response_model=DocumentPublic)
 def create_document(
     *, session: SessionDep, current_user: CurrentUser,
-      background_tasks: BackgroundTasks,
-      file: UploadFile = File(...), 
+      background_tasks: BackgroundTasks, # noqa: ARG001
+      file: UploadFile = File(...),
 ) -> Any:
     key = None
     try:
-        user_id = current_user.id
         key = upload_file_to_s3(file, str(current_user.id))
     except Exception as e:
         raise HTTPException(500, f"Failed to upload file. Error: {str(e)}")
 
     try:
         url = generate_s3_url(key)
-    except Exception as e:
+    except Exception:
         raise HTTPException(500, f"Could not generate URL for file key: {key}")
 
     document_in = DocumentCreate(
         filename=file.filename,
         content_type=file.content_type,
-        size=file.size,  
+        size=file.size,
         s3_url=url,
     )
 
@@ -42,4 +43,4 @@ def create_document(
     print("Document created, starting background task...")
     # background_tasks.add_task(generate_questions, document.id)
 
-    return document
+    return document

backend/app/core/db.py

@@ -16,7 +16,6 @@ def init_db(session: Session) -> None:
     # Tables should be created with Alembic migrations
     # But if you don't want to use migrations, create
     # the tables un-commenting the next lines
-    from sqlmodel import SQLModel
 
     # This works because the models are already imported and registered from app.models
     SQLModel.metadata.create_all(engine)

backend/app/models.py

@@ -118,4 +118,4 @@ class TokenPayload(SQLModel):
 
 class NewPassword(SQLModel):
     token: str
-    new_password: str = Field(min_length=8, max_length=40)
+    new_password: str = Field(min_length=8, max_length=40)

backend/app/s3.py

@@ -1,6 +1,8 @@
-import boto3
 import uuid
+
+import boto3
 from fastapi import UploadFile
+
 from app.core.config import settings
 
 s3 = boto3.client(
@@ -25,4 +27,4 @@ def upload_file_to_s3(file: UploadFile, user_id: str) -> str:
 
 
 def generate_s3_url(key: str) -> str:
-    return f"https://{settings.S3_BUCKET_NAME}.s3.amazonaws.com/{key}"
+    return f"https://{settings.S3_BUCKET_NAME}.s3.amazonaws.com/{key}"

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

@@ -1,11 +1,10 @@
-import uuid
 
-from fastapi.testclient import TestClient
-from sqlmodel import Session
+import io
 from unittest.mock import patch
+
+from fastapi.testclient import TestClient
+
 from app.core.config import settings
-from app.tests.utils.document import create_random_document
-import io
 
 
 def skip_test_create_document(
@@ -45,7 +44,7 @@ def test_create_document(
             },
         )
 
-    assert response.status_code == 200, f"Unexpected response status code"
+    assert response.status_code == 200, "Unexpected response status code"
     content = response.json()
     assert "id" in content, "actual response: " + str(content)
     assert "document-slug" in content["s3_url"], "S3 URL should match mocked value"