Added logo model

cms/admin.py

@@ -1,6 +1,6 @@
 from django.contrib import admin
 
-from cms.models import InfoBanner, MenuItem, MessageTemplate, Post, PostLink
+from cms.models import InfoBanner, MenuItem, MessageTemplate, Post, PostLink, Logo
 
 
 @admin.register(MenuItem)
@@ -16,6 +16,15 @@ class MenuItemAdmin(admin.ModelAdmin):
     )
 
 
+@admin.register(Logo)
+class LogoAdmin(admin.ModelAdmin):
+    list_display = (
+        'image',
+        'name',
+        'disabled'
+    )
+
+
 @admin.register(Post)
 class PostAdmin(admin.ModelAdmin):
     list_display = (

cms/models.py

@@ -3,6 +3,7 @@
 from django.core.exceptions import ValidationError
 from django.db import models
 from django.utils.timezone import now
+from base.models import RestrictedFileField
 
 from competition.models import Event, Series
 
@@ -111,6 +112,24 @@ class Meta:
     # zobrazované aj možno niekde inde
 
 
+class Logo(models.Model):
+    """
+    Logo
+    """
+    class Meta:
+        verbose_name = 'logo'
+        verbose_name_plural = 'logá'
+
+        ordering = ['name', ]
+
+    name = models.CharField(verbose_name='názov loga', max_length=150)
+    disabled = models.BooleanField()
+    image = RestrictedFileField(
+        content_types=['image/svg+xml', 'image/png'],
+        upload_to='logo_images/',
+        verbose_name='Logo', null=True, blank=True)
+
+
 class InfoBanner(ModelWithVisibility):
     class Meta:
         verbose_name = 'Informácia v pohyblivom baneri'

cms/serializers.py

@@ -27,6 +27,12 @@ class Meta:
         model = models.Post
         fields = '__all__'
 
+@ts_interface(context='cms')
+class LogoSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = models.Logo
+        fields = '__all__'
+
 
 @ts_interface(context='cms')
 class InfoBannerSerializer(serializers.ModelSerializer):

cms/urls.py

@@ -11,6 +11,7 @@
 router.register(r'menu-item', views.MenuItemViewSet)
 router.register(r'info-banner', views.InfoBannerViewSet)
 router.register(r'message-template', views.MessageTemplateViewSet)
+router.register(r'logo', views.LogoViewSet)
 
 
 urlpatterns = [

cms/views.py

@@ -1,12 +1,14 @@
-from rest_framework import viewsets
+from rest_framework import viewsets, exceptions, status
 from rest_framework.decorators import action
 from rest_framework.response import Response
 
-from cms.models import InfoBanner, MenuItem, MessageTemplate, Post
-from cms.permissions import PostPermission
+from cms.models import InfoBanner, MenuItem, MessageTemplate, Post, Logo
+from cms.permissions import PostPermission, IsAdminUser
 from cms.serializers import (InfoBannerSerializer, MenuItemShortSerializer,
-                             MessageTemplateSerializer, PostSerializer)
+                             MessageTemplateSerializer, PostSerializer, LogoSerializer)
 
+from django.http import HttpResponse
+from base.utils import mime_type
 
 class MenuItemViewSet(viewsets.ReadOnlyModelViewSet):
     """Položky menu"""
@@ -45,6 +47,47 @@ def visible(self, request):
         return Response(serializer.data)
 
 
+class LogoViewSet(viewsets.ModelViewSet):
+    """Publikácie(výsledky, brožúrky, časopisy, ...)"""
+    queryset = Logo.objects.all()
+    serializer_class = LogoSerializer
+    permission_classes = (PostPermission,)
+
+    def perform_create(self, serializer):
+        '''
+        Vola sa pri vytvarani objektu,
+        checkuju sa tu permissions, ci user vie vytvorit publication v danom evente
+        '''
+        event = serializer.validated_data['event']
+        if event.can_user_modify(self.request.user):
+            serializer.save()
+        else:
+            raise exceptions.PermissionDenied(
+                'Nedostatočné práva na vytvorenie tohoto objektu')
+
+    @action(methods=['get'], detail=True, url_path='download')
+    def download_logo(self, request):
+        """Stiahne logo"""
+        logo = self.get_object()
+        response = HttpResponse(
+            logo.file, content_type=mime_type(logo.file))
+        response['Content-Disposition'] = f'attachment; filename="{logo.name}"'
+        return response
+
+    @action(methods=['post'], detail=False, url_path='upload', permission_classes=[IsAdminUser])
+    def upload_publication(self, request):
+        """Nahrá súbor publikácie"""
+        if 'file' not in request.data:
+            raise exceptions.ParseError(detail='Request neobsahoval súbor')
+
+        file = request.data['file']
+        if mime_type(file) not in ['application/jpg', 'application/png']:
+            raise exceptions.ParseError(detail='Nesprávny formát')
+
+        return Response(status=status.HTTP_201_CREATED)
+
+
+
 class InfoBannerViewSet(viewsets.ModelViewSet):
     """Správy v čiernom info banneri"""
     serializer_class = InfoBannerSerializer