Speakers + Sponsors editing enabled.

Signed-off-by: Bojan Jovanovic <[email protected]>

Author: Bojan Jovanovic

.gitignore

@@ -7,5 +7,5 @@
 *.sqlite3
 *.py[cod]
 __pycache__
-/core/static
+
 /staticfiles/

pyconbalkan/conference/abstractions.py

@@ -1,9 +1,49 @@
+from django.contrib import admin
 from django.db import models
+from django.utils.translation import ugettext_lazy as _
+
 from pyconbalkan.conference.models import Conference
 
 
+def _get_default_conference():
+    return Conference.objects.first().id
+
+
 class AbstractConference(models.Model):
-    conference = models.ForeignKey(Conference, on_delete=models.CASCADE)
+    conference = models.ForeignKey(
+        Conference, on_delete=models.CASCADE, default=_get_default_conference
+    )
 
     class Meta:
-        abstract = True
+        abstract = True
+
+
+class ConferenceFilter(admin.SimpleListFilter):
+    title = _('Conference year')
+    parameter_name = 'conference'
+
+    def choices(self, changelist):
+        for lookup, title in self.lookup_choices:
+            yield {
+                'selected': self.value() is lookup or self.value() == str(lookup),
+                'query_string': changelist.get_query_string(
+                    {self.parameter_name: lookup}
+                ),
+                'display': title,
+            }
+
+    def lookups(self, request, model_admin):
+        choices = []
+        for _ in Conference.objects.all():
+            if _ == request.conference:
+                choices.append((None, "{} Current".format(_)))
+            else:
+                choices.append((_.id, str(_)))
+        return choices
+
+    def queryset(self, request, queryset):
+        return queryset.filter(conference=self.value() or request.conference.pk)
+
+
+class ConferenceAbstractAdmin(admin.ModelAdmin):
+    list_filter = (ConferenceFilter,)

pyconbalkan/speaker/admin.py

@@ -1,14 +1,15 @@
 from django.contrib import admin
 from markdownx.admin import MarkdownxModelAdmin
 
+from pyconbalkan.conference.abstractions import ConferenceAbstractAdmin
 from pyconbalkan.speaker.models import Speaker, SpeakerPhoto
 
 
 class SpeakerImageInline(admin.TabularInline):
     model = SpeakerPhoto
 
 
-class SpeakerAdmin(MarkdownxModelAdmin):
+class SpeakerAdmin(ConferenceAbstractAdmin, MarkdownxModelAdmin):
     inlines = [SpeakerImageInline]
 
 

pyconbalkan/speaker/migrations/0011_speaker_conference.py

@@ -0,0 +1,21 @@
+# Generated by Django 2.1.7 on 2019-03-03 16:46
+
+from django.db import migrations, models
+import django.db.models.deletion
+import pyconbalkan.conference.abstractions
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('conference', '0007_auto_20190227_0738'),
+        ('speaker', '0010_speaker_country'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='speaker',
+            name='conference',
+            field=models.ForeignKey(default=pyconbalkan.conference.abstractions._get_default_conference, on_delete=django.db.models.deletion.CASCADE, to='conference.Conference'),
+        ),
+    ]

pyconbalkan/speaker/models.py

@@ -1,10 +1,11 @@
 from django.db import models
 from django.db.models import CASCADE
 
+from pyconbalkan.conference.abstractions import AbstractConference
 from pyconbalkan.core.models import Person, ActiveModel
 
 
-class Speaker(ActiveModel, Person):
+class Speaker(AbstractConference, ActiveModel, Person):
     keynote = models.BooleanField(default=False)
 
     def __str__(self):

pyconbalkan/sponsors/admin.py

@@ -1,7 +1,10 @@
 from django.contrib import admin
+
+from pyconbalkan.conference.abstractions import ConferenceAbstractAdmin
 from pyconbalkan.sponsors.models import Package, PackageItem, Sponsor, Sponsoring
 
-admin.site.register(Sponsor)
+
+admin.site.register(Sponsor, ConferenceAbstractAdmin)
 admin.site.register(Sponsoring)
 admin.site.register(Package)
 admin.site.register(PackageItem)

pyconbalkan/sponsors/migrations/0010_sponsor_conference.py

@@ -0,0 +1,21 @@
+# Generated by Django 2.1.7 on 2019-03-03 14:42
+
+from django.db import migrations, models
+import django.db.models.deletion
+import pyconbalkan.conference.abstractions
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('conference', '0007_auto_20190227_0738'),
+        ('sponsors', '0009_auto_20190303_1426'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='sponsor',
+            name='conference',
+            field=models.ForeignKey(default=pyconbalkan.conference.abstractions._get_default_conference, on_delete=django.db.models.deletion.CASCADE, to='conference.Conference'),
+        ),
+    ]

pyconbalkan/sponsors/models.py

@@ -5,6 +5,7 @@
 from markdownx.models import MarkdownxField
 from djmoney.models.fields import MoneyField
 
+from pyconbalkan.conference.abstractions import AbstractConference
 from pyconbalkan.core.models import ActiveModel
 
 
@@ -25,7 +26,7 @@ class SponsorshipLevel(DjangoChoices):
 }
 
 
-class Sponsor(models.Model):
+class Sponsor(AbstractConference):
     name = models.CharField(max_length=256)
     description = MarkdownxField()
     level = models.CharField(max_length=16, choices=SponsorshipLevel.choices)