from django.shortcuts import render, redirect
from app01 import models
from app01.utils.pagination import Pagination


def depart_list(request):
    """部门列表"""

    # 去数据库获取所有的部门列表

    queryset = models.Department.objects.all()
    print("-------")
    for i in queryset:
        print(i)
    print("----------")
    page_object = Pagination(request, queryset, page_size=5)
    context = {
        'queryset': page_object.page_queryset,
        'page_string': page_object.html()
    }

    return render(request, 'depart_list.html', context)


def depart_add(request):
    """添加部门"""
    if request.method == "GET":
        return render(request, 'depart_add.html')

    # 获取用户post提交的数据
    title = request.POST.get("title")

    # 保留到数据库
    models.Department.objects.create(title=title)

    # 重定向回列表
    return redirect("/depart/list/")


def depart_delete(request, nid):
    """删除部门"""
    # 获取id
    nid = request.GET.get('nid')
    # 删除
    models.Department.objects.filter(id=nid).delete()

    # 重定向列表
    return redirect("/depart/list/")


def depart_edit(request, nid):
    if request.method == "GET":
        row_object = models.Department.objects.filter(id=nid).first()
        # print(row_object.id, row_object.title)
        return render(request, 'depart_edit.html', {'row_object': row_object})

    # 根据id找到数据库的数据进行修改
    title = request.POST.get("title")
    models.Department.objects.filter(id=nid).update(title=title)

    return redirect("/depart/list/")