CodeMirror/ArchiveBox
1
0
Fork 0
mirror of synced 2024-06-22 16:10:54 +12:00
Code Issues Projects Releases Wiki Activity

add video subtitles and description to full-text index

Browse source
This commit is contained in:
Nick Sweeting 2021-04-10 07:22:20 -04:00
parent 1e3c1e3c83
commit 193df5c8d3
3 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
archivebox/config.py
View file

@ -126,6 +126,7 @@ CONFIG_SCHEMA: Dict[str, ConfigDefaultDict] = {
'--write-sub',
'--all-subs',
'--write-auto-sub',
'--convert-subs=srt',
'--yes-playlist',
'--continue',
'--ignore-errors',

2
archivebox/core/models.py
View file

@ -41,7 +41,7 @@ class Tag(models.Model):
Based on django-taggit model
"""
id = models.AutoField(primary_key=True, serialize=False, verbose_name='ID')
name = models.CharField(unique=True, blank=False, max_length=100)
# slug is autoset on save from name, never set it manually

13
archivebox/extractors/media.py
View file

@ -70,11 +70,24 @@ def save_media(link: Link, out_dir: Optional[Path]=None, timeout: int=MEDIA_TIME
finally:
timer.end()
# add video description and subtitles to full-text index
index_texts = [
text_file.read_text(encoding='utf-8').strip()
for text_file in (
*output_path.glob('*.description'),
*output_path.glob('*.srt'),
*output_path.glob('*.vtt'),
*output_path.glob('*.lrc'),
*output_path.glob('*.lrc'),
)
]
return ArchiveResult(
cmd=cmd,
pwd=str(out_dir),
cmd_version=YOUTUBEDL_VERSION,
output=output,
status=status,
index_texts=index_texts,
**timer.stats,
)