43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
import logging
|
|
import pytz
|
|
from datetime import datetime, timedelta
|
|
from books.models import Book
|
|
from django.core.management.base import BaseCommand
|
|
from scrobbles.models import Scrobble
|
|
from vrobbler.apps.books.koreader import fix_long_play_stats_for_scrobbles
|
|
from vrobbler.apps.scrobbles.utils import timestamp_user_tz_to_utc
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Grace period between page reads for it to be a new scrobble
|
|
SESSION_GAP_SECONDS = 1800 # a half hour
|
|
|
|
def update_scrobble_from_page_data(scrobble, commit=True):
|
|
page_list = list(scrobble.book_page_data.items())
|
|
first_page_start_ts = datetime.fromtimestamp(page_list[0][1]["start_ts"])
|
|
last_page_end_ts = datetime.fromtimestamp(page_list[-1][1]["end_ts"])
|
|
|
|
if (
|
|
datetime(2023, 10, 15) <= first_page_start_ts <= datetime(2023, 12, 15)
|
|
):
|
|
first_page_start_ts.replace(tzinfo=pytz.timezone("Europe/Paris"))
|
|
last_page_end_ts.replace(tzinfo=pytz.timezone("Europe/Paris"))
|
|
else:
|
|
first_page_start_ts.replace(tzinfo=pytz.timezone("US/Eastern"))
|
|
last_page_end_ts.replace(tzinfo=pytz.timezone("US/Eastern"))
|
|
|
|
scrobble.timestamp = first_page_start_ts
|
|
scrobble.stop_timestamp = last_page_end_ts
|
|
if commit:
|
|
scrobble.save(update_fields=["timestamp", "stop_timestamp"])
|
|
|
|
|
|
class Command(BaseCommand):
|
|
def handle(self, *args, **options):
|
|
scrobbles_to_create = []
|
|
|
|
for scrobble in Scrobble.objects.filter(media_type="Book", source="KOReader"):
|
|
update_scrobble_from_page_data(scrobble)
|