91 lines
3.2 KiB
Python
91 lines
3.2 KiB
Python
import os
|
|
from typing import List
|
|
from urllib.parse import unquote, urlparse
|
|
import uuid
|
|
|
|
from sqlmodel import delete
|
|
from api.models import LogMetadata
|
|
from api.routers.presentation.models import (
|
|
PresentationUpdateRequest,
|
|
PresentationAndSlides,
|
|
)
|
|
from api.services.logging import LoggingService
|
|
from api.sql_models import PresentationSqlModel, SlideSqlModel
|
|
from api.utils import (
|
|
download_files,
|
|
get_presentation_dir,
|
|
get_presentation_images_dir,
|
|
replace_file_name,
|
|
)
|
|
from api.services.database import get_sql_session
|
|
from api.services.instances import temp_file_service
|
|
|
|
|
|
class UpdateSlideModelsHandler:
|
|
|
|
def __init__(self, data: PresentationUpdateRequest):
|
|
self.data = data
|
|
self.presentation_id = data.presentation_id
|
|
self.session = str(uuid.uuid4())
|
|
self.temp_dir = temp_file_service.create_temp_dir()
|
|
|
|
self.presentation_dir = get_presentation_dir(self.presentation_id)
|
|
|
|
def __del__(self):
|
|
temp_file_service.cleanup_temp_dir(self.temp_dir)
|
|
|
|
async def post(self, logging_service: LoggingService, log_metadata: LogMetadata):
|
|
logging_service.logger.info(
|
|
logging_service.message(self.data.model_dump(mode="json")),
|
|
extra=log_metadata.model_dump(),
|
|
)
|
|
|
|
presentation_id = self.data.presentation_id
|
|
new_slides = self.data.slides
|
|
|
|
images_dir = get_presentation_images_dir(self.presentation_id)
|
|
|
|
# Handle images
|
|
images_local_paths = []
|
|
images_download_links = []
|
|
for new_slide in new_slides:
|
|
new_images = new_slide.images or []
|
|
|
|
for i, image in enumerate(new_images):
|
|
if image.startswith("http"):
|
|
parsed_url = unquote(urlparse(image).path)
|
|
image_name = replace_file_name(
|
|
os.path.basename(parsed_url), str(uuid.uuid4())
|
|
)
|
|
image_path = os.path.join(images_dir, image_name)
|
|
images_local_paths.append(image_path)
|
|
images_download_links.append(image)
|
|
getattr(new_slide, "images")[i] = image_path
|
|
|
|
if images_download_links:
|
|
await download_files(images_download_links, images_local_paths)
|
|
|
|
with get_sql_session() as sql_session:
|
|
slide_sql_models = [
|
|
SlideSqlModel(**each.model_dump(mode="json")) for each in new_slides
|
|
]
|
|
to_update_slides_ids = [each.id for each in slide_sql_models]
|
|
sql_session.exec(
|
|
delete(SlideSqlModel).where(SlideSqlModel.id.in_(to_update_slides_ids))
|
|
)
|
|
sql_session.add_all(slide_sql_models)
|
|
sql_session.commit()
|
|
for each in slide_sql_models:
|
|
sql_session.refresh(each)
|
|
presentation = sql_session.get(PresentationSqlModel, presentation_id)
|
|
|
|
response = PresentationAndSlides(
|
|
presentation=presentation, slides=slide_sql_models
|
|
)
|
|
response = response.to_response_dict()
|
|
|
|
logging_service.logger.info(
|
|
logging_service.message(response),
|
|
extra=log_metadata.model_dump(),
|
|
)
|
|
return response
|