Create bookimage endpoint

Author: RezuanDzibov

book/models.py

@@ -26,8 +26,8 @@ class Book(UUIDMixin, BaseModel): #type: ignore
     edition = Column(String)
     available = Column(Boolean, default=True)
     release_date = Column(DATE)
-    created = Column(DateTime, default=datetime.now)
-    updated = Column(DateTime, server_default=func.now())
+    created = Column(DateTime, default=datetime.now) # TODO created_at
+    updated = Column(DateTime, server_default=func.now()) # TODO on_update updated_at
     description = Column(String)
     language = Column(String)
     pages = Column(Integer)

image/endpoints.py

@@ -1,18 +1,23 @@
-from email.mime import image
 from fastapi import APIRouter, Depends, BackgroundTasks
+from sqlalchemy.ext.asyncio import AsyncSession
 
-from utils import write_file
-from image.schemas import BookImageCreate
+from core.db import get_session
+from image.schemas import BookImageCreate, BookImageRetrieve
+from image import services as image_services
 
 
 router = APIRouter()
 
 
-@router.post('/create')
+@router.post('/create', response_model=BookImageRetrieve)
 async def create_image(
     background_tasks: BackgroundTasks, 
-    form: BookImageCreate = Depends(BookImageCreate.as_form)
+    form: BookImageCreate = Depends(BookImageCreate.as_form), 
+    session: AsyncSession = Depends(get_session)
 ):
-
-    # background_tasks.add_task(write_file, file)
-    return form.file.filename
+    image = await image_services.insert_image(
+        session=session, 
+        background_tasks=background_tasks, 
+        form=form
+    )
+    return image

image/models.py

@@ -17,11 +17,10 @@
 class ImageMixin:
     title = Column(String(length=255))
     available = Column(Boolean, default=True)
-    created = Column(DateTime, default=datetime.now)
+    created = Column(DateTime, default=datetime.now) # TODO created_at
     image_path = Column(String)
 
 
-
 class BookImage(UUIDMixin, ImageMixin, BaseModel):   # type: ignore
     __tablename__ = 'book_image'
 

image/schemas.py

@@ -1,5 +1,7 @@
+from datetime import datetime
+
 from fastapi import UploadFile, Form
-from pydantic import BaseModel, UUID4, constr
+from pydantic import BaseModel, UUID4, constr, validator
 
 
 class BookImageCreate(BaseModel):
@@ -16,4 +18,18 @@ def as_form(
         available: bool = Form(...),
         file: UploadFile = Form(...)
     ):
-        return cls(book_id=book_id, title=title, available=available, file=file)    
+        return cls(book_id=book_id, title=title, available=available, file=file)
+
+
+class BookImageRetrieve(BaseModel):
+    id: UUID4
+    book_id: UUID4
+    title: str
+    available: bool
+    created: datetime
+    
+    class Config:
+        orm_mode = True
+        json_encoders = {
+            datetime: lambda v: v.strftime("%Y-%m-%d %H:%M:%S")
+        }

image/services.py

@@ -0,0 +1,32 @@
+from fastapi import Depends, BackgroundTasks
+from sqlalchemy import insert
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from image.schemas import BookImageCreate
+from image.models import BookImage
+from utils import write_file
+
+
+async def insert_image(
+    session: AsyncSession,
+    background_tasks: BackgroundTasks, 
+    form: BookImageCreate = Depends(BookImageCreate.as_form)
+):
+    file = form.file
+    filepath = f"media/{file.filename}"
+    form_data = form.dict()
+    form_data['book_id'] = form_data['book_id'].hex
+    form_data['image_path'] = filepath
+    del form_data['file']
+    statement = insert(BookImage).values(form_data).returning(
+        BookImage.id, 
+        BookImage.book_id, 
+        BookImage.title,
+        BookImage.available, 
+        BookImage.created
+    )
+    result = await session.execute(statement)
+    await session.commit()
+    image = result.one()
+    background_tasks.add_task(write_file, filepath, file)
+    return image

main.py

@@ -1,13 +1,22 @@
+import os
+from pathlib import Path
+
 from fastapi import FastAPI
 from fastapi_pagination import add_pagination
 
 from routes import routes
 from core.db import database
+from config import BASE_DIR
 
 
 app = FastAPI(title='Library API')
 app.include_router(routes)
 
+MEDIA = BASE_DIR.joinpath('media')
+
+if not os.path.isdir(MEDIA):
+    Path.mkdir(MEDIA)
+
 add_pagination(app)
 
 

utils.py

@@ -12,6 +12,6 @@ def extract_objects(objects: tuple):
         yield object_tuple[0]
 
 
-def write_file(file: UploadFile = File(...)):
-    with open(f'{file.filename}', 'wb') as buffer:
+def write_file(filepath, file: UploadFile = File(...)):
+    with open(filepath, 'wb') as buffer:
         shutil.copyfileobj(file.file, buffer)