import json import logging from datetime import datetime, timedelta from dateutil.parser import parse from django.conf import settings from django.db.models.fields import timezone from django.views.decorators.csrf import csrf_exempt from django.views.generic.list import ListView from rest_framework import status from rest_framework.decorators import api_view from rest_framework.response import Response from scrobbles.models import Scrobble from scrobbles.serializers import ScrobbleSerializer from videos.models import Series, Video logger = logging.getLogger(__name__) TRUTHY_VALUES = [ 'true', '1', 't', 'y', 'yes', 'yeah', 'yup', 'certainly', 'uh-huh', ] class RecentScrobbleList(ListView): model = Scrobble def get_queryset(self): return Scrobble.objects.filter(in_progress=False) @csrf_exempt @api_view(['GET']) def scrobble_endpoint(request): """List all Scrobbles, or create a new Scrobble""" scrobble = Scrobble.objects.all() serializer = ScrobbleSerializer(scrobble, many=True) return Response(serializer.data) @csrf_exempt @api_view(['POST']) def jellyfin_websocket(request): data_dict = request.data media_type = data_dict["ItemType"] # Check if it's a TV Episode video_dict = { "title": data_dict["Name"], "imdb_id": data_dict["Provider_imdb"], "video_type": Video.VideoType.MOVIE, "year": data_dict["Year"], } if media_type == 'Episode': series_name = data_dict["SeriesName"] series, series_created = Series.objects.get_or_create(name=series_name) video_dict['video_type'] = Video.VideoType.TV_EPISODE video_dict["tv_series_id"] = series.id video_dict["episode_number"] = data_dict["EpisodeNumber"] video_dict["season_number"] = data_dict["SeasonNumber"] video_dict["tvdb_id"] = data_dict.get("Provider_tvdb", None) video_dict["tvrage_id"] = data_dict.get("Provider_tvrage", None) video, video_created = Video.objects.get_or_create(**video_dict) if video_created: video.overview = data_dict["Overview"] video.tagline = data_dict["Tagline"] video.run_time_ticks = data_dict["RunTimeTicks"] video.run_time = data_dict["RunTime"] video.save() # Now we run off a scrobble timestamp = parse(data_dict["UtcTimestamp"]) scrobble_dict = { 'video_id': video.id, 'user_id': request.user.id, 'in_progress': True, } existing_finished_scrobble = ( Scrobble.objects.filter( video=video, user_id=request.user.id, in_progress=False ) .order_by('-modified') .first() ) minutes_from_now = timezone.now() + timedelta(minutes=15) if ( existing_finished_scrobble and existing_finished_scrobble.modified < minutes_from_now ): logger.info( 'Found a scrobble for this video less than 15 minutes ago, holding off scrobbling again' ) return Response(video_dict, status=status.HTTP_204_NO_CONTENT) scrobble, scrobble_created = Scrobble.objects.get_or_create( **scrobble_dict ) if scrobble_created: # If we newly created this, capture the client we're watching from scrobble.source = data_dict['ClientName'] scrobble.source_id = data_dict['MediaSourceId'] else: last_tick = scrobble.playback_position_ticks # Update a found scrobble with new position and timestamp scrobble.playback_position_ticks = data_dict["PlaybackPositionTicks"] scrobble.playback_position = data_dict["PlaybackPosition"] scrobble.timestamp = parse(data_dict["UtcTimestamp"]) scrobble.is_paused = data_dict["IsPaused"] in TRUTHY_VALUES scrobble.save() # If we hit our completion threshold, save it and get ready # to scrobble again if we re-watch this. if scrobble.percent_played >= getattr( settings, "PERCENT_FOR_COMPLETION", 95 ): scrobble.in_progress = False scrobble.playback_position_ticks = video.run_time_ticks scrobble.save() if scrobble.percent_played % 5 == 0: logger.info(f"You are {scrobble.percent_played}% through {video}") return Response(video_dict, status=status.HTTP_201_CREATED)