Skip to content

Commit c9c17ee

Browse files
committed
Chore: 完善下载功能
1 parent 4901ded commit c9c17ee

2 files changed

Lines changed: 27 additions & 0 deletions

File tree

updater/mian.go

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -168,6 +168,27 @@ func GetDownloadLink(md5hash string) string {
168168
return link
169169
}
170170

171+
func DownloadFile(url, filename string) error {
172+
resp, err := http.Get(url)
173+
if err != nil {
174+
fmt.Println("Error:", err.Error())
175+
return err
176+
}
177+
defer resp.Body.Close()
178+
f, err := os.Create(filename)
179+
if err != nil {
180+
fmt.Println("Error:", err.Error())
181+
return err
182+
}
183+
defer f.Close()
184+
_, err = io.Copy(f, resp.Body)
185+
if err != nil {
186+
fmt.Println("Error:", err.Error())
187+
return err
188+
}
189+
return nil
190+
}
191+
171192
func Update(CurrentVersion string) {
172193
NewVersion, _ := CheckUpdate()
173194
v1, err := version.NewVersion(NewVersion)
@@ -204,6 +225,11 @@ func Update(CurrentVersion string) {
204225
return
205226
}
206227
fmt.Println("Download link:", link)
228+
err := DownloadFile(link, k)
229+
if err != nil {
230+
fmt.Println("Error:", err)
231+
return
232+
}
207233
}
208234
}
209235

utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -412,6 +412,7 @@ def daily_answer(self):
412412
self.pause()
413413
continue
414414
else:
415+
# 正常填空题
415416
self.send_log("查看提示")
416417
click_result: TaskDetail = self.tasker.post_task("查看提示").wait().get()
417418
if not click_result.nodes:

0 commit comments

Comments
 (0)