blob: 795617a1970c057465f24276aa66145abf015fc0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
import logging
import re
import importlib
import sys
from utils import NoExtractorException
extrs = [
'bandcamp'
]
class DlJob():
def __init__(self, url, output, extractor=None):
if extractor in extrs:
cls = self._get_class(extractor)
self.extr = cls(reg=None, url=url)
else:
self.extr = self._find(url)
self.output = output
self._albums = []
if not self.extr:
raise NoExtractorException("No extractor found for " + url + ".")
def _find(self, url):
for cls in self._list_extractors():
match = cls.pattern.match(url)
if match:
return cls(match, url)
def _list_extractors(self):
for extr in iter(extrs):
module = importlib.import_module('.'+extr, __package__)
yield from self._add_module(module)
def _add_module(self, module):
classes = self._get_classes(module)
for cls in classes:
cls.pattern = re.compile(cls.pattern)
return classes
def _get_classes(self, module):
return [
cls for cls in module.__dict__.values() if (
hasattr(cls, "pattern") and cls.__module__ == module.__name__
)
]
def _get_class(self, extractor):
module = importlib.import_module('.' + extractor, __package__)
classes = self._get_classes(module)
for cls in classes:
if cls.__name__ == extractor:
return cls
def run(self):
self.extr.get_albums()
self.extr.download_albums(self.output)
|
