# from application.ninja_cof import api
# Author 臧成龙
# coding=utf-8
# @Time    : 2022/5/29 01:36
# @File    : user.py
# @Software: PyCharm
# @qq: 939589097

from typing import List

from django.contrib.auth.hashers import make_password
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 Users
from utils.fu_crud import create, delete, retrieve
from utils.fu_ninja import FuFilters, MyPagination
from utils.fu_response import FuResponse
from utils.usual import get_user_info_from_token

router = Router()


class Filters(FuFilters):
    name: str = Field(None, alias="name")
    mobile: str = Field(None, alias="mobile")
    status: bool = Field(None, alias="status")
    dept_id__in: list = Field(None, alias="dept_ids[]")
    id: int = Field(None, alias="id")


class SchemaIn(ModelSchema):
    dept_id: int = Field(None, alias="dept")
    post: list = []
    role: list = []

    class Config:
        model = Users
        model_exclude = ['id', 'groups', 'user_permissions', 'is_superuser', 'dept', 'post', 'role', 'password',
                         'create_datetime', 'update_datetime']


class SchemaOut(ModelSchema):
    class Config:
        model = Users
        model_exclude = ['password']


@router.post("/user", response=SchemaOut)
def create_user(request, data: SchemaIn):
    data_dic = data.dict()
    data_dic['password'] = make_password('123456')
    post = data_dic.pop('post')
    role = data_dic.pop('role')
    user = create(request, data_dic, Users)
    # 为多对多关系添加数据
    user.post.set(post)
    user.role.set(role)
    return user


@router.delete("/user/{user_id}")
def delete_user(request, user_id: int):
    delete(user_id, Users)
    return {"success": True}


@router.put("/user/{user_id}", response=SchemaOut)
def update_user(request, user_id: int, payload: SchemaIn):
    user = get_object_or_404(Users, id=user_id)
    for attr, value in payload.dict().items():
        if attr == 'role':
            user.role.set(value)
        elif attr == 'post':
            user.post.set(value)
        else:
            setattr(user, attr, value)
    user.save()
    return user


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


@router.get("/user/all/list", response=List[SchemaOut])
def all_list_user(request):
    qs = retrieve(request, Users)
    return qs


@router.get("/user/{user_id}", response=SchemaOut)
def get_user(request, user_id: int):
    user = get_object_or_404(Users, id=user_id)
    return user


class SchemaIn(Schema):
    id: int
    password: str


@router.post("/user/set/repassword", response=SchemaOut)
def repassword(request, data: SchemaIn):
    request_user = get_user_info_from_token(request)
    request_user_id = request_user['id']
    update_id = data.id
    if request_user_id == update_id:
        user = get_object_or_404(Users, id=update_id)
        user.set_password(data.password)
        user.save()
        return FuResponse(msg='密码修改成功')
    else:
        return FuResponse(code=403, msg='对不起, 只能修改自己的密码')


@router.put("/user/reset/password/{id}", response=SchemaOut)
def reset_password(request, id:int):
    user = get_object_or_404(Users, id=id)
    user.set_password('123456')
    user.save()
    return FuResponse(msg='密码重置成功')