140 lines
5.3 KiB
Python
140 lines
5.3 KiB
Python
from rest_framework import viewsets
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView, View
|
|
from rest_framework.request import Request
|
|
|
|
from django.http import HttpResponseBadRequest
|
|
from django.db.models.query import QuerySet
|
|
|
|
from django.core.validators import validate_email
|
|
from django.core.exceptions import ValidationError
|
|
|
|
from .models import Apartament, User
|
|
from .serializer import (ApartamentListSerializer,
|
|
ApartamentDetailSerializer,
|
|
PsychTestAddResultSerializer,
|
|
PublicUserSerializer)
|
|
|
|
import json, math, random, re
|
|
|
|
class ApartamentViewSet(viewsets.ReadOnlyModelViewSet):
|
|
"""Вывод списка квартир или отдельной квартиры"""
|
|
|
|
def get_queryset(self):
|
|
apartaments = Apartament.objects.all()
|
|
return apartaments
|
|
|
|
def get_serializer_class(self):
|
|
if self.action == 'list':
|
|
return ApartamentListSerializer
|
|
elif self.action == "retrieve":
|
|
return ApartamentDetailSerializer
|
|
|
|
class ApartamentGetManyViewSet(viewsets.ReadOnlyModelViewSet):
|
|
"""Вывод отдельных квартир для сравнения"""
|
|
def get_queryset(self): # получение id квартир для избранного или сравнения
|
|
pk = self.request.query_params.get("pk", None) # получение id пользователя
|
|
|
|
if self.action == "retrieve": # если для страницы сравнения
|
|
apartaments_id = User.objects.get(pk=pk).apartaments_for_comparison.split(',') # получение id квартир
|
|
elif self.action == "list": # если для страницы избранного
|
|
apartaments_id = User.objects.get(pk=pk).favorites_apartaments.split(',') # получение id квартир
|
|
|
|
queryset = []
|
|
for i in apartaments_id:
|
|
queryset.append(Apartament.objects.get(pk=i))
|
|
|
|
return queryset
|
|
|
|
def retrieve(self, request, *args, **kwargs):
|
|
queryset = self.get_queryset()
|
|
apartaments = []
|
|
for i in queryset:
|
|
apartaments.append(ApartamentDetailSerializer(i).data)
|
|
return Response({'results': apartaments})
|
|
|
|
def list(self, request, *args, **kwargs):
|
|
queryset = self.get_queryset()
|
|
apartaments = []
|
|
for i in queryset:
|
|
apartaments.append(ApartamentListSerializer(i).data)
|
|
return Response({'results': apartaments})
|
|
|
|
class ApartmentFilter(viewsets.ViewSet):
|
|
"""Вывод списка квартир или отдельной квартиры"""
|
|
|
|
def list(self, req: Request):
|
|
filters = dict(req.data);
|
|
unfiltered = Apartament.objects.all();
|
|
filtered = [];
|
|
|
|
# TODO: Some better converting practice?
|
|
filters['price_range']['from'] = int(filters['price_range']['from']);
|
|
filters['price_range']['to'] = int(filters['price_range']['to']);
|
|
filters['area_range']['from'] = int(filters['area_range']['from']);
|
|
filters['area_range']['to'] = int(filters['area_range']['to']);
|
|
filters['rooms'] = int(filters['rooms'])
|
|
|
|
for entry in unfiltered:
|
|
if (filters['price_range']['from'] >= entry.price and (filters['price_range']['to'] != -1 and filters['price_range']['to'] <= entry.price)):
|
|
continue;
|
|
if (filters['area_range']['from'] >= entry.perimetrs and (filters['area_range']['to'] != -1 and filters['area_range']['to'] <= entry.perimetrs)):
|
|
continue;
|
|
if (filters['rooms'] != -1 and entry.rooms != filters['rooms']):
|
|
continue;
|
|
|
|
filtered.append(entry);
|
|
|
|
return Response(ApartamentListSerializer(filtered, many=True).data);
|
|
|
|
|
|
class PsychTestAddResultViewSet(viewsets.ViewSet):
|
|
def get_object(self, pk):
|
|
return User.objects.filter(pk=pk)
|
|
def update(self, request, pk):
|
|
user = self.get_object(pk)
|
|
result = request.query_params.get("result", None)
|
|
user.update(psych_test_result=result)
|
|
return Response(PsychTestAddResultSerializer(User.objects.get(pk=pk)).data)
|
|
|
|
class CompatibleUsersView(viewsets.ViewSet):
|
|
def list(self, req: Request):
|
|
user_data = dict(req.data);
|
|
|
|
# TODO: Verify auth
|
|
vvsu_login = user_data['openid'];
|
|
|
|
# Exclude already viewed users
|
|
exclude = [];
|
|
if ('exclude' in user_data.keys()):
|
|
exclude = user_data['exclude'];
|
|
|
|
try:
|
|
validate_email(vvsu_login);
|
|
except ValidationError:
|
|
return Request({'error': 'bad login'}, 400);
|
|
|
|
try:
|
|
this_user = User.objects.get(openid_addr=vvsu_login);
|
|
except User.DoesNotExist:
|
|
return Response({'error': 'user not found'}, 404);
|
|
|
|
score = this_user.psych_test_result;
|
|
|
|
users_query = User.objects.all();
|
|
|
|
users = [];
|
|
|
|
for user in users_query:
|
|
if (abs(user.psych_test_result - score) < 20):
|
|
users.append(PublicUserSerializer(user).data);
|
|
|
|
random.shuffle(users);
|
|
users = users[:7];
|
|
|
|
for i in range(3):
|
|
users.append(PublicUserSerializer(random.choice(users_query)).data);
|
|
|
|
random.shuffle(users);
|
|
|
|
return Response(users); |