72 lines
3.1 KiB
Python
72 lines
3.1 KiB
Python
from ninja_extra import api_controller, ControllerBase, route
|
|
from ninja import Query
|
|
from apps.project.models import Abbreviation
|
|
from ninja_jwt.authentication import JWTAuth
|
|
from ninja_extra.permissions import IsAuthenticated
|
|
from ninja.pagination import paginate
|
|
from utils.chen_pagination import MyPagination
|
|
from django.db import transaction
|
|
from django.contrib.auth import get_user_model
|
|
from typing import List
|
|
from utils.chen_crud import multi_delete
|
|
from utils.chen_response import ChenResponse
|
|
from apps.dict.schema import DeleteSchema, AbbreviationOut, AbbreviationListInputSchema
|
|
|
|
Users = get_user_model()
|
|
|
|
@api_controller("/system", tags=['缩略语接口'], auth=JWTAuth(), permissions=[IsAuthenticated])
|
|
class AbbreviationController(ControllerBase):
|
|
@route.get("/abbreviation/getlist", response=List[AbbreviationOut], url_name="abbreviation-search")
|
|
@transaction.atomic
|
|
@paginate(MyPagination)
|
|
def get_abbreviation_list(self, payload: AbbreviationListInputSchema = Query(...)):
|
|
for attr, value in payload.__dict__.items():
|
|
if getattr(payload, attr) is None:
|
|
setattr(payload, attr, '')
|
|
qs = Abbreviation.objects.filter(title__icontains=payload.title, des__icontains=payload.des)
|
|
return qs
|
|
|
|
# 单独获取
|
|
@route.get("/abbreviation/index", response=List[AbbreviationOut], url_name="abbreviation-all")
|
|
@transaction.atomic
|
|
def get_contact_index(self):
|
|
qs = Abbreviation.objects.all()
|
|
return qs
|
|
|
|
@route.post("/abbreviation/save", response=AbbreviationOut, url_name='abbreviation-create')
|
|
@transaction.atomic
|
|
def create_abbreviation(self, data: AbbreviationListInputSchema):
|
|
for attr, value in data.__dict__.items():
|
|
if getattr(data, attr) is None:
|
|
setattr(data, attr, '')
|
|
# 判重key
|
|
assert_dict = data.dict()
|
|
key_qs = Abbreviation.objects.filter(title=data.title)
|
|
if len(key_qs) > 0:
|
|
return ChenResponse(code=400, status=400, message="缩略语重复，请修改...")
|
|
# 正常添加
|
|
qs = Abbreviation.objects.create(**assert_dict)
|
|
return qs
|
|
|
|
@route.put("/abbreviation/update/{id}", response=AbbreviationOut, url_name='abbreviation-update')
|
|
@transaction.atomic
|
|
def update_contact(self, id: int, data: AbbreviationListInputSchema):
|
|
for attr, value in data.__dict__.items():
|
|
if getattr(data, attr) is None:
|
|
setattr(data, attr, '')
|
|
key_qs = Abbreviation.objects.filter(title=data.title)
|
|
if len(key_qs) > 1:
|
|
return ChenResponse(code=400, status=400, message="缩略语重复，请修改...")
|
|
# 查询id
|
|
qs = Abbreviation.objects.get(id=id)
|
|
for attr, value in data.__dict__.items():
|
|
setattr(qs, attr, value)
|
|
qs.save()
|
|
return qs
|
|
|
|
@route.delete('/abbreviation/delete', url_name='abbreviation-delete')
|
|
@transaction.atomic
|
|
def delete_contact(self, data: DeleteSchema):
|
|
multi_delete(data.ids, Abbreviation)
|
|
return ChenResponse(message='单位或公司删除成功')
|