local
/
youtuber
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Set metadata and fix dl path

master v0.3
bel 2020-02-01 20:03:47 +00:00
parent cb0eea6140
commit 740f02bf6b
3 changed files with 45 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.go
View File

@ -57,7 +57,7 @@ func do(client *youtubedl.Client) error {
log.Printf("already exists: %s", target)
continue
}
if err := client.Download(result.Link, path.Join(config.Root, target)); err != nil {
if err := client.Download(result.Link, target); err != nil {
errs += ", " + err.Error()
}
}

40
youtubedl/client.go
View File

@ -15,15 +15,41 @@ func New() (*Client, error) {
}
func (c *Client) Download(video, local string) error {
cmd := exec.Command("youtube-dl", "-f", "22", "-o", local, "--write-sub", "--write-auto-sub", "--sub-format", "srt", video)
log.Println(cmd.Path, cmd.Args)
out, err := cmd.CombinedOutput()
if err != nil {
err = fmt.Errorf("%v: %s", err, out)
errs := []error{}
for _, extras := range [][]string{
[]string{"--add-metadata", "--metadata-from-title", ".*[0-9]/[0-9][0-9] (?P<title>.+)"},
[]string{},
} {
args := append([]string{
"youtube-dl",
"-f",
"22",
"-o",
local,
"--write-sub",
"--write-auto-sub",
"--sub-format",
"srt",
}, extras...)
args = append(args, video)
cmd := exec.Command(args[0], args[1:]...)
log.Println(cmd.Path, cmd.Args)
out, err := cmd.CombinedOutput()
if err != nil {
log.Printf("failed with %v: %s", err, out)
errs = append(errs, fmt.Errorf("%v: %s", err, out))
} else {
log.Printf("passed with %s", out)
errs = nil
break
}
}
if len(errs) > 0 {
err := fmt.Errorf("%v", errs)
return err
}
cmd = exec.Command("python3", "-m", "vtt_to_srt", path.Dir(local))
out, err = cmd.CombinedOutput()
cmd := exec.Command("python3", "-m", "vtt_to_srt", path.Dir(local))
out, err := cmd.CombinedOutput()
if err != nil {
err = fmt.Errorf("%v: %s", err, out)
return err

20
youtubedl/install.go
View File

@ -7,7 +7,7 @@ import (
)
func install() error {
if err := installPyPip3(); err != nil {
if err := installPyPip3FFMPEG(); err != nil {
return err
}
if err := installYtdl(); err != nil {
@ -19,21 +19,23 @@ func install() error {
return nil
}
func installPyPip3() error {
func installPyPip3FFMPEG() error {
hasPy := exec.Command("python3", "--version")
hasPip := exec.Command("pip3", "--version")
hasFFMPEG := exec.Command("which", "ffmpeg")
if err := hasPy.Run(); err != nil {
} else if err := hasPip.Run(); err != nil {
} else if err := hasFFMPEG.Run(); err != nil {
} else {
return nil
}
for _, combo := range [][]string{
[]string{"sudo", "apt", "install", "python3", "pip3"},
[]string{"apt", "install", "python3", "pip3"},
[]string{"apk", "add", "python3", "py3-pip"},
[]string{"sudo", "apk", "add", "python3", "py3-pip"},
[]string{"/sbin/apk", "add", "python3", "py3-pip"},
[]string{"sudo", "/sbin/apk", "add", "python3", "py3-pip"},
[]string{"sudo", "apt", "install", "python3", "pip3", "ffmpeg"},
[]string{"apt", "install", "python3", "pip3", "ffmpeg"},
[]string{"apk", "add", "python3", "py3-pip", "ffmpeg"},
[]string{"sudo", "apk", "add", "python3", "py3-pip", "ffmpeg"},
[]string{"/sbin/apk", "add", "python3", "py3-pip", "ffmpeg"},
[]string{"sudo", "/sbin/apk", "add", "python3", "py3-pip", "ffmpeg"},
} {
cmd := exec.Command(combo[0], combo[1:]...)
err := cmd.Run()
@ -42,7 +44,7 @@ func installPyPip3() error {
}
log.Println("%v: %v", err, combo)
}
return errors.New("cannot get python3 and pip3")
return errors.New("cannot get python3 and pip3 and ffmpeg")
}
func installYtdl() error {