71 lines
2.8 KiB
Python
71 lines
2.8 KiB
Python
import pandas as pd
|
|
from django.db.transaction import atomic
|
|
from django.http import Http404
|
|
from rest_framework.exceptions import APIException
|
|
from rest_framework.permissions import IsAdminUser, IsAuthenticated
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
|
|
from ..serializers import SuppressionAdministresSerializer
|
|
from ..utils.alimentation_decorators import (data_perf_logger_factory,
|
|
get_data_logger)
|
|
from ..utils.decorators import execution_time, query_count
|
|
from ..utils_extraction import (DataFrameTypes, FileTypes, read_files_by_type,
|
|
to_table_suppression_administres)
|
|
from ..utils_insertion import suppression_administres
|
|
|
|
|
|
class SuppressionAdministresView(APIView):
|
|
""" Vue pour supprimer les administrés de la base """
|
|
|
|
permission_classes = [IsAuthenticated, IsAdminUser]
|
|
serializer_class = SuppressionAdministresSerializer
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.logger = get_data_logger(self)
|
|
|
|
|
|
def get(self, request):
|
|
return Response("Formulaire d\'OGURE NG permettant de supprimer des administés de la base de données")
|
|
|
|
|
|
@atomic
|
|
@execution_time(logger_factory=data_perf_logger_factory)
|
|
@query_count(logger_factory=data_perf_logger_factory)
|
|
def post(self, request):
|
|
"""
|
|
Charge le(s) fichier(s) et met à jour la base.
|
|
|
|
:param request: requête, contient les fichiers
|
|
:type request: class:`rest_framework.request.Request`
|
|
|
|
:raises: class:`rest_framework.exceptions.APIException`
|
|
|
|
:return: réponse
|
|
:rtype: class:`rest_framework.response.Response`
|
|
"""
|
|
try:
|
|
validator = self.serializer_class(data=request.data)
|
|
validator.is_valid(raise_exception=True)
|
|
adm_suppr = 0
|
|
|
|
df_adm_suppr = read_files_by_type({
|
|
FileTypes.ADM_SUPPR: validator.validated_data.get('administres')
|
|
}).get(DataFrameTypes.ADM_SUPPR)
|
|
|
|
if df_adm_suppr is not None:
|
|
df = to_table_suppression_administres(df_adm_suppr)
|
|
self.logger.info('Extraction des administés à supprimer ------> Succès')
|
|
adm_suppr = suppression_administres(df)
|
|
self.logger.info('Suppression des administrés ------> Succès')
|
|
else:
|
|
self.logger.info('Mise à jour ignorée : suppression des administrés')
|
|
return Response({f'Suppression de {adm_suppr} administré(s) réussie'})
|
|
except (Http404, APIException):
|
|
raise
|
|
except BaseException:
|
|
message = "Echec de la suppression"
|
|
self.logger.exception(message)
|
|
raise APIException(message)
|