Merge pull request #6 from proevo/features/sponsors

Features/sponsors

Author: jefftriplett

pydotorg/settings.py

@@ -87,9 +87,10 @@
 
     'boxes',
     'cms',
-    'pages',
     'companies',
     'jobs',
+    'pages',
+    'sponsors',
     'successstories',
 )
 

pydotorg/urls.py

@@ -36,6 +36,7 @@
 
     url(r'^jobs/', include('jobs.urls', namespace='jobs')),
     url(r'^box/', include('boxes.urls')),
+    url(r'^sponsors/', include('sponsors.urls')),
     url(r'^success-stories/', include('successstories.urls')),
 
     # admin

sponsors/__init__.py

@@ -0,0 +1,18 @@
+from django.contrib import admin
+
+from .models import Sponsor
+from cms.admin import ContentManageableModelAdmin
+
+
+class SponsorAdmin(ContentManageableModelAdmin):
+    raw_id_fields = ['company']
+
+    def get_list_filter(self, request):
+        fields = list(super().get_list_filter(request))
+        return fields + ['is_published']
+
+    def get_list_display(self, request):
+        fields = list(super().get_list_display(request))
+        return fields + ['is_published']
+
+admin.site.register(Sponsor, SponsorAdmin)

sponsors/admin.py

@@ -0,0 +1,21 @@
+from django.db.models import Manager
+from django.db.models.query import QuerySet
+
+
+class SponsorQuerySet(QuerySet):
+    def draft(self):
+        return self.filter(is_published=False)
+
+    def published(self):
+        return self.filter(is_published=True)
+
+
+class SponsorManager(Manager):
+    def get_query_set(self):
+        return SponsorQuerySet(self.model, using=self._db)
+
+    def draft(self):
+        return self.get_query_set().draft()
+
+    def published(self):
+        return self.get_query_set().published()

sponsors/managers.py

@@ -0,0 +1,25 @@
+from django.conf import settings
+from django.db import models
+from markupfield.fields import MarkupField
+
+from .managers import SponsorManager
+from cms.models import ContentManageable
+from companies.models import Company
+
+
+DEFAULT_MARKUP_TYPE = getattr(settings, 'DEFAULT_MARKUP_TYPE', 'restructuredtext')
+
+
+class Sponsor(ContentManageable):
+    company = models.ForeignKey(Company)
+    content = MarkupField(default_markup_type=DEFAULT_MARKUP_TYPE, blank=True)
+    is_published = models.BooleanField(default=False)
+
+    objects = SponsorManager()
+
+    class Meta:
+        verbose_name = 'sponsor'
+        verbose_name_plural = 'sponsors'
+
+    def __str__(self):
+        return self.company.name

sponsors/models.py

@@ -0,0 +1,3 @@
+from .test_models import *
+from .test_views import *
+from .. import admin     # coverage FTW

sponsors/tests/__init__.py

@@ -0,0 +1,25 @@
+from django.test import TestCase
+
+from ..models import Sponsor
+from companies.models import Company
+
+
+class SponsorModelTests(TestCase):
+    def setUp(self):
+        self.company1 = Company.objects.create(name='Python')
+
+        self.Sponsor1 = Sponsor.objects.create(
+            company=self.company1,
+            is_published=True)
+
+        self.company2 = Company.objects.create(name='Python Hidden')
+
+        self.Sponsor2 = Sponsor.objects.create(
+            company=self.company2,
+            is_published=False)
+
+    def test_draft(self):
+        self.assertQuerysetEqual(Sponsor.objects.draft(), ['<Sponsor: Python Hidden>'])
+
+    def test_published(self):
+        self.assertQuerysetEqual(Sponsor.objects.published(), ['<Sponsor: Python>'])

sponsors/tests/test_models.py

@@ -0,0 +1,28 @@
+from django.core.urlresolvers import reverse
+from django.test import TestCase
+
+from ..models import Sponsor
+from companies.models import Company
+
+
+class SponsorViewTests(TestCase):
+    urls = 'sponsors.urls'
+
+    def setUp(self):
+        self.company1 = Company.objects.create(name='Python')
+
+        self.Sponsor1 = Sponsor.objects.create(
+            company=self.company1,
+            is_published=True)
+
+        self.company2 = Company.objects.create(name='Python Hidden')
+
+        self.Sponsor2 = Sponsor.objects.create(
+            company=self.company2,
+            is_published=False)
+
+    def test_sponsor_list(self):
+        url = reverse('sponsor_list')
+        r = self.client.get(url)
+        self.assertEqual(r.status_code, 200)
+        self.assertEqual(len(r.context['sponsors']), 1)

sponsors/tests/test_views.py

@@ -0,0 +1,8 @@
+from django.conf.urls import patterns, url
+
+from . import views
+
+
+urlpatterns = patterns('',
+    url(r'^$', views.SponsorList.as_view(), name='sponsor_list'),
+)