|
@@ -1,7 +1,10 @@
|
|
|
import json
|
|
|
import logging
|
|
|
+from datetime import datetime, timedelta
|
|
|
|
|
|
-import dateutil
|
|
|
+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
|
|
@@ -30,6 +33,9 @@ TRUTHY_VALUES = [
|
|
|
class RecentScrobbleList(ListView):
|
|
|
model = Scrobble
|
|
|
|
|
|
+ def get_queryset(self):
|
|
|
+ return Scrobble.objects.filter(in_progress=False)
|
|
|
+
|
|
|
|
|
|
@csrf_exempt
|
|
|
@api_view(['GET', 'POST'])
|
|
@@ -41,58 +47,92 @@ def scrobble_list(request):
|
|
|
return Response(serializer.data)
|
|
|
|
|
|
elif request.method == 'POST':
|
|
|
- data_dict = json.loads(request.data["_content"])
|
|
|
+ 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.objects.get_or_create(name=series_name)
|
|
|
+ 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_num"] = data_dict["EpisodeNumber"]
|
|
|
- video_dict["season_num"] = data_dict["SeasonNumber"]
|
|
|
- video_dict["tvdb_id"] = data_dict["Provider_tvdb"]
|
|
|
- video_dict["tvrage_id"] = data_dict["Provider_tvrage"]
|
|
|
+ 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, _created = Video.objects.get_or_create(**video_dict)
|
|
|
+ video, video_created = Video.objects.get_or_create(**video_dict)
|
|
|
|
|
|
- video.year = data_dict["Year"]
|
|
|
- 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()
|
|
|
+ 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 = dateutil.parser.parse(data_dict["UtcTimestamp"])
|
|
|
+ scrobble.timestamp = parse(data_dict["UtcTimestamp"])
|
|
|
scrobble.is_paused = data_dict["IsPaused"] in TRUTHY_VALUES
|
|
|
- scrobble.played_to_completion = (
|
|
|
- data_dict["PlayedToCompletion"] in TRUTHY_VALUES
|
|
|
- )
|
|
|
scrobble.save()
|
|
|
|
|
|
- logger.info(f"You are {scrobble.percent_played}% through {video}")
|
|
|
+ # 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)
|
|
|
- # return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|