Colin Powell
fe47d916e9
Biggest thing here is adding the ability to scrobble until the video is 95% done and then not scrobble the same video file again for 15 minutes. This seems hacky, but in practice works pretty well, so long as you don't monkey around with the 95% completion thing. Unlike music, videos are generally watched all the way through.
139 lines
4.6 KiB
Python
139 lines
4.6 KiB
Python
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', 'POST'])
|
|
def scrobble_list(request):
|
|
"""List all Scrobbles, or create a new Scrobble"""
|
|
if request.method == 'GET':
|
|
scrobble = Scrobble.objects.all()
|
|
serializer = ScrobbleSerializer(scrobble, many=True)
|
|
return Response(serializer.data)
|
|
|
|
elif request.method == 'POST':
|
|
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)
|