Archives/telebot
2
0
Fork 0
mirror of https://github.com/tucnak/telebot synced 2024-11-03 09:40:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

add method GetFile and refactor code

Browse Source
This commit is contained in:
robatioor 2019-03-03 08:44:14 +03:30
parent e6cfaa8339
commit 5601dd0354
1 changed files with 25 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
bot.go
View File

@ -877,14 +877,12 @@ func (b *Bot) FileByID(fileID string) (File, error) {
// Download saves the file from Telegram servers locally.
//
// Maximum file size to download is 20 MB.
func (b *Bot) Download(f *File, localFilename string) error {
g, err := b.FileByID(f.FileID)
func (b *Bot) Download(file *File, localFilename string) error {
reader, err := b.GetFile(file)
if err != nil {
return err
return wrapSystem(err)
}
url := fmt.Sprintf("%s/file/bot%s/%s",
b.URL, b.Token, g.FilePath)
defer reader.Close()
out, err := os.Create(localFilename)
if err != nil {
@ -892,23 +890,34 @@ func (b *Bot) Download(f *File, localFilename string) error {
}
defer out.Close()
resp, err := http.Get(url)
_, err = io.Copy(out, reader)
if err != nil {
return wrapSystem(err)
}
defer resp.Body.Close()
_, err = io.Copy(out, resp.Body)
if err != nil {
return wrapSystem(err)
}
g.FileLocal = localFilename
*f = g
file.FileLocal = localFilename
return nil
}
// GetFile from Telegram servers
func (b *Bot) GetFile(file *File) (io.ReadCloser, error) {
f, err := b.FileByID(file.FileID)
if err != nil {
return nil, err
}
url := fmt.Sprintf("%s/file/bot%s/%s",
b.URL, b.Token, f.FilePath)
resp, err := http.Get(url)
if err != nil {
return nil, err
}
// set FilePath
*file = f
return resp.Body, nil
}
// StopLiveLocation should be called to stop broadcasting live message location
// before Location.LivePeriod expires.
//