-
Notifications
You must be signed in to change notification settings - Fork 1
/
views.py
50 lines (41 loc) · 1.66 KB
/
views.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
# from django.shortcuts import render
from rest_framework import viewsets
from rest_framework.response import Response
from rest_framework.parsers import JSONParser
from .serializers import PlaceRequestSerializer
from .recommend1.recommend1 import recommend1
from .recommend2.recommend2 import recommend2
import json
# Create your views here.
class STCViewSet(viewsets.ViewSet):
@classmethod
def list(self, request):
recommender1 = recommend1()
UserID = request.query_params["User_ID"]
latitude = request.query_params["lat"]
longitude = request.query_params["long"]
keywords = request.query_params["keywords"]
epsilon = request.query_params["epsilon"]
requestData = {"User_ID": UserID,
"lat": latitude,
"long": longitude,
"keywords": keywords,
"epsilon": epsilon}
serializer = PlaceRequestSerializer(data=requestData)
if serializer.is_valid():
requestResult1 = recommender1.recommend(
latitude, longitude, keywords, epsilon
)
requestResult2 = recommend2.recommend(
UserID, latitude, longitude, epsilon
)
if requestResult2 is not None:
result = {"result": requestResult1+requestResult2}
else:
result = {"result": requestResult1}
jsonRequest = json.dumps(result, ensure_ascii=False)
return Response(
json.JSONDecoder().decode(jsonRequest)["result"],
status=200
)
return Response(serializer.errors, status=400)