# -*- coding: utf-8 -*-
# @Time    : 2023/8/8 16:40
# @Author  : tx
# @File    : msds.py
# @Description :

import datetime
import os
import subprocess

from fastapi import APIRouter, Depends, File, Request, UploadFile
from starlette.responses import FileResponse
from tortoise.queryset import QuerySet, Q
from fastapi.encoders import jsonable_encoder

from helper import login_required, respond_to
from models import Msds

router = APIRouter(prefix='/msds', dependencies=[Depends(login_required)])


@router.get('', summary='msds数据库')
async def index(keyword: str = '', page_no: int = 1, page_size: int = 10):
    """
    获取msds数据库
    :param keyword: 按药剂信息搜索
    :param page_no: 分页页码，默认为1
    :param page_size: 分页大小，默认为10
    :return:
    """
    offset = (page_no - 1) * page_size
    query = QuerySet(Msds)
    if keyword:
        query = query.filter(Q(name__contains=keyword) | Q(english_name__contains=keyword))

    count = await query.count()
    msds_objs = await query.limit(page_size).offset(offset)
    result_list = jsonable_encoder(msds_objs)

    return respond_to(data={'data': result_list, 'count': count})


@router.post('/import', summary='新增msds/上传msds附件')
async def create(request: Request, drug_name: str, cas_number: str, file: UploadFile = File(...)):
    """
    :param request: 请求
    :param id: 药剂模板ID
    :param file: 模板文件
    """
    if not (file.filename and file.filename.rsplit('.', 1)[1].lower() in ['pdf']):
        return respond_to(code=400, desc='请上传正确的文件格式')
    current_user = request.state.current_user

    try:
        current_month = datetime.datetime.now().strftime("%Y-%m")
        save_folder = os.path.join("/static/msds_file", current_month)
        os.makedirs(save_folder, exist_ok=True)
        save_path = os.path.join(save_folder, file.filename)
        if os.path.exists(save_path):
            return respond_to(code=400, desc='同名附件已存在，上传失败')
        with open(save_path, "wb") as buffer:
            buffer.write(await file.read())
    except Exception as e:
        print("附件上传有误e", e)
        return respond_to(code=400, desc='附件上传有误')
    msds = {
        "upload_user_id": current_user.id,
        "upload_at": datetime.datetime.now(),
        "path": os.path.abspath(save_folder),
        "file_name": file.filename,
        "name": drug_name,
        "cas_number": cas_number
    }
    data = await Msds.create(**msds)

    return respond_to(data=data)


@router.get("/pdf/{filename:path}", summary='附件查看')
async def read_pdf(filename: str):
    # 调用函数获取并删除数据
    file_path = os.getcwd()+f"/static/{filename}"
    try:
        if os.name == 'posix':
            # Linux系统
            os.system(f"xdg-open {file_path}")
        elif os.name == 'nt':
            # Windows系统
            os.startfile(file_path)
        else:
            print("不支持的操作系统")
    except Exception as e:
        print(f"打开文件失败: {e}")