Skip to content

Commit 002dd46

Browse files
committed
Adding reasonable route.
1 parent 404e710 commit 002dd46

2 files changed

Lines changed: 29 additions & 2 deletions

File tree

api_v2/models/document.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,5 +118,10 @@ def as_text(self):
118118
def get_absolute_url(self):
119119
return reverse(self.__name__, kwargs={"pk": self.pk})
120120

121+
def search_result_extra_fields(self):
122+
return {
123+
"school":self.school.key,
124+
}
125+
121126
class Meta:
122127
abstract = True

api_v2/serializers/search.py

Lines changed: 24 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,12 @@
44

55
from api_v2 import models
66
from api import models as v1
7-
from .item import ItemSerializer
7+
from django.urls import reverse
88

99
class SearchResultSerializer(serializers.ModelSerializer):
1010
object = serializers.SerializerMethodField(method_name='get_object')
1111
document = serializers.SerializerMethodField(method_name='get_document')
12+
route = serializers.SerializerMethodField(method_name='get_route')
1213

1314

1415
class Meta:
@@ -20,6 +21,7 @@ class Meta:
2021
'object',
2122
'object_model',
2223
'schema_version',
24+
'route',
2325
'rank',
2426
'text',
2527
'highlighted']
@@ -63,4 +65,24 @@ def get_document(self, obj):
6365
return {
6466
'key': doc.key,
6567
'name': doc.name
66-
}
68+
}
69+
70+
def get_route(self, obj):
71+
# May want to split this out into v1 and v2?
72+
route_lookup = {
73+
"Item":"items",
74+
"Creature":"creatures",
75+
"Spell":"spells",
76+
"CharacterClass":"class",
77+
"Monster":"monsters",
78+
"MagicItem":"magicitems",
79+
"Section":"sections",
80+
"Background":"backgrounds",
81+
"Subrace":"subraces",
82+
"Feat":"feats",
83+
"Race":"races",
84+
"Plane":"planes",
85+
}
86+
87+
route = "{}/{}/".format(obj.schema_version,route_lookup[obj.object_model])
88+
return route

0 commit comments

Comments
 (0)