day2作业笔记

模型models/articles.pyfrom tortoise import Model,fields class Category(Model): idfields.IntField(pkTrue,auto_incrementTrue,description分类ID) namefields.CharField(max_length50,description分类名称) class Meta: tabledb_category class Author(Model): idfields.IntField(pkTrue,auto_incrementTrue,description作者ID) namefields.CharField(max_length50,description作者名称) class Meta: tabledb_author class Article(Model): idfields.IntField(pkTrue,auto_incrementTrue,description文章ID) titlefields.CharField(max_length200,description文章标题) contentfields.TextField(description文章内容) summaryfields.CharField(max_length200,description文章摘要) categoryfields.ForeignKeyField(models.Category,description分类) authorfields.ForeignKeyField(models.Author,description作者) tagsfields.ManyToManyField(models.Tag,related_namearticles,description标签) view_countfields.IntField(default0,description文章浏览量) statusfields.SmallIntField(choices[(0,草稿),(1,已发布)],default0,description文章状态) class Meta: tabledb_articlemodels/tags.pyfrom tortoise import Model,fields class Tag(Model): idfields.IntField(pkTrue,auto_incrementTrue,description标签ID) namefields.CharField(max_length50,description标签名称) colorfields.CharField(max_length20,default#409EFF,description标签颜色) class Meta: tabledb_tag接口1.分类增删改查功能from fastapi import APIRouter, HTTPException from pydantic import BaseModel from apps.models import Category category_routerAPIRouter() category_router.get(/category/,tags获取所有分类) async def get_all_category(): categoriesawait Category.all() return categories class CategoryCreateRequest(BaseModel): name:str category_router.post(/category/,tags新建分类) async def add_category(request:CategoryCreateRequest): categoryawait Category.create(**request.model_dump()) return category class CategoryUpdateRequest(BaseModel): name:str category_router.put(category/{id},tags编辑分类) async def update_category(id:int,request:CategoryUpdateRequest): categoryawait Category.get_or_none(idid) if not category: raise HTTPException(status_code404,detail分类不存在) await category.update_from_dict(request.model_dump()).save() return category category_router.delete(category/{id},tags删除分类) async def delete_category(id:int): categoryawait Category.get_or_none(idid) await category.delete() return {message:删除成功,id:id}2.文章增删改查功能from fastapi import APIRouter, Depends, HTTPException from pydantic import BaseModel, computed_field from tortoise.contrib.pydantic import pydantic_model_creator from apps.models import Article, Tag from utils.jwt import login_required article_routerAPIRouter() ArticleOutpydantic_model_creator(Article,nameArticleOut) class TagOut(BaseModel): id:int name:str color:str class CategoryOut(BaseModel): id:int name:str class AuthorOut(BaseModel): id:int name:str class ArticleListOut(BaseModel): id:int title:str summary:str view_count:int status:int category:CategoryOut author:AuthorOut tags:list[TagOut] computed_field property def status_text(self)-str: return 草稿 if self.status0 else 已发布 article_router.get(/article/,tags文章列表(条件筛选)) async def article_list(userDepends(login_required),category:intNone,title:strNone,): queryArticle.all() if category is not None: queryquery.filter(category_idcategory) if title is not None: queryquery.filter(title__containstitle) articles await query.prefetch_related(category, author, tags) return [ ArticleListOut( ida.id, titlea.title, summarya.summary, view_counta.view_count, statusa.status, categoryCategoryOut(ida.category.id, namea.category.name), authorAuthorOut(ida.author.id, namea.author.name), tags[TagOut(idt.id, namet.name, colort.color) for t in a.tags] ) for a in articles ] class ArticleCreateRequest(BaseModel): title:str content:str summary:str category:int tags:list[int] status:int0 article_router.post(/article/,tags新建文章) async def create_article(request:ArticleCreateRequest,userDepends(login_required)): tag_idsrequest.tags datarequest.model_dump(exclude{tags}) articleawait Article.create(**data,author_iduser.id) if tag_ids: tagsawait Tag.filter(id__intag_ids) await article.tags.add(*tags) return await ArticleOut.from_tortoise_orm(article) class ArticleUpdateRequest(BaseModel): title:str content:str summary:str category:int tags:list[int] status:int0 article_router.put(/article/{id},tags编辑文章) async def update_article(id:int,request:ArticleUpdateRequest,userDepends(login_required)): article_idawait Article.get_or_none(idid) if not article_id: raise HTTPException(status_code404,detail文章不存在) tag_ids request.tags data request.model_dump(exclude{tags}, exclude_noneTrue) await article_id.update_from_dict(data).save() if tag_ids is not None: tags await Tag.filter(id__intag_ids) await article_id.tags.clear() await article_id.tags.add(*tags) return await ArticleOut.from_tortoise_orm(article_id) article_router.delete(/article/{id},tags删除文章) async def delete_article(id:int,userDepends(login_required)): article_idawait Article.get_or_none(idid) if not article_id: raise HTTPException(status_code404,detail文章不存在) await article_id.delete() return {message:删除成功}3.标签增删查功能from fastapi import APIRouter, HTTPException from pydantic import BaseModel, Field from apps.models import Tag tag_routerAPIRouter() tag_router.get(/tag/,tags获取所有标签) async def tag_list(): tagsawait Tag.all() return tags class TagCreateRequest(BaseModel): name:str color:str tag_router.post(/tag/,tags创建标签) async def add_tag(request:TagCreateRequest): existsawait Tag.filter(namerequest.name).first() if exists: raise HTTPException(status_code400,detail标签名称已存在) tagawait Tag.create(**request.model_dump()) return tag tag_router.delete(/tag/{id},tags删除标签) async def delete_tag(id:int): tagawait Tag.get_or_none(idid) await tag.delete() return {message:标签删除成功}