You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
256 lines
9.3 KiB
256 lines
9.3 KiB
import warnings
|
|
from io import BytesIO
|
|
|
|
import pandas as pd
|
|
from django.db.models import Q
|
|
from django.http import HttpResponse
|
|
from django.http import JsonResponse
|
|
from rest_framework import status
|
|
from rest_framework import status as http_status
|
|
from rest_framework.decorators import action
|
|
from rest_framework.decorators import api_view
|
|
from rest_framework.decorators import permission_classes
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
from rest_framework.viewsets import ReadOnlyModelViewSet
|
|
|
|
from service import models
|
|
from service import pagination
|
|
from service import serializers
|
|
from service import utils
|
|
from service.permissions import UserPermission
|
|
from service.utils import load_data
|
|
|
|
|
|
class AOViewSet(ReadOnlyModelViewSet):
|
|
serializer_class = serializers.AOSerializer
|
|
queryset = models.AO.objects
|
|
permission_classes = [UserPermission]
|
|
|
|
|
|
class PlacementPointViewSet(ReadOnlyModelViewSet):
|
|
serializer_class = serializers.PlacementPointSerializer
|
|
queryset = models.PlacementPoint.objects
|
|
pagination_class = pagination.MyPagination
|
|
permission_classes = [UserPermission]
|
|
|
|
def get_queryset(self):
|
|
qs = self.queryset.all()
|
|
location_ids = self.request.GET.get('location_ids[]')
|
|
prediction_first = self.request.GET.get('prediction_first[]')
|
|
prediction_current = self.request.GET.get('prediction_current[]')
|
|
categories = self.request.GET.get('categories[]')
|
|
status = self.request.GET.get('status[]')
|
|
delta = self.request.GET.get('delta[]')
|
|
fact = self.request.GET.get('fact[]')
|
|
age = self.request.GET.get('age[]')
|
|
included = self.request.GET.get('included[]')
|
|
excluded = self.request.GET.get('excluded[]')
|
|
plan_first = self.request.GET.get('plan_first[]')
|
|
plan_current = self.request.GET.get('plan_current[]')
|
|
rayons = self.request.GET.get('rayon[]')
|
|
aos = self.request.GET.get('ao[]')
|
|
if location_ids:
|
|
location_ids = list(location_ids.split(','))
|
|
qs = qs.filter(pk__in=location_ids)
|
|
if prediction_first:
|
|
prediction_first = list(prediction_first.split(','))
|
|
qs = qs.filter(prediction_first__range=prediction_first)
|
|
if prediction_current:
|
|
prediction_current = list(prediction_current.split(','))
|
|
qs = qs.filter(prediction_current__range=prediction_current)
|
|
if categories:
|
|
categories = list(categories.split(','))
|
|
qs = qs.filter(category__in=categories)
|
|
if status:
|
|
status = list(status.split(','))
|
|
qs = qs.filter(status__in=status)
|
|
if delta:
|
|
delta = list(delta.split(','))
|
|
qs = qs.filter(delta__range=delta)
|
|
if fact:
|
|
fact = list(fact.split(','))
|
|
qs = qs.filter(fact__range=fact)
|
|
if age:
|
|
age = list(age.split(','))
|
|
qs = qs.filter(age__range=age)
|
|
if plan_first:
|
|
plan_first = list(plan_first.split(','))
|
|
qs = qs.filter(plan_first__range=plan_first)
|
|
if plan_current:
|
|
plan_current = list(plan_current.split(','))
|
|
qs = qs.filter(plan_current__range=plan_current)
|
|
if rayons:
|
|
rayons = list(rayons.split(','))
|
|
qs = qs.filter(rayon_id__in=rayons)
|
|
if aos:
|
|
aos = list(aos.split(','))
|
|
qs = qs.filter(okrug_id__in=aos)
|
|
if excluded:
|
|
excluded = list(excluded.split(','))
|
|
qs = qs.filter(~Q(pk__in=excluded))
|
|
if included:
|
|
inclded = list(included.split(','))
|
|
qs2 = models.PlacementPoint.objects.filter(pk__in=inclded).all()
|
|
qs = (qs | qs2).distinct()
|
|
return qs
|
|
|
|
@action(detail=False, methods=['get'])
|
|
def filters(self, request):
|
|
qs = self.get_queryset()
|
|
keys = (
|
|
'age_day', 'prediction_first', 'prediction_current',
|
|
'plan_first', 'plan_current', 'fact', 'delta_first',
|
|
'delta_current', 'flat_cnt', 'year_bld', 'levels',
|
|
'doors', 'flats_cnt', 'popul_home', 'popul_job',
|
|
'other_post_cnt', 'target_post_cnt', 'yndxfood_cnt',
|
|
'yndxfood_sum', 'yndxfood_cnt_cst',
|
|
)
|
|
temp_data = {
|
|
key: [
|
|
x for x in list(set(qs.values_list(key, flat=True))) if
|
|
x is not None
|
|
]
|
|
for key in keys
|
|
}
|
|
data = {
|
|
key: [
|
|
min(temp_data[key]), max(temp_data[key]),
|
|
] if temp_data[key] else [0, 100] for key in keys
|
|
}
|
|
return Response(data, status=status.HTTP_200_OK)
|
|
|
|
@action(detail=False, methods=['get'])
|
|
def search_address(self, request):
|
|
address = self.request.GET.get('address')
|
|
qs = self.get_queryset()
|
|
qs = qs.filter(address__icontains=address)
|
|
|
|
page = self.paginate_queryset(qs)
|
|
if page is not None:
|
|
serializer = self.get_serializer(page, many=True)
|
|
return self.get_paginated_response(serializer.data)
|
|
|
|
serializer = self.get_serializer(qs, many=True)
|
|
return Response(serializer.data)
|
|
|
|
@action(detail=False, methods=['put'])
|
|
def update_status(self, request):
|
|
qs = self.get_queryset()
|
|
new_status = self.request.GET.get('status')
|
|
if not new_status:
|
|
return Response({'message': 'No status'}, 400)
|
|
qs.update(**{'status': new_status})
|
|
return Response(
|
|
{'message': 'status updated'},
|
|
status=http_status.HTTP_200_OK,
|
|
)
|
|
|
|
@action(detail=False, methods=['put'])
|
|
def update_fact(self, request):
|
|
point_id = request.GET.get('postamat_id')
|
|
fact = request.GET.get('fact')
|
|
if not point_id or not fact or not fact.isdigit():
|
|
return Response(status=http_status.HTTP_400_BAD_REQUEST)
|
|
qs = models.PlacementPoint.objects.filter(postamat_id=point_id)
|
|
if not qs:
|
|
return Response(status=http_status.HTTP_404_NOT_FOUND)
|
|
qs.update(**{'fact': fact})
|
|
return Response({'message': 'fact updated'}, status=http_status.HTTP_200_OK)
|
|
|
|
@action(detail=False, methods=['put'])
|
|
def update_postamat_id(self, request):
|
|
postamat_id = request.GET.get('postamat_id')
|
|
point_id = request.GET.get('id')
|
|
if not point_id or not postamat_id:
|
|
return Response(status=http_status.HTTP_400_BAD_REQUEST)
|
|
qs = models.PlacementPoint.objects.filter(pk=point_id)
|
|
if not qs:
|
|
return Response(status=http_status.HTTP_404_NOT_FOUND)
|
|
qs.update(**{'postamat_id': postamat_id})
|
|
return Response({'message': 'Postamat id updated'}, status=http_status.HTTP_200_OK)
|
|
|
|
@action(detail=False, methods=['get'])
|
|
def to_excel(self, request):
|
|
qs = self.get_queryset()
|
|
data = pd.DataFrame(list(qs.values()))
|
|
data['start_date'] = data['start_date'].dt.tz_localize(None)
|
|
data['sample_trn'] = data['sample_trn'].astype(int)
|
|
with BytesIO() as b:
|
|
with pd.ExcelWriter(b) as writer:
|
|
data.to_excel(
|
|
writer, sheet_name='Placement Points',
|
|
index=False,
|
|
)
|
|
|
|
filename = 'placement_points.xlsx'
|
|
res = HttpResponse(
|
|
b.getvalue(),
|
|
content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
|
|
)
|
|
res['Content-Disposition'] = f'attachment; filename={filename}'
|
|
return res
|
|
|
|
@action(detail=False, methods=['get'])
|
|
def get_10k(self, request):
|
|
if models.PlacementPoint.objects.count() > 10000:
|
|
qs = \
|
|
models.PlacementPoint.objects.order_by(
|
|
'-prediction_current',
|
|
).all()[
|
|
10000
|
|
]
|
|
return Response({'prediction_current': qs.prediction_current}, 200)
|
|
return Response(
|
|
{
|
|
'prediction_current': models.PlacementPoint.objects.order_by(
|
|
'prediction_current',
|
|
).first().prediction_current,
|
|
},
|
|
200,
|
|
)
|
|
|
|
|
|
class refresh_placement_points(APIView):
|
|
@staticmethod
|
|
def post(request):
|
|
warnings.filterwarnings('ignore')
|
|
file = request.FILES['file']
|
|
load_data(file)
|
|
return Response(status=http_status.HTTP_200_OK)
|
|
|
|
|
|
class load_ao_and_rayons(APIView):
|
|
@staticmethod
|
|
def post(request):
|
|
warnings.filterwarnings('ignore')
|
|
file_ao = request.FILES['file_ao']
|
|
file_rayon = request.FILES['file_rayon']
|
|
utils.load_ao_and_rayons(file_ao, file_rayon)
|
|
return Response(status=http_status.HTTP_200_OK)
|
|
|
|
|
|
@api_view(['POST'])
|
|
def upload_rivals(request):
|
|
warnings.filterwarnings('ignore')
|
|
file_rivals = request.FILES['file_rivals']
|
|
utils.load_rivals(file_rivals)
|
|
return JsonResponse({'message': 'OK'})
|
|
|
|
|
|
@api_view(['POST'])
|
|
def upload_dist(request):
|
|
warnings.filterwarnings('ignore')
|
|
file_dist = request.FILES['file_dist']
|
|
utils.load_dist(file_dist)
|
|
return JsonResponse({'message': 'OK'})
|
|
|
|
|
|
@api_view(['GET'])
|
|
@permission_classes([IsAuthenticated])
|
|
def get_current_user(request):
|
|
return JsonResponse(
|
|
{'groups': [gr.name for gr in request.user.groups.all()]},
|
|
)
|