Johan Dahlin
5169a67966
All checks were successful
Deploy Quartz site to GitHub Pages / build (push) Successful in 2m12s
81 lines
2.0 KiB
Python
81 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import subprocess
|
|
import sys
|
|
import os
|
|
import glob
|
|
|
|
def get_clipboard():
|
|
result = subprocess.run(['pbpaste'], capture_output=True, text=True)
|
|
return result.stdout.strip()
|
|
|
|
def set_clipboard(text):
|
|
subprocess.run(['pbcopy'], input=text, text=True)
|
|
|
|
def get_url_from_dialog():
|
|
applescript = '''
|
|
display dialog "Enter YouTube URL:" default answer "" with title "Download Swedish Subtitles" buttons {"Cancel", "OK"} default button "OK"
|
|
set userInput to text returned of result
|
|
return userInput
|
|
'''
|
|
try:
|
|
result = subprocess.run(['osascript', '-e', applescript], capture_output=True, text=True, check=True)
|
|
return result.stdout.strip()
|
|
except subprocess.CalledProcessError:
|
|
return None
|
|
|
|
def clean_subtitles(srt_content):
|
|
lines = srt_content.strip().split('\n')
|
|
text_lines = []
|
|
seen_lines = set()
|
|
|
|
for line in lines:
|
|
line = line.strip()
|
|
if not line or line.isdigit() or '-->' in line:
|
|
continue
|
|
if line in seen_lines:
|
|
continue
|
|
seen_lines.add(line)
|
|
text_lines.append(line)
|
|
|
|
return ' '.join(text_lines)
|
|
|
|
def download_subtitles(url):
|
|
cmd = [
|
|
'yt-dlp',
|
|
'--write-auto-sub',
|
|
'--sub-lang', 'sv',
|
|
'--skip-download',
|
|
'--convert-subs', 'srt',
|
|
'-o', '%(id)s.%(ext)s',
|
|
url
|
|
]
|
|
|
|
subprocess.run(cmd, capture_output=True, text=True, check=True)
|
|
|
|
srt_files = glob.glob('*.sv.srt')
|
|
|
|
if srt_files:
|
|
subtitle_file = srt_files[0]
|
|
with open(subtitle_file, 'r', encoding='utf-8') as f:
|
|
raw_content = f.read()
|
|
cleaned_content = clean_subtitles(raw_content)
|
|
os.remove(subtitle_file)
|
|
return cleaned_content
|
|
|
|
sys.exit(1)
|
|
|
|
if __name__ == '__main__':
|
|
url = get_clipboard()
|
|
|
|
if not url or 'youtube.com' not in url and 'youtu.be' not in url:
|
|
url = get_url_from_dialog()
|
|
if not url:
|
|
sys.exit(1)
|
|
|
|
subtitles = download_subtitles(url)
|
|
set_clipboard(subtitles)
|
|
|
|
|
|
|