Merge pull request #87 from brunoamaral/add-rss-feeds

Add rss feeds via Django

django/admin/urls.py

@@ -16,25 +16,31 @@
 from django.contrib import admin
 from django.urls import include, path, re_path
 from rest_framework import routers
-from api import views
+from api.views import *
+from rss.views import *
 
 router = routers.DefaultRouter()
-router.register(r'articles', views.ArticleViewSet, views.RelevantList)
-router.register(r'trials', views.TrialViewSet)
-router.register(r'sources', views.SourceViewSet)
+router.register(r'articles', ArticleViewSet, RelevantList)
+router.register(r'trials', TrialViewSet)
+router.register(r'sources', SourceViewSet)
 
 
 urlpatterns = [
 	path('admin/', admin.site.urls),
 	path('api-auth/', include('rest_framework.urls')),
-	path('articles/all/', views.AllArticleViewSet.as_view()),
-	path('trials/all/', views.AllTrialViewSet.as_view()),
-	re_path('^articles/relevant/$', views.RelevantList.as_view()),
-	path('articles/prediction/none/', views.ArticlesPredictionNone.as_view()),
-	re_path('^articles/source/(?P<source>.+)/$', views.ArticlesBySourceList.as_view()),
-	re_path('^trials/source/(?P<source>.+)/$', views.TrialsBySourceList.as_view()),
-	re_path('^articles/unsent/$', views.UnsentList.as_view()),
-	path('articles/related/', views.RelatedArticles.as_view({'get': 'list'})),
-	path('articles/count/', views.ArticlesCount.as_view({'get': 'list'})),
+	path('articles/all/', AllArticleViewSet.as_view()),
+	path('trials/all/', AllTrialViewSet.as_view()),
+	re_path('^articles/relevant/$', RelevantList.as_view()),
+	path('articles/prediction/none/', ArticlesPredictionNone.as_view()),
+	re_path('^articles/source/(?P<source>.+)/$', ArticlesBySourceList.as_view()),
+	re_path('^trials/source/(?P<source>.+)/$', TrialsBySourceList.as_view()),
+	re_path('^articles/unsent/$', UnsentList.as_view()),
+	path('articles/related/', RelatedArticles.as_view({'get': 'list'})),
+	path('articles/count/', ArticlesCount.as_view({'get': 'list'})),
 	path('', include(router.urls)),
+	path('feed/latest/articles/', LatestArticlesFeed()),
+	path('feed/latest/trials/', LatestTrialsFeed()),
+	path('feed/machine-learning/', MachineLearningFeed()),
+	path('feed/articles/prediction/none/', ToPredictFeed()),
+
 ]

django/rss/admin.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

django/rss/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class RssConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'rss'

django/rss/models.py

@@ -0,0 +1,3 @@
+from django.db import models
+
+# Create your models here.

django/rss/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

django/rss/views.py

@@ -0,0 +1,78 @@
+from django.shortcuts import render
+
+# Create your views here.
+from django.contrib.syndication.views import Feed
+from django.urls import reverse
+from gregory.models import Articles, Trials
+
+class LatestArticlesFeed(Feed):
+	title = "Gregory MS - latest research articles"
+	link = "/articles/"
+	description = "Real time results for research on Multiple Sclerosis."
+
+	def items(self):
+		return Articles.objects.order_by('-discovery_date')[:5]
+
+	def item_title(self, item):
+		return item.title
+
+	def item_description(self, item):
+		return item.summary
+
+	# # item_link is only needed if NewsItem has no get_absolute_url method.
+	def item_link(self, item):
+		return 'https://gregory-ms.com/articles/' + str(item.pk) + '/' 
+
+class LatestTrialsFeed(Feed):
+	title = "Gregory MS - latest clinical trials"
+	link = "/trials/"
+	description = "Real time results for research on Multiple Sclerosis."
+
+	def items(self):
+		return Trials.objects.order_by('-discovery_date')[:5]
+
+	def item_title(self, item):
+		return item.title
+
+	def item_description(self, item):
+		return item.summary
+
+	# # item_link is only needed if NewsItem has no get_absolute_url method.
+	def item_link(self, item):
+		return 'https://gregory-ms.com/trials/' + str(item.pk) + '/'
+
+class MachineLearningFeed(Feed):
+	title = "Gregory MS - Relevant articles by machine learning"
+	link = "/articles/"
+	description = "Real time results for research on Multiple Sclerosis."
+
+	def items(self):
+		return Articles.objects.filter(ml_prediction_gnb=True)[:20]
+
+	def item_title(self, item):
+		return item.title
+
+	def item_description(self, item):
+		return item.summary
+
+	# # item_link is only needed if NewsItem has no get_absolute_url method.
+	def item_link(self, item):
+		return 'https://gregory-ms.com/articles/' + str(item.pk) + '/' 
+
+class ToPredictFeed(Feed):
+	title = "Gregory MS - Relevant articles by machine learning"
+	link = "/articles/"
+	description = "Real time results for research on Multiple Sclerosis."
+
+	def items(self):
+		return Articles.objects.filter(ml_prediction_gnb=None)[:20]
+
+	def item_title(self, item):
+		return item.title
+
+	def item_description(self, item):
+		return item.summary
+
+	# # item_link is only needed if NewsItem has no get_absolute_url method.
+	def item_link(self, item):
+		return 'https://gregory-ms.com/articles/' + str(item.pk) + '/'