import json import logging from datetime import datetime from celery import states from django.conf import settings from django.contrib.auth.decorators import login_required from django.contrib.auth.mixins import LoginRequiredMixin from django.core.cache import cache from django.db.models import Avg, Count, F from django.http import HttpResponse from django.views.generic import DetailView, ListView from django.views.generic.list import MultipleObjectMixin from django_celery_results.models import TaskResult from games.tasks import update_roms from .models import Developer, Game, GameCollection, GameSystem, Genre, Publisher logger = logging.Logger(__name__) IN_PROGRESS_STATES = [states.PENDING, states.STARTED, states.RETRY] class RecentGameList(ListView): model = Game paginate_by = 10 queryset = Game.objects.order_by("-created")[:70] def get_context_data(self, **kwargs): cached_game_id = cache.get("todays_game_id", None) if not cached_game_id: todays_game = ( Game.objects.filter(rating__gte=settings.FEATURED_THRESHOLD, featured_on__isnull=True) .order_by("?") .first() ) featured_collection = GameCollection.objects.filter( slug="emus-featured" ).first() if featured_collection: featured_collection.games.add(todays_game) featured_collection.save() cache.set("todays_game_id", todays_game.id, settings.FEATURED_GAME_DURATION) todays_game.featured_on = datetime.now() todays_game.save(update_fields=["featured_on"]) else: todays_game = Game.objects.get(id=cached_game_id) return super(RecentGameList, self).get_context_data( todays_game=todays_game, **kwargs, ) class LibraryGameList(ListView): template_name = "games/game_library_list.html" model = Game paginate_by = 400 default_sort_key = "-created" def get_context_data(self, **kwargs): game_system_slug = self.request.GET.get("game_system") order_by = self.request.GET.get("order_by", self.default_sort_key) if order_by[0] == "-": order_by = order_by[1:] object_list = Game.objects.order_by(F(order_by).desc(nulls_last=True)) else: object_list = Game.objects.order_by(F(order_by).asc(nulls_last=True)) if game_system_slug: object_list = object_list.filter( game_system__retropie_slug=game_system_slug ) context = super(LibraryGameList, self).get_context_data( object_list=object_list, **kwargs ) return context class FilterableBaseListView(ListView): def get_queryset(self, **kwargs): order_by = self.request.GET.get("order_by", "name") queryset = super().get_queryset(**kwargs) queryset = queryset.annotate(num_games=Count("game")).annotate( rating=Avg("game__rating") ) if order_by[0] == "-": order_by = order_by[1:] queryset = queryset.order_by(F(order_by).desc(nulls_last=True)) else: queryset = queryset.order_by(F(order_by).asc(nulls_last=True)) return queryset class PublisherList(FilterableBaseListView): model = Publisher class DeveloperList(FilterableBaseListView): model = Developer class GenreList(FilterableBaseListView): model = Genre class GameDetail(DetailView): model = Game class GamePlayDetail(DetailView, LoginRequiredMixin): template_name = "games/game_play_detail.html" model = Game class GameSystemList(ListView): model = GameSystem class GameSystemDetail(DetailView, MultipleObjectMixin): model = GameSystem paginate_by = 20 def get_context_data(self, **kwargs): object_list = Game.objects.filter(game_system=self.get_object()) context = super(GameSystemDetail, self).get_context_data( object_list=object_list, **kwargs ) return context class GenreDetail(DetailView, MultipleObjectMixin): model = Genre paginate_by = 20 def get_context_data(self, **kwargs): object_list = Game.objects.filter(genre=self.get_object()) context = super(GenreDetail, self).get_context_data( object_list=object_list, **kwargs ) return context class PublisherDetail(DetailView, MultipleObjectMixin): model = Publisher paginate_by = 20 def get_context_data(self, **kwargs): object_list = Game.objects.filter(publisher=self.get_object()) context = super(PublisherDetail, self).get_context_data( object_list=object_list, **kwargs ) return context class DeveloperDetail(DetailView, MultipleObjectMixin): model = Developer paginate_by = 20 def get_context_data(self, **kwargs): object_list = Game.objects.filter(developer=self.get_object()) context = super(DeveloperDetail, self).get_context_data( object_list=object_list, **kwargs ) return context class GameCollectionList(ListView): model = GameCollection class GameCollectionDetail(DetailView, LoginRequiredMixin): model = GameCollection def get_context_data(self, **kwargs): collection = self.get_object() order_by = self.request.GET.get("order_by", "release_date") object_list = collection.games.all() if order_by[0] == "-": order_by = order_by[1:] object_list = object_list.order_by(F(order_by).desc(nulls_last=True)) else: object_list = object_list.order_by(F(order_by).asc(nulls_last=True)) context = super(GameCollectionDetail, self).get_context_data( object_list=object_list, **kwargs ) return context @login_required def trigger_rom_update(request): full_scan = request.GET.get("full_scan", False) if full_scan == "true": full_scan = True game_systems = request.GET.getlist("game_systems") if not game_systems: game_systems = list(settings.GAME_SYSTEM_DEFAULTS.keys()) try: update_roms.delay(game_systems, full_scan=full_scan) except FileNotFoundError: return HttpResponse( json.dumps({"success": False, "msg": "Skyscraper is not installed"}), status=400, content_type="application/json", ) return HttpResponse( json.dumps({"success": True, "msg": "Library scan started"}), status=200, content_type="application/json", ) @login_required def library_update_status(request): update_task = TaskResult.objects.filter( task_name="games.tasks.update_roms", status__in=IN_PROGRESS_STATES, ).first() if not update_task: state = states.SUCCESS else: state = update_task.status return HttpResponse( json.dumps({"state": state}), status=200, content_type="application/json", )