Support ynet articles as well

3 files changed, 16 insertions, 5 deletions

diff --git a/newsdiff/core/views.py b/newsdiff/core/views.py
index 3c339f6..29176d4 100644
--- a/newsdiff/core/views.py
+++ b/newsdiff/core/views.py
@@ -6,17 +6,15 @@ from reversion.models import Version
 from .models import *
 
 
-class HaaretzArticleView(DetailView):
+class ArticleView(DetailView):
 
     context_object_name = 'article'
-    model = HaaretzArticle
-    slug_field = 'haaretz_id'
     slug_url_kwarg = 'id'
 
-    template_name = 'haaretz-article.html'
+    template_name = 'article.html'
 
     def get_context_data(self, **kwargs):
-        context = super(HaaretzArticleView, self).get_context_data(**kwargs)
+        context = super(ArticleView, self).get_context_data(**kwargs)
         versions = reversion.get_for_object(self.object)
         version_diffs = [(v1, v2) for v1, v2 in zip(versions, versions[1:])]
         context['diffs'] = [{
@@ -24,3 +22,15 @@ class HaaretzArticleView(DetailView):
         } for (v1, v2) in version_diffs]
         # context['diffs'] = [(v1, v2) for v1, v2 in zip(versions, versions[1:])]
         return context
+
+
+class HaaretzArticleView(ArticleView):
+
+    model = HaaretzArticle
+    slug_field = 'haaretz_id'
+
+
+class YnetArticleView(ArticleView):
+
+    model = YnetArticle
+    slug_field = 'ynet_id'
diff --git a/newsdiff/templates/haaretz-article.html b/newsdiff/templates/article.html
index 90e2778..90e2778 100644
--- a/newsdiff/templates/haaretz-article.html
+++ b/newsdiff/templates/article.html
diff --git a/newsdiff/urls.py b/newsdiff/urls.py
index e24a740..613516d 100644
--- a/newsdiff/urls.py
+++ b/newsdiff/urls.py
@@ -10,6 +10,7 @@ admin.autodiscover()
 
 urlpatterns = patterns('',
     url(r'^haaretz/(?P<id>[\d\.]+)$', HaaretzArticleView.as_view()),
+    url(r'^ynet/(?P<id>[\d]+)$', YnetArticleView.as_view()),
     url(r'^admin/', include(admin.site.urls)),
     url(r'^robots.txt', lambda x: HttpResponse('User-agent: *\nDisallow: /', mimetype='text/plain'))
 )