from app.model.bussiness_model import ProjectImage, ProjectInfo, ProjectImgLeafer, ProjectImgLabel
from app.model.schemas.project_info_schemas import ProjectInfoIn, ProjectInfoOut
from app.model.schemas.project_image_schemas import ProjectImgLeaferLabel, ProjectImgLeaferOut
from app.model.crud import project_info_crud as pic
from app.model.crud import project_image_crud as pimc
from app.model.crud import project_img_leafer_label_crud as pillc
from app.util import os_utils as os
from app.util import random_utils as ru
from app.config.config_reader import datasets_url, runs_url, images_url

from sqlalchemy.orm import Session
from typing import List
from fastapi import UploadFile


def add_project(info: ProjectInfoIn, session: Session, user_id: int):
    """
    新建项目，完善数据，并创建对应的文件夹
    :param info: 项目信息
    :param session: 数据库session
    :param user_id: 用户id
    :return:
    """
    project_info = ProjectInfo(**info.dict())
    project_info.user_id = user_id
    project_info.project_no = ru.random_str(6)
    project_info.project_status = "0"
    project_info.train_version = 0
    os.create_folder(datasets_url, project_info.project_no)
    os.create_folder(runs_url, project_info.project_no)
    pic.add_project(project_info, session)
    return project_info


def upload_project_image(project_info: ProjectInfoOut, files: List[UploadFile], session: Session):
    """
    上传项目的图片
    :param files: 上传的图片
    :param project_info: 项目信息
    :param session:
    :return:
    """
    images = []
    for file in files:
        image = ProjectImage()
        image.project_id = project_info.id
        # 保存原图
        path = os.save_images(images_url, project_info.project_no, file=file)
        image.image_url = path
        # 生成缩略图
        thumb_image_url = images_url + "\\thumb\\" + project_info.project_no + "\\" + ru.random_str(10) + ".jpg"
        os.create_thumbnail(path, thumb_image_url)
        image.thumb_image_url = thumb_image_url
        images.append(image)
    pimc.add_image_batch(images, session)


def save_img_label(img_leafer_label: ProjectImgLeaferLabel, session: Session):
    """
    保存图片的标签框选信息，每次保存都会针对图片的信息全部删除，然后重新保存
    :param img_leafer_label:
    :param session:
    :return:
    """
    img_leafer = ProjectImgLeafer(**img_leafer_label)
    pillc.save_img_leafer(img_leafer, session)
    label_infos = img_leafer_label.label_infos
    img_labels = []
    for label_info in label_infos:
        img_label = ProjectImgLabel(**label_info.dict())
        img_label.image_id = img_leafer_label.image_id
        img_labels.append(img_label)
    pillc.save_img_label_batch(img_leafer_label.image_id, img_labels, session)


def get_img_leafer(image_id: int, session: Session):
    """
    根据图片id查询图片的leafer信息
    :param image_id:
    :param session:
    :return:
    """
    img_leafer = pillc.get_img_leafer(image_id, session)
    img_leafer_out = ProjectImgLeaferOut.from_orm(img_leafer).dict()
    return img_leafer_out