blob: 97b0f5521834f1184b22cd69468a7fb864ac2f8a (
plain) (
blame)
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
51
52
|
from rest_framework.response import Response
from rest_framework.viewsets import ViewSet
from .models import SnakeName
from .serializers import SnakeNameSerializer
class SnakeNameViewSet(ViewSet):
"""
View of snake names for the bot's snake cog from our first code jam's winners.
## Routes
### GET /bot/snake-names
By default, return a single random snake name in the following format:
>>> {
... 'name': "Python",
... 'scientific': "Langus greatus"
... }
If the `get_all` query parameter is given, for example using...
$ curl api.pythondiscord.local:8000/bot/snake-names?get_all=yes
...then the API will return all snake names and scientific names in the database,
for example:
>>> [
... {'name': "Python 3", 'scientific': "Langus greatus"},
... {'name': "Python 2", 'scientific': "Langus decentus"}
... ]
"""
serializer_class = SnakeNameSerializer
def get_queryset(self):
return SnakeName.objects.all()
def list(self, request):
if request.query_params.get('get_all'):
queryset = self.get_queryset()
serialized = self.serializer_class(queryset, many=True)
return Response(serialized.data)
single_snake = SnakeName.objects.order_by('?').first()
if single_snake is not None:
body = {
'name': single_snake.name,
'scientific': single_snake.scientific
}
return Response(body)
return Response({})
|