From 7272263cff460649a8343b52c586f367830b2f79 Mon Sep 17 00:00:00 2001
From: cdle <798731886@qq.com>
Date: Tue, 17 Aug 2021 15:12:18 +0800
Subject: [PATCH] x

---
 models/task.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/models/task.go b/models/task.go
index 49dc217..85ac0e6 100644
--- a/models/task.go
+++ b/models/task.go
@@ -75,6 +75,7 @@ func runTask(task *Task, msgs ...interface{}) string {
 		f, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0777)
 		if err != nil {
 			logs.Warn("打开%s失败，", path, err)
+			return ""
 		}
 		url := task.Path
 		if strings.Contains(url, "raw.githubusercontent.com") {
@@ -86,6 +87,23 @@ func runTask(task *Task, msgs ...interface{}) string {
 		}
 		io.Copy(f, r.Body)
 		f.Close()
+	} else {
+		if task.Name != task.Path {
+			f, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0777)
+			if err != nil {
+				logs.Warn("打开%s失败，", path, err)
+				return ""
+			}
+			f2, err := os.Open(task.Path)
+			if err != nil {
+				f.Close()
+				logs.Warn("打开%s失败，", path, err)
+				return ""
+			}
+			io.Copy(f, f2)
+			f2.Close()
+			f.Close()
+		}
 	}
 	lan := Config.Node
 	if strings.Contains(task.Name, ".py") {