40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
|
from youtube_dl.extractor import *
|
||
|
|
import re
|
||
|
|
import os
|
||
|
|
|
||
|
|
def list_non_suitable_extractors(age_limit):
|
||
|
|
"""
|
||
|
|
Return a list of extractors that are suitable for the given age,
|
||
|
|
sorted by extractor ID.
|
||
|
|
"""
|
||
|
|
|
||
|
|
return sorted(
|
||
|
|
filter(lambda ie: not ie.is_suitable(age_limit), gen_extractors()),
|
||
|
|
key=lambda ie: ie.IE_NAME.lower())
|
||
|
|
|
||
|
|
non_suitable_extractors = list_non_suitable_extractors(0)
|
||
|
|
|
||
|
|
extractors_dir = 'youtube_dl/extractor/'
|
||
|
|
|
||
|
|
with open(extractors_dir + '__init__.py') as f:
|
||
|
|
content = f.read()
|
||
|
|
for extractor in non_suitable_extractors:
|
||
|
|
content = re.sub(r'from \.' + extractor.__module__.split('.')[-1] + r' import (?:[A-Za-z0-9,\s]+|\([^\)]+\))\n', '', content)
|
||
|
|
|
||
|
|
with open(extractors_dir + '__init__.py', 'w') as f:
|
||
|
|
f.write(content)
|
||
|
|
|
||
|
|
with open(extractors_dir + 'generic.py') as f:
|
||
|
|
content = f.read()
|
||
|
|
for extractor in non_suitable_extractors:
|
||
|
|
content = re.sub(r'from \.' + extractor.__module__.split('.')[-1] + r' import (?:[A-Za-z0-9,\s]+|\([^\)]+\))\n', '', content)
|
||
|
|
content = re.sub(r'#[^#]+' + extractor.__module__.split('.')[-1] + r'[^#]+', '', content)
|
||
|
|
|
||
|
|
with open(extractors_dir + 'generic.py', 'w') as f:
|
||
|
|
f.write(content)
|
||
|
|
|
||
|
|
for extractor in non_suitable_extractors:
|
||
|
|
extractor_filename = extractors_dir + extractor.__module__.split('.')[-1] + '.py'
|
||
|
|
if os.path.isfile(extractor_filename):
|
||
|
|
os.remove(extractor_filename)
|