Compare commits
3 Commits
Author | SHA1 | Date | |
---|---|---|---|
752dce9751 | |||
8d11b58cec | |||
35694eaed2 |
10
README.md
10
README.md
@@ -1,3 +1,11 @@
|
|||||||
# ytbatchdownloader
|
# ytbatchdownloader
|
||||||
|
|
||||||
A simple Youtube downloader to load multiple videos.
|
A simple Youtube downloader to load multiple videos.
|
||||||
|
|
||||||
|
## Credits
|
||||||
|
|
||||||
|
Used libraries:
|
||||||
|
|
||||||
|
[youtube downloader](https://github.com/kkdai/youtube)
|
||||||
|
|
||||||
|
[uiprogress](https://github.com/gosuri/uiprogress)
|
||||||
|
24
download.go
24
download.go
@@ -10,6 +10,7 @@ import (
|
|||||||
"sync"
|
"sync"
|
||||||
//"path/filepath"
|
//"path/filepath"
|
||||||
|
|
||||||
|
"github.com/gosuri/uiprogress"
|
||||||
. "github.com/kkdai/youtube"
|
. "github.com/kkdai/youtube"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -36,6 +37,11 @@ func (b *Batch) Start() {
|
|||||||
//That should be enough for yt links.
|
//That should be enough for yt links.
|
||||||
scanner := bufio.NewScanner(linkfile)
|
scanner := bufio.NewScanner(linkfile)
|
||||||
|
|
||||||
|
//Start rendering for progressbar
|
||||||
|
//This is the initialization the actual bars
|
||||||
|
//will be instantiated in the go routines
|
||||||
|
uiprogress.Start()
|
||||||
|
|
||||||
var wg sync.WaitGroup
|
var wg sync.WaitGroup
|
||||||
|
|
||||||
for scanner.Scan() {
|
for scanner.Scan() {
|
||||||
@@ -67,22 +73,28 @@ func downloadVideo(destDir, url string, wg *sync.WaitGroup) {
|
|||||||
re := regexp.MustCompile("(mp4|webm|3gpp)")
|
re := regexp.MustCompile("(mp4|webm|3gpp)")
|
||||||
fileext := re.FindString(y.StreamList[0]["type"])
|
fileext := re.FindString(y.StreamList[0]["type"])
|
||||||
filename := fmt.Sprintf("%s/%s.%s", destDir, y.StreamList[0]["title"], fileext)
|
filename := fmt.Sprintf("%s/%s.%s", destDir, y.StreamList[0]["title"], fileext)
|
||||||
|
filenameRune := []rune(y.StreamList[0]["title"])
|
||||||
|
shortFilename := string(filenameRune[0:20]) + "..."
|
||||||
|
|
||||||
log.Println("Start downloading:", filename)
|
|
||||||
go func() {
|
go func() {
|
||||||
|
bar := uiprogress.AddBar(100).AppendCompleted()
|
||||||
|
|
||||||
|
//Custom decorator (the short name in front of the bar)
|
||||||
|
bar.PrependFunc(func(b *uiprogress.Bar) string {
|
||||||
|
return shortFilename
|
||||||
|
})
|
||||||
|
|
||||||
var i int64 = 0
|
var i int64 = 0
|
||||||
for i < 100 {
|
for i < 100 {
|
||||||
i = <-y.DownloadPercent
|
i = <-y.DownloadPercent
|
||||||
if i%10 == 0 {
|
bar.Set(int(i))
|
||||||
log.Println(i, "%", y.StreamList[0]["title"])
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
|
//start the actual download
|
||||||
err := y.StartDownload(filename)
|
err := y.StartDownload(filename)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Println("Failed to download", url)
|
log.Println("Failed to download", url)
|
||||||
log.Println("Message:", err.Error())
|
log.Println("Message:", err.Error())
|
||||||
} else {
|
|
||||||
log.Println("Finished downloading:", filename)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user