2018-04-12 16:41:43 +00:00
|
|
|
from requests.exceptions import RequestException
|
|
|
|
|
2018-04-11 21:13:33 +00:00
|
|
|
from funkwhale_api.taskapp import celery
|
|
|
|
|
|
|
|
from . import library as lb
|
|
|
|
from . import models
|
|
|
|
|
|
|
|
|
2018-04-12 16:41:43 +00:00
|
|
|
@celery.app.task(
|
|
|
|
name='federation.scan_library',
|
|
|
|
autoretry_for=[RequestException],
|
|
|
|
retry_backoff=30,
|
|
|
|
max_retries=5)
|
2018-04-11 21:13:33 +00:00
|
|
|
@celery.require_instance(models.Library, 'library')
|
2018-04-12 16:41:43 +00:00
|
|
|
def scan_library(library, until=None):
|
2018-04-11 21:13:33 +00:00
|
|
|
if not library.federation_enabled:
|
|
|
|
return
|
|
|
|
|
|
|
|
data = lb.get_library_data(library.url)
|
|
|
|
scan_library_page.delay(
|
2018-04-12 16:41:43 +00:00
|
|
|
library_id=library.id, page_url=data['first'], until=until)
|
2018-04-11 21:13:33 +00:00
|
|
|
|
|
|
|
|
2018-04-12 16:41:43 +00:00
|
|
|
@celery.app.task(
|
|
|
|
name='federation.scan_library_page',
|
|
|
|
autoretry_for=[RequestException],
|
|
|
|
retry_backoff=30,
|
|
|
|
max_retries=5)
|
2018-04-11 21:13:33 +00:00
|
|
|
@celery.require_instance(models.Library, 'library')
|
2018-04-12 16:41:43 +00:00
|
|
|
def scan_library_page(library, page_url, until=None):
|
2018-04-11 21:13:33 +00:00
|
|
|
if not library.federation_enabled:
|
|
|
|
return
|
|
|
|
|
|
|
|
data = lb.get_library_page(library, page_url)
|
|
|
|
lts = []
|
|
|
|
for item_serializer in data['items']:
|
2018-04-12 16:41:43 +00:00
|
|
|
item_date = item_serializer.validated_data['published']
|
|
|
|
if until and item_date < until:
|
|
|
|
return
|
2018-04-11 21:13:33 +00:00
|
|
|
lts.append(item_serializer.save())
|
2018-04-12 16:41:43 +00:00
|
|
|
|
|
|
|
next_page = data.get('next')
|
|
|
|
if next_page and next_page != page_url:
|
|
|
|
scan_library_page.delay(library_id=library.id, page_url=next_page)
|