|
@@ -6,17 +6,18 @@ import (
|
|
|
"mongodb"
|
|
|
qu "qfw/util"
|
|
|
"sort"
|
|
|
- util "spiderutil"
|
|
|
+ sp "spiderutil"
|
|
|
"strconv"
|
|
|
"strings"
|
|
|
"time"
|
|
|
- u "util"
|
|
|
+ "util"
|
|
|
)
|
|
|
|
|
|
type Task struct {
|
|
|
*xweb.Action
|
|
|
- taskList xweb.Mapper `xweb:"/center/newtask/tasklist"` //任务管理
|
|
|
- taskEdit xweb.Mapper `xweb:"/center/newtask/edit/(.*)"` //编辑任务
|
|
|
+ taskList xweb.Mapper `xweb:"/center/newtask/tasklist"` //任务管理
|
|
|
+ taskEdit xweb.Mapper `xweb:"/center/newtask/edit/(.*)"` //编辑任务
|
|
|
+ taskTagCode xweb.Mapper `xweb:"/center/newtask/tagcode"` //标记爬虫
|
|
|
}
|
|
|
|
|
|
func (t *Task) TaskList() {
|
|
@@ -32,16 +33,16 @@ func (t *Task) TaskList() {
|
|
|
draw, _ := t.GetInteger("draw")
|
|
|
start, _ := t.GetInteger("start")
|
|
|
limit, _ := t.GetInteger("length")
|
|
|
- if auth == u.Role_Admin {
|
|
|
+ if auth == util.Role_Admin {
|
|
|
if t.Method() == "GET" {
|
|
|
events := []string{}
|
|
|
- for k, _ := range util.Config.Uploadevents {
|
|
|
+ for k, _ := range sp.Config.Uploadevents {
|
|
|
events = append(events, k)
|
|
|
}
|
|
|
sort.Strings(events)
|
|
|
t.T["events"] = events
|
|
|
t.T["modifyusers"] = getModifyUsers()
|
|
|
- t.Render("newtask.html", &t.T)
|
|
|
+ t.Render("newtasklist.html", &t.T)
|
|
|
} else {
|
|
|
query := queryCriteria(userid, platform, taskState, event, stype)
|
|
|
if search != "" {
|
|
@@ -67,15 +68,15 @@ func (t *Task) TaskList() {
|
|
|
sort = `{` + sortb + `,"l_complete":1}`
|
|
|
}
|
|
|
qu.Debug("query:", query, sort)
|
|
|
- task, _ := u.MgoEB.Find("newtask", query, sort, nil, false, start, limit)
|
|
|
- count := u.MgoEB.Count("newtask", query)
|
|
|
+ task, _ := util.MgoEB.Find("newtask", query, sort, nil, false, start, limit)
|
|
|
+ count := util.MgoEB.Count("newtask", query)
|
|
|
page := start / 10
|
|
|
if len(*task) > 0 {
|
|
|
for k, v := range *task {
|
|
|
v["num"] = k + 1 + page*10
|
|
|
- v["encode"] = util.Se.Encode2Hex(fmt.Sprint(v["s_code"]))
|
|
|
+ v["encode"] = sp.Se.Encode2Hex(fmt.Sprint(v["s_code"]))
|
|
|
//根据code查询luaconfig
|
|
|
- lua, _ := u.MgoEB.FindOneByField("luaconfig", map[string]interface{}{"code": qu.ObjToString(v["s_code"])}, `{"href":1}`)
|
|
|
+ lua, _ := util.MgoEB.FindOneByField("luaconfig", map[string]interface{}{"code": qu.ObjToString(v["s_code"])}, `{"href":1}`)
|
|
|
if len(*lua) > 0 {
|
|
|
v["href"] = (*lua)["href"]
|
|
|
} else {
|
|
@@ -100,7 +101,7 @@ func (t *Task) TaskEdit(ids string) error {
|
|
|
query := map[string]interface{}{
|
|
|
"_id": mongodb.StringTOBsonId(id),
|
|
|
}
|
|
|
- task, _ := u.MgoEB.FindOne("newtask", query)
|
|
|
+ task, _ := util.MgoEB.FindOne("newtask", query)
|
|
|
if task != nil && len(*task) > 0 {
|
|
|
(*task)["l_comeintime"] = time.Unix((*task)["l_comeintime"].(int64), 0).Format("2006-01-02 15:04:05")
|
|
|
(*task)["l_complete"] = time.Unix((*task)["l_complete"].(int64), 0).Format("2006-01-02 15:04:05")
|
|
@@ -121,7 +122,7 @@ func (t *Task) TaskEdit(ids string) error {
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
- t.T["encode"] = util.Se.Encode2Hex(fmt.Sprint((*task)["s_code"]))
|
|
|
+ t.T["encode"] = sp.Se.Encode2Hex(fmt.Sprint((*task)["s_code"]))
|
|
|
t.T["id"] = id
|
|
|
t.T["task"] = *task
|
|
|
t.T["param"] = param
|
|
@@ -131,11 +132,11 @@ func (t *Task) TaskEdit(ids string) error {
|
|
|
t.T["xgTime"] = qu.ObjToString(t.GetSession(id))
|
|
|
}
|
|
|
t.DelSession(id)
|
|
|
- if auth == u.Role_Admin {
|
|
|
+ if auth == util.Role_Admin {
|
|
|
return t.Render("taskedit.html", &t.T)
|
|
|
- } else if auth == u.Role_Dev {
|
|
|
+ } else if auth == util.Role_Dev {
|
|
|
return t.Render("mytaskedit.html", &t.T)
|
|
|
- } else if auth == u.Role_Examine {
|
|
|
+ } else if auth == util.Role_Examine {
|
|
|
return t.Render("auditedit.html", &t.T)
|
|
|
}
|
|
|
}
|
|
@@ -143,12 +144,39 @@ func (t *Task) TaskEdit(ids string) error {
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
+func (t *Task) TaskTagCode() {
|
|
|
+ defer qu.Catch()
|
|
|
+ taskId := t.GetString("id")
|
|
|
+ code := t.GetString("code")
|
|
|
+ stype := t.GetString("stype")
|
|
|
+ qu.Debug(taskId, code, stype)
|
|
|
+ //关闭任务
|
|
|
+ update := map[string]interface{}{
|
|
|
+ "$set": map[string]interface{}{
|
|
|
+ "i_state": 6,
|
|
|
+ "l_checktime": time.Now().Unix(),
|
|
|
+ },
|
|
|
+ }
|
|
|
+ util.MgoEB.UpdateById("newtask", taskId, update)
|
|
|
+ //标记爬虫
|
|
|
+ /*
|
|
|
+ tags:{任务类型:标记时间}
|
|
|
+ */
|
|
|
+ update = map[string]interface{}{
|
|
|
+ "$set": map[string]interface{}{
|
|
|
+ "codetags." + stype: time.Now().Unix(),
|
|
|
+ },
|
|
|
+ }
|
|
|
+ util.MgoEB.Update("luaconfig", map[string]interface{}{"code": code}, update, false, false)
|
|
|
+ t.Write("y")
|
|
|
+}
|
|
|
+
|
|
|
func getModifyUsers() []map[string]interface{} {
|
|
|
query := map[string]interface{}{
|
|
|
"i_auth": 1,
|
|
|
"i_delete": 0,
|
|
|
}
|
|
|
- user, _ := u.MgoEB.Find("user", query, nil, nil, false, -1, -1)
|
|
|
+ user, _ := util.MgoEB.Find("user", query, nil, nil, false, -1, -1)
|
|
|
return *user
|
|
|
}
|
|
|
|