You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Dict
|
|
|
|
from ...config import app_config
|
|
from ...models import MediaCollection
|
|
from ..helpers import expect_suitable_extractor
|
|
from .base import CollectionExtractor
|
|
from .tt_rss import TtRssCollectionExtractor, TtRssConnectionParameter
|
|
from .youtube import YouTubeCollectionExtractor
|
|
|
|
|
|
tt_rss_params = TtRssConnectionParameter(**app_config["extractors"]["tt_rss"])
|
|
COLLECTION_EXTRACTORS: Dict[str, CollectionExtractor] = {
|
|
"tt-rss": TtRssCollectionExtractor(
|
|
params=tt_rss_params,
|
|
label_filter=-1033,
|
|
mark_as_read=True,
|
|
),
|
|
"youtube": YouTubeCollectionExtractor(),
|
|
}
|
|
|
|
|
|
def collection_expect_extractor(uri: str) -> CollectionExtractor:
|
|
return expect_suitable_extractor(
|
|
extractor_list=COLLECTION_EXTRACTORS.values(),
|
|
uri=uri,
|
|
)
|
|
|
|
|
|
def collection_update(
|
|
collection: MediaCollection,
|
|
check_cache_expired: bool = True,
|
|
) -> None:
|
|
ex = collection_expect_extractor(collection.uri)
|
|
ex.update_object(
|
|
object=collection,
|
|
check_cache_expired=check_cache_expired,
|
|
)
|
|
|
|
|
|
def collection_extract_uri(uri: str) -> MediaCollection:
|
|
elem: MediaCollection = CollectionExtractor.check_uri(uri)
|
|
ex = collection_expect_extractor(uri)
|
|
if not elem:
|
|
elem = ex.extract_and_store(uri)
|
|
else:
|
|
ex.update_object(elem, check_cache_expired=False)
|
|
return elem
|