from django.shortcuts import render

# Create your views here.
from ninja import Router
from django.shortcuts import get_object_or_404
from typing import List

from ninja.pagination import paginate, PageNumberPagination

from .models import NewsSection, News
from .schema import (
    NewsSectionSchema, NewsSectionCreateSchema, NewsSectionUpdateSchema,
    NewsSchema, NewsCreateSchema, NewsUpdateSchema
)

router = Router(tags=['news'])


# ================= 新闻模块接口 =================

@router.get("/news_section/", response=List[NewsSectionSchema])
@paginate(PageNumberPagination, page_size=20)
def list_sections(request):
    return NewsSection.objects.all()


@router.get("/news_section/{section_id}", response=NewsSectionSchema)
def retrieve_section(request, section_id: int):
    return get_object_or_404(NewsSection, id=section_id)


@router.post("/news_section/", response=NewsSectionSchema)
def create_section(request, data: NewsSectionCreateSchema):
    section = NewsSection.objects.create(**data.dict())
    return section


@router.put("/news_section/{section_id}", response=NewsSectionSchema)
def update_section(request, section_id: int, data: NewsSectionUpdateSchema):
    section = get_object_or_404(NewsSection, id=section_id)
    for attr, value in data.dict(exclude_unset=True).items():
        setattr(section, attr, value)
    section.save()
    return section


@router.delete("/news_section/{section_id}", response={"success": bool})
def delete_section(request, section_id: int):
    section = get_object_or_404(NewsSection, id=section_id)
    section.delete()
    return {"success": True}


# ================= 新闻接口 =================

@router.get("/news/", response=List[NewsSchema])
@paginate(PageNumberPagination, page_size=20)
def list_news(request, section_id: int = 0, title: str = ''):
    news = News.objects.filter(title__contains=title)
    if section_id > 0:
        return news.filter(section_id=section_id).all()
    return news


@router.get("/news/{news_id}", response=NewsSchema)
def retrieve_news(request, news_id: int):
    return get_object_or_404(News, id=news_id)


@router.post("/news/", response=NewsSchema)
def create_news(request, data: NewsCreateSchema):
    news = News.objects.create(**data.dict())
    return news


@router.put("/news/{news_id}", response=NewsSchema)
def update_news(request, news_id: int, data: NewsUpdateSchema):
    news = get_object_or_404(News, id=news_id)
    for attr, value in data.dict(exclude_unset=True).items():
        setattr(news, attr, value)
    news.save()
    return news


@router.delete("/news/{news_id}", response={"success": bool})
def delete_news(request, news_id: int):
    news = get_object_or_404(News, id=news_id)
    news.delete()
    return {"success": True}