uri_suitable: Explicitly check for "is not None" & use helper methods

- helper for convertion from boolean to SuitableLevel
master
Felix Stupp 2 years ago
parent 54748a9540
commit 49cff6f300
Signed by: zocker
GPG Key ID: 93E1BD26F6B02FB7

@ -48,7 +48,7 @@ class YouTubeCollectionExtractor(CollectionExtractor[Dict]):
super().__init__("youtube")
def uri_suitable(self, uri: str) -> SuitableLevel:
return SuitableLevel.ALWAYS if self.__uri_regex.match(uri) else SuitableLevel.NO
return SuitableLevel.always_or_no(self.__uri_regex.match(uri) is not None)
def can_extract_offline(self, uri: str) -> bool:
return True

@ -81,7 +81,7 @@ class YoutubeMediaExtractor(MediaExtractor[YoutubeVideoData]):
super().__init__("youtube")
def uri_suitable(self, uri: str) -> SuitableLevel:
return SuitableLevel.always_or_no(self.__uri_regex.match(uri))
return SuitableLevel.always_or_no(self.__uri_regex.match(uri) is not None)
def _get_author_data(self, data: YoutubeVideoData) -> Optional[AuthorExtractedData]:
return AuthorExtractedData(

Loading…
Cancel
Save