# from application.ninja_cof import api
# Author 臧成龙
# coding=utf-8
# @Time    : 2022/5/15 21:47
# @File    : dept.py
# @Software: PyCharm
# @qq: 939589097

from typing import List

from django.shortcuts import get_object_or_404
from ninja import Field, ModelSchema, Query, Router, Schema
from ninja.pagination import paginate
from system.models import Dept
from utils.fu_crud import create, delete, retrieve, update
from utils.fu_ninja import FuFilters, MyPagination
from utils.fu_response import FuResponse
from utils.list_to_tree import list_to_route, list_to_tree

router = Router()


class Filters(FuFilters):
    name: str = Field(None, alias="name")
    status: bool = Field(None, alias="status")
    id: str = Field(None, alias="id")


class SchemaIn(ModelSchema):
    parent_id: int = None

    class Config:
        model = Dept
        model_exclude = ['id', 'parent', 'create_datetime', 'update_datetime']


class SchemaOut(ModelSchema):
    class Config:
        model = Dept
        model_fields = "__all__"
    # model_fields = []


@router.post("/dept", response=SchemaOut)
def create_dept(request, data: SchemaIn):
    dept = create(request, data, Dept)
    return dept


@router.delete("/dept/{dept_id}")
def delete_dept(request, dept_id: int):
    delete(dept_id, Dept)
    return {"success": True}


@router.put("/dept/{dept_id}", response=SchemaOut)
def update_dept(request, dept_id: int, data: SchemaIn):
    dept = update(request, dept_id, data, Dept)
    return dept


@router.get("/dept", response=List[SchemaOut])
@paginate(MyPagination)
def list_dept(request, filters: Filters = Query(...)):
    qs = retrieve(request, Dept, filters)
    return qs


@router.get("/dept/{dept_id}", response=SchemaOut)
def get_dept(request, dept_id: int):
    dept = get_object_or_404(Dept, id=dept_id)
    return dept


@router.get("/dept/list/tree")
def list_dept_tree(request, filters: Filters = Query(...)):
    qs = retrieve(request, Dept, filters).values()
    dept_tree = list_to_tree(list(qs))
    return FuResponse(data=dept_tree)