48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
import logging
|
|
|
|
from celery import shared_task
|
|
from django.contrib.auth import get_user_model
|
|
from scrobbles.models import (
|
|
AudioScrobblerTSVImport,
|
|
KoReaderImport,
|
|
LastFmImport,
|
|
)
|
|
|
|
from vrobbler.apps.scrobbles.stats import build_yesterdays_charts_for_user
|
|
|
|
logger = logging.getLogger(__name__)
|
|
User = get_user_model()
|
|
|
|
|
|
@shared_task
|
|
def process_lastfm_import(import_id):
|
|
lastfm_import = LastFmImport.objects.filter(id=import_id).first()
|
|
if not lastfm_import:
|
|
logger.warn(f"LastFmImport not found with id {import_id}")
|
|
|
|
lastfm_import.process()
|
|
|
|
|
|
@shared_task
|
|
def process_tsv_import(import_id):
|
|
tsv_import = AudioScrobblerTSVImport.objects.filter(id=import_id).first()
|
|
if not tsv_import:
|
|
logger.warn(f"AudioScrobblerTSVImport not found with id {import_id}")
|
|
|
|
tsv_import.process()
|
|
|
|
|
|
@shared_task
|
|
def process_koreader_import(import_id):
|
|
koreader_import = KoReaderImport.objects.filter(id=import_id).first()
|
|
if not koreader_import:
|
|
logger.warn(f"KOReaderImport not found with id {import_id}")
|
|
|
|
koreader_import.process()
|
|
|
|
|
|
@shared_task
|
|
def create_yesterdays_charts():
|
|
for user in User.objects.all():
|
|
build_yesterdays_charts_for_user(user)
|