Implemented search using elasticsearch

sdslabs · Apr 10, 2020 · cbc58f0 · cbc58f0
1 parent ddfdeb3
commit cbc58f0
Showing 1 changed file with 40 additions and 0 deletions.
diff --git a/rest_api/documents.py b/rest_api/documents.py
@@ -0,0 +1,40 @@
+from django_elasticsearch_dsl import Document, Index
+from rest_api.models import Course, Department, File
+
+courses = Index('courses')
+departments = Index('departments')
+files = Index('files')
+
+@courses.doc_type
+class CourseDocument(Document):
+    class Django:
+        model = Course
+
+        fields = [
+            'id',
+            'title',
+            'code',
+        ]
+
+@departments.doc_type
+class DepartmentDocument(Document):
+    class Django:
+        model = Department
+
+        fields = [
+            'id',
+            'title',
+            'abbreviation',
+        ]
+
+@files.doc_type
+class FileDocument(Document):
+    class Django:
+        model = File
+
+        fields = [
+            'id',
+            'title',
+            'fileext',
+            'filetype'
+        ]