lostplaces-backend/django_lostplaces/lostplaces/views/explorer_views.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from django.views import View

from django.utils.translation import ugettext_lazy as _

from django.shortcuts import render, redirect, get_object_or_404
from django.urls import reverse_lazy

from lostplaces.common import get_all_subclasses
from lostplaces.views.base_views import IsAuthenticatedMixin
from lostplaces.models.models import Explorer
from lostplaces.models.place import Place, PlaceAsset
from lostplaces.forms import ExplorerChangeForm, ExplorerUserChangeForm

class ExplorerProfileView(IsAuthenticatedMixin, View):
    def get(self, request, explorer_id):
        explorer = get_object_or_404(Explorer, pk=explorer_id)
        place_list = Place.objects.filter(submitted_by=explorer)
        place_count = place_list.count()

        context={
            'explorer': explorer,
            'place_count': place_count,
            'place_list': place_list,
            'assets': {}
        }

        asset_count = 0
        for subclass in get_all_subclasses(PlaceAsset): # kinda ugly, but PlaceAsset cannot be abstract according to django ORM
            objects = subclass.objects.filter(submitted_by=explorer)
            context['assets'][subclass.__name__.lower()+'s'] = objects
            asset_count += objects.count()
        
        context['asset_count'] = asset_count
                
        return render(
            request=request,
            template_name='explorer/profile.html',
            context=context
        )

class ExplorerProfileUpdateView(IsAuthenticatedMixin, View):
    success_message = ''
    permission_denied_message = ''

    def get(self, request, *args, **kwargs):
        context = {
            'explorer_image_url': request.user.explorer.profile_image.url,
            'explorer_user_change_form': ExplorerUserChangeForm(instance=request.user),
            'explorer_change_form': ExplorerChangeForm(instance=request.user.explorer)
        }
        return render(request, 'explorer/profile_update.html', context)