front.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941
  1. // front
  2. package front
  3. import (
  4. "fmt"
  5. "io/ioutil"
  6. "log"
  7. qu "qfw/util"
  8. mgdb "qfw/util/mongodb"
  9. "qfw/util/redis"
  10. "regexp"
  11. "sort"
  12. "spider"
  13. util "spiderutil"
  14. "strconv"
  15. "strings"
  16. "sync"
  17. "time"
  18. u "util"
  19. "github.com/go-xweb/httpsession"
  20. "github.com/go-xweb/xweb"
  21. "github.com/lauyoume/gopinyin"
  22. "github.com/tealeg/xlsx"
  23. "gopkg.in/mgo.v2/bson"
  24. )
  25. type Front struct {
  26. *xweb.Action
  27. login xweb.Mapper `xweb:"/"`
  28. logout xweb.Mapper `xweb:"/center/logout"` //退出
  29. loadIndex xweb.Mapper `xweb:"/center"` //控制中心
  30. spidernew xweb.Mapper `xweb:"/center/spider"` //爬虫保存
  31. reg xweb.Mapper `xweb:"/center/reg"` //爬虫注册
  32. assign xweb.Mapper `xweb:"/center/user/assign"` //分配爬虫
  33. loadSpider xweb.Mapper `xweb:"/center/spider/edit/(.*)"` //爬虫加载
  34. viewSpider xweb.Mapper `xweb:"/center/spider/view/(.*)"` //爬虫查看
  35. downSpider xweb.Mapper `xweb:"/center/spider/download/(.*)"` //爬虫下载
  36. upState xweb.Mapper `xweb:"/center/spider/upstate"` //爬虫状态更新
  37. assort xweb.Mapper `xweb:"/center/spider/assort"` //审核人员分类
  38. batchShelves xweb.Mapper `xweb:"/center/spider/batchShelves"` //爬虫状态更新
  39. checktime xweb.Mapper `xweb:"/center/spider/checktime"` //爬虫核对
  40. disables xweb.Mapper `xweb:"/center/spider/disable"` //批量作废
  41. changeEvent xweb.Mapper `xweb:"/center/changeEvent"` //节点更新
  42. getJson xweb.Mapper `xweb:"/center/spider/json"` //
  43. delRedis xweb.Mapper `xweb:"/center/spider/delRedis"` //清理Redis
  44. updateEventOrState xweb.Mapper `xweb:"/center/spider/updateeventorstate"` //修改爬虫的节点和状态
  45. spiderModel xweb.Mapper `xweb:"/center/model"` //获取补充模型
  46. runStep xweb.Mapper `xweb:"/center/run"` //方法测试
  47. spiderPass xweb.Mapper `xweb:"/center/spider/pass"` //整体测试
  48. runPinYin xweb.Mapper `xweb:"/center/runpy"` //获取拼音
  49. saveStep xweb.Mapper `xweb:"/center/save"` //保存脚本
  50. loadModel xweb.Mapper `xweb:"/center/gmodel/(.*)"` //加载模型
  51. importdata xweb.Mapper `xweb:"/center/importdata"` //导入脚本
  52. importfile xweb.Mapper `xweb:"/center/importfile"` //批量导入爬虫
  53. oldedit xweb.Mapper `xweb:"/center/oldedit"` //老文件编辑
  54. findName xweb.Mapper `xweb:"/center/findname"` //即时查询名称
  55. checkrepeat xweb.Mapper `xweb:"/center/spider/isrepeat"` //脚本代码判重
  56. Base Base
  57. OtherBase OtherBase
  58. Step1 Step1
  59. Step2 Step2
  60. Step3 Step3
  61. StepRe3 StepRe3
  62. U U
  63. luaList xweb.Mapper `xweb:"/center/lualist.html"` //脚本管理
  64. user xweb.Mapper `xweb:"/center/user.html"` //用户管理
  65. delUser xweb.Mapper `xweb:"/center/user/del"` //删除用户
  66. updateUser xweb.Mapper `xweb:"/center/user/updateUser"` //修改用户信息
  67. checkUsenamer xweb.Mapper `xweb:"/center/user/checkUsenamer"` //校验用户名的唯一性
  68. checkEmail xweb.Mapper `xweb:"/center/user/checkEmail"` //校验邮箱的唯一性
  69. saveNewUser xweb.Mapper `xweb:"/center/user/saveNewUser"` //添加用户
  70. getCity xweb.Mapper `xweb:"/center/getCity"` //获取城市
  71. }
  72. const role_admin, role_examine, role_dev = 3, 2, 1 //管理员,审核员,开发员
  73. const Sp_state_0, Sp_state_1, Sp_state_2, Sp_state_3, Sp_state_4, Sp_state_5, Sp_state_6, Sp_state_7 = 0, 1, 2, 3, 4, 5, 6, 7 //0待完成,1待审核,2打回,3发布,4作废,5已上架,6已下架,7其他
  74. var spinfos sync.Map = sync.Map{}
  75. var SessMap map[string]*httpsession.Session
  76. var AutoTpl map[string]interface{}
  77. var Mails *util.Mail
  78. var Reg = regexp.MustCompile(`(http|https)://([\w]+\.)+[\w]+`)
  79. var Transfercode map[string]interface{}
  80. var LuaStateMap = map[int]string{
  81. 0: "待完成",
  82. 1: "待审核",
  83. 2: "未通过",
  84. 3: "已通过",
  85. 4: "已作废",
  86. 5: "已上架",
  87. 6: "已下架",
  88. 7: "无发布",
  89. 8: "需登录",
  90. 9: "无法处理",
  91. 10: "已删除",
  92. }
  93. func (f *Front) Login() error {
  94. username := f.GetString("username")
  95. password := f.GetString("password")
  96. f.SetSession("password", password)
  97. password = util.Se.EncodeString(password)
  98. query := bson.M{
  99. "s_name": username,
  100. "s_pass": password,
  101. }
  102. user := *mgdb.FindOne("user", query)
  103. if user != nil && user["i_delete"] == 0 {
  104. f.SetSession("userid", user["_id"].(bson.ObjectId).Hex())
  105. f.SetSession("username", user["s_fullname"])
  106. f.SetSession("loginuser", user["s_name"])
  107. f.SetSession("email", user["s_email"])
  108. f.SetSession("auth", user["i_auth"])
  109. comeintime := time.Unix(user["l_comeintime"].(int64), 0).Format("2006-01-02")
  110. f.SetSession("comeintime", comeintime)
  111. if qu.IntAll(user["i_auth"]) > role_admin {
  112. return f.Redirect("/center/user.html")
  113. } else if qu.IntAll(user["i_auth"]) == role_dev {
  114. return f.Redirect("/center/mytask")
  115. } else {
  116. return f.Redirect("/center")
  117. }
  118. } else {
  119. if username != "" {
  120. f.T["fail"] = "fail"
  121. }
  122. return f.Render("login.html", &f.T)
  123. }
  124. }
  125. //用户管理
  126. func (f *Front) User() {
  127. if f.Method() == "POST" {
  128. auth := qu.IntAll(f.GetSession("auth"))
  129. if auth > role_admin {
  130. start, _ := f.GetInteger("start")
  131. limit, _ := f.GetInteger("length")
  132. draw, _ := f.GetInteger("draw")
  133. query := bson.M{
  134. "i_delete": 0, //可用用户
  135. "i_auth": bson.M{
  136. "$lt": auth,
  137. },
  138. }
  139. user := *mgdb.Find("user", query, nil, nil, false, start, limit)
  140. count := mgdb.Count("user", query)
  141. page := start / 10
  142. for k, v := range user {
  143. v["num"] = k + 1 + page*10
  144. v["l_comeintime"] = time.Unix(v["l_comeintime"].(int64), 0).Format("2006-01-02")
  145. v["s_pass"] = util.Se.DecodeString(v["s_pass"].(string))
  146. v["userid"] = v["_id"].(bson.ObjectId).Hex()
  147. }
  148. f.ServeJson(map[string]interface{}{
  149. "draw": draw,
  150. "data": user,
  151. "recordsFiltered": count,
  152. "recordsTotal": count,
  153. })
  154. }
  155. } else {
  156. f.Render("user.html")
  157. }
  158. }
  159. //删除用户
  160. func (f *Front) DelUser() {
  161. userid := f.GetString("userid")
  162. auth := qu.IntAll(f.GetSession("auth"))
  163. if auth > role_admin {
  164. query := bson.M{
  165. "_id": bson.ObjectIdHex(userid),
  166. }
  167. update := bson.M{
  168. "$set": bson.M{
  169. "i_delete": 1,
  170. },
  171. }
  172. ok := mgdb.Update("user", query, update, false, false)
  173. if ok {
  174. f.ServeJson(map[string]interface{}{
  175. "status": "y",
  176. })
  177. } else {
  178. f.ServeJson(map[string]interface{}{
  179. "status": "n",
  180. })
  181. }
  182. } else {
  183. f.ServeJson(map[string]interface{}{
  184. "status": "e",
  185. })
  186. }
  187. }
  188. //修改用户
  189. func (f *Front) UpdateUser() {
  190. // username := f.GetString("username")
  191. password := f.GetString("password")
  192. userid := f.GetString("userid")
  193. f.SetSession("password", password)
  194. userAuth := f.GetString("userAuth")
  195. auth := qu.IntAll(f.GetSession("auth"))
  196. self := f.GetString("self")
  197. //log.Println("userid----:", userid, "username----:", username, "password----:", password, "auth----:", auth, "self----:", self)
  198. query := bson.M{
  199. "_id": bson.ObjectIdHex(userid),
  200. }
  201. update := bson.M{}
  202. if "y" == self { //修改个人信息 只修改个人密码
  203. password = util.Se.EncodeString(password)
  204. update = bson.M{
  205. "$set": bson.M{
  206. "s_pass": password,
  207. },
  208. }
  209. } else if "y" != self && auth == 4 { //修改他人信息 只修改他人权限
  210. if userAuth == "开发员" || userAuth == "审核员" || userAuth == "管理员" {
  211. switch userAuth {
  212. case "开发员":
  213. update = bson.M{
  214. "$set": bson.M{
  215. "i_auth": 1,
  216. },
  217. }
  218. case "审核员":
  219. update = bson.M{
  220. "$set": bson.M{
  221. "i_auth": 2,
  222. },
  223. }
  224. case "管理员":
  225. update = bson.M{
  226. "$set": bson.M{
  227. "i_auth": 3,
  228. },
  229. }
  230. }
  231. }
  232. } else {
  233. f.ServeJson("没有权限!")
  234. return
  235. }
  236. ok := mgdb.Update("user", query, update, false, false)
  237. if ok {
  238. f.ServeJson(map[string]interface{}{
  239. "status": "y",
  240. })
  241. } else {
  242. f.ServeJson(map[string]interface{}{
  243. "status": "n",
  244. })
  245. }
  246. }
  247. func (f *Front) CheckUsenamer() {
  248. username := f.GetString("username")
  249. if username != "" {
  250. query := bson.M{
  251. "s_name": username,
  252. }
  253. user := *mgdb.FindOne("user", query)
  254. if user != nil {
  255. f.ServeJson(map[string]interface{}{
  256. "status": "hasUser",
  257. })
  258. } else {
  259. f.ServeJson(map[string]interface{}{
  260. "status": "notHasUser",
  261. })
  262. }
  263. }
  264. }
  265. func (f *Front) CheckEmail() {
  266. email := f.GetString("email")
  267. //校验邮箱
  268. if email != "" {
  269. query := bson.M{
  270. "s_email": email,
  271. }
  272. user := *mgdb.FindOne("user", query)
  273. if user != nil {
  274. f.ServeJson(map[string]interface{}{
  275. "status": "hasEmail",
  276. })
  277. } else {
  278. f.ServeJson(map[string]interface{}{
  279. "status": "notHasEmail",
  280. })
  281. }
  282. }
  283. }
  284. //新增用户
  285. func (f *Front) SaveNewUser() {
  286. auth := qu.IntAll(f.GetSession("auth"))
  287. if auth != 4 {
  288. f.ServeJson("没有权限!")
  289. return
  290. } else {
  291. i_auth := 1
  292. username := f.GetString("username")
  293. password := f.GetString("password")
  294. relname := f.GetString("relname")
  295. password = util.Se.EncodeString(password)
  296. email := f.GetString("email")
  297. userAuth := f.GetString("userAuth")
  298. if userAuth == "开发员" {
  299. i_auth = 1
  300. } else if userAuth == "审核员" {
  301. i_auth = 2
  302. } else {
  303. i_auth = 3
  304. }
  305. time := time.Now().Unix()
  306. save := bson.M{
  307. "s_name": username,
  308. "s_fullname": relname,
  309. "s_email": email,
  310. "s_pass": password,
  311. "i_auth": i_auth,
  312. "i_delete": 0,
  313. "l_comeintime": time,
  314. }
  315. ok := mgdb.Save("user", save)
  316. if ok != "" {
  317. f.ServeJson(map[string]interface{}{
  318. "status": "y",
  319. })
  320. } else {
  321. f.ServeJson(map[string]interface{}{
  322. "status": "n",
  323. })
  324. }
  325. }
  326. }
  327. func (f *Front) Logout() {
  328. email := f.GetSession("email").(string)
  329. f.DelSession("username")
  330. f.DelSession("userid")
  331. f.DelSession("email")
  332. f.DelSession("user")
  333. f.DelSession("loginuser")
  334. f.DelSession("auth")
  335. delete(SessMap, email)
  336. f.Redirect("/center")
  337. }
  338. //控制中心
  339. func (f *Front) LoadIndex() {
  340. auth := qu.IntAll(f.GetSession("auth"))
  341. if f.Method() == "POST" {
  342. start, _ := f.GetInteger("start")
  343. limit, _ := f.GetInteger("length")
  344. draw, _ := f.GetInteger("draw")
  345. searchStr := f.GetString("search[value]")
  346. event, _ := f.GetInteger("taskEvent") //节点
  347. //searchN := strings.Replace(searchStr, " ", "", -1)
  348. //search := strings.Replace(searchN, "\n", "", -1)
  349. search := strings.TrimSpace(searchStr)
  350. state, _ := f.GetInteger("state")
  351. urgency, _ := f.GetInteger("urgency") //节点
  352. platform := f.GetString("platform")
  353. query := bson.M{}
  354. if event > -1 {
  355. query["event"] = event
  356. }
  357. if urgency > -1 {
  358. query["urgency"] = urgency
  359. }
  360. if platform != "-1" {
  361. query["platform"] = platform
  362. }
  363. if auth == role_examine { //审核员
  364. if state > -1 {
  365. query["state"] = state
  366. } else {
  367. query["state"] = Sp_state_1
  368. }
  369. if search != "" {
  370. query["$or"] = []interface{}{
  371. bson.M{"code": bson.M{"$regex": search}},
  372. bson.M{"createuser": bson.M{"$regex": search}},
  373. bson.M{"param_common.1": bson.M{"$regex": search}},
  374. }
  375. }
  376. } else if auth == role_dev { //开发员
  377. if state > -1 {
  378. query["state"] = state
  379. }
  380. query["createuserid"] = f.GetSession("userid")
  381. query["$or"] = []interface{}{
  382. bson.M{"code": bson.M{"$regex": search}},
  383. bson.M{"createuser": bson.M{"$regex": search}},
  384. bson.M{"param_common.1": bson.M{"$regex": search}},
  385. }
  386. } else { //管理员
  387. if state > -1 {
  388. query["state"] = state
  389. }
  390. query["$or"] = []interface{}{
  391. bson.M{"code": bson.M{"$regex": search}},
  392. bson.M{"createuser": bson.M{"$regex": search}},
  393. bson.M{"param_common.1": bson.M{"$regex": search}},
  394. }
  395. }
  396. sort := `{"%s":%d}`
  397. orderIndex := f.GetString("order[0][column]")
  398. orderName := f.GetString(fmt.Sprintf("columns[%s][data]", orderIndex))
  399. orderType := 1
  400. if f.GetString("order[0][dir]") != "asc" {
  401. orderType = -1
  402. }
  403. if orderName == "param_common" {
  404. orderName = orderName + ".1"
  405. }
  406. sort = fmt.Sprintf(sort, orderName, orderType)
  407. page := start / 10
  408. //log.Println("sort", sort, orderName)
  409. luas := *mgdb.Find("luaconfig", query, sort, list_fields, false, start, limit)
  410. count := mgdb.Count("luaconfig", query)
  411. for k, v := range luas {
  412. v["num"] = k + 1 + page*10
  413. if v["modifytime"] != nil {
  414. v["modifytime"] = time.Unix(v["modifytime"].(int64), 0).Format("2006-01-02 15:04:05")
  415. } else {
  416. v["modifytime"] = "-"
  417. }
  418. if v["modifyuser"] == nil {
  419. v["modifytime"] = "-"
  420. }
  421. v["encode"] = util.Se.Encode2Hex(fmt.Sprint(v["code"]))
  422. if v["event"] == nil { //节点
  423. v["event"] = 0
  424. }
  425. //v["state"] = LuaStateMap[qu.IntAll(v["state"])]
  426. }
  427. f.ServeJson(map[string]interface{}{"draw": draw, "data": luas, "recordsFiltered": count, "recordsTotal": count})
  428. } else {
  429. events := []string{}
  430. for k, _ := range util.Config.Uploadevents {
  431. events = append(events, k)
  432. }
  433. sort.Strings(events)
  434. f.T["events"] = events
  435. f.Render("index.html", &f.T)
  436. }
  437. }
  438. func (f *Front) Checkrepeat() {
  439. code := f.GetString("code")
  440. one := *mgdb.FindOne("luaconfig", bson.M{"code": code})
  441. if len(one) > 0 {
  442. f.ServeJson("y")
  443. } else {
  444. f.ServeJson("n")
  445. }
  446. }
  447. //新建
  448. func (f *Front) Spidernew() error {
  449. auth := qu.IntAll(f.GetSession("auth"))
  450. if auth != role_admin {
  451. return nil
  452. }
  453. copy := f.GetString("copy")
  454. if copy != "" {
  455. one := *mgdb.FindOne("luaconfig", bson.M{"code": copy})
  456. delete(one, "_id")
  457. delete(one, "code")
  458. base := one["param_common"].([]interface{})
  459. base[0] = ""
  460. base[1] = ""
  461. one["param_common"] = base
  462. f.T["lua"] = one
  463. }
  464. f.T["isflow"] = 1 //新建爬虫时,初始化isflow的值
  465. f.T["actiontext"] = "新建"
  466. f.T["restate"] = 4 //此处设置restate=4无意义,只为了页面不报错
  467. return f.Render("spideredit.html", &f.T)
  468. }
  469. //得到模型
  470. func (f *Front) SpiderModel() {
  471. f.ServeJson(util.Config.Model)
  472. }
  473. func (f *Front) RunPinYin() {
  474. word := f.GetString("word")
  475. str := gopinyin.Convert(word, true)
  476. f.Write(str)
  477. }
  478. type U struct {
  479. User string
  480. Name string
  481. Pwd string
  482. }
  483. func (f *Front) Reg() {
  484. }
  485. func (f *Front) Assort() {
  486. state, _ := f.GetInteger("state")
  487. code := f.GetString("code")
  488. sql := bson.M{
  489. "$set": bson.M{
  490. "state": state,
  491. "modifytime": time.Now().Unix(),
  492. },
  493. }
  494. queryT := bson.M{
  495. "code": code,
  496. }
  497. //下架爬虫
  498. lua := *mgdb.FindOne("luaconfig", queryT)
  499. upresult, err := spider.UpdateSpiderByCodeState(code, "6", qu.IntAll(lua["event"]))
  500. qu.Debug("下架爬虫:", code, upresult, err)
  501. if upresult && err == nil {
  502. //更新爬虫
  503. mgdb.Update("luaconfig", queryT, sql, false, false)
  504. //关闭任务
  505. query := bson.M{
  506. "s_code": code,
  507. }
  508. s := *mgdb.Find("task", query, nil, nil, false, -1, -1)
  509. if s != nil {
  510. var idArr []string
  511. for _, v := range s {
  512. idArr = append(idArr, v["_id"].(bson.ObjectId).Hex())
  513. }
  514. for _, id := range idArr {
  515. sql = bson.M{
  516. "$set": bson.M{
  517. "i_state": 6,
  518. },
  519. }
  520. mgdb.Update("task", bson.M{"_id": bson.ObjectIdHex(id)}, sql, false, false)
  521. }
  522. }
  523. }
  524. f.ServeJson(bson.M{"upresult": upresult})
  525. }
  526. func (f *Front) Importfile() {
  527. auth := qu.IntAll(f.GetSession("auth"))
  528. if auth != role_admin {
  529. f.ServeJson("没有权限")
  530. return
  531. }
  532. if f.Method() == "POST" {
  533. mf, _, err := f.GetFile("xlsx")
  534. errorinfo := map[string]interface{}{}
  535. if err == nil {
  536. binary, _ := ioutil.ReadAll(mf)
  537. xls, _ := xlsx.OpenBinary(binary)
  538. sheet := xls.Sheets[0]
  539. rows := sheet.Rows
  540. for k, v := range rows {
  541. if k != 0 {
  542. cells := v.Cells
  543. if cells[1].Value != "" {
  544. o := make(map[string]interface{})
  545. o["name"] = cells[0].Value
  546. o["code"] = cells[1].Value
  547. o["channel"] = cells[2].Value
  548. o["channeladdr"] = cells[3].Value
  549. o["author"] = cells[4].Value
  550. o["timestamp"] = time.Now().Unix()
  551. o["status"] = 1
  552. o["next"] = cells[4].Value
  553. o["event"] = cells[5].Value
  554. o["historyevent"] = cells[6].Value
  555. if cells[7].Value == "是" {
  556. o["isflow"] = 1
  557. } else {
  558. o["isflow"] = 0
  559. }
  560. if cells[8].Value == "紧急" {
  561. o["urgency"] = 1
  562. } else {
  563. o["urgency"] = 0
  564. }
  565. o["platform"] = cells[9].Value
  566. // table := cells[6].Value
  567. // o["table"] = table
  568. // o["transfercode"] = qu.IntAll(Transfercode[table])
  569. query := bson.M{"code": cells[1].Value}
  570. rs := *mgdb.FindOne("import", query)
  571. if len(rs) > 0 {
  572. errorinfo[cells[1].Value] = "第" + strconv.Itoa(k) + "行重复,已经过滤"
  573. } else {
  574. ok, name := pf(o) //保存爬虫
  575. if ok == false {
  576. errorinfo[cells[1].Value] = "第" + strconv.Itoa(k) + "行找不到作者,已经过滤"
  577. } else {
  578. o["author"] = name
  579. mgdb.Save("import", o)
  580. }
  581. }
  582. }
  583. }
  584. }
  585. f.ServeJson(errorinfo)
  586. } else {
  587. f.ServeJson(false)
  588. }
  589. }
  590. }
  591. func pf(o map[string]interface{}) (bool, string) {
  592. AutoTpl["Base.SpiderName"] = o["name"]
  593. AutoTpl["Base.SpiderCode"] = o["code"]
  594. AutoTpl["Base.SpiderChannel"] = o["channel"]
  595. AutoTpl["Base.SpiderTargetChannelUrl"] = o["channeladdr"]
  596. author := o["author"].(string)
  597. one := *mgdb.FindOne("user", bson.M{"s_email": author})
  598. id := one["_id"].(bson.ObjectId).Hex()
  599. if len(one) == 0 {
  600. return false, ""
  601. }
  602. common := []interface{}{
  603. AutoTpl["Base.SpiderCode"],
  604. AutoTpl["Base.SpiderName"],
  605. AutoTpl["Base.SpiderChannel"],
  606. AutoTpl["Base.SpiderDownDetailPage"],
  607. AutoTpl["Base.SpiderStartPage"],
  608. AutoTpl["Base.SpiderMaxPage"],
  609. AutoTpl["Base.SpiderRunRate"],
  610. //AutoTpl["Base.Spider2Collection"],
  611. "bidding", //爬虫导入新建默认为bidding
  612. AutoTpl["Base.SpiderPageEncoding"],
  613. AutoTpl["Base.SpiderStoreMode"],
  614. AutoTpl["Base.SpiderStoreToMsgEvent"],
  615. AutoTpl["Base.SpiderTargetChannelUrl"],
  616. AutoTpl["Base.SpiderLastDownloadTime"],
  617. AutoTpl["Base.SpiderIsHistoricalMend"],
  618. AutoTpl["Base.SpiderIsMustDownload"],
  619. }
  620. ptime := []interface{}{
  621. AutoTpl["Step1.DateFormat"],
  622. AutoTpl["Step1.Address"],
  623. AutoTpl["Step1.ContentChooser"],
  624. }
  625. list := []interface{}{
  626. AutoTpl["Step2.Listadd"],
  627. AutoTpl["Step2.Listadds"],
  628. AutoTpl["Step2.BlockChooser"],
  629. AutoTpl["Step2.AddressChooser"],
  630. AutoTpl["Step2.TitleChooser"],
  631. AutoTpl["Step2.DateChooser"],
  632. AutoTpl["Step2.DateFormat"],
  633. }
  634. content := []interface{}{
  635. AutoTpl["Step3.ContentChooser"],
  636. AutoTpl["Step3.ElementChooser"],
  637. }
  638. param := map[string]interface{}{}
  639. param["param_common"] = common
  640. param["param_common"] = common
  641. //向导模式
  642. param["param_time"] = ptime
  643. param["param_list"] = list
  644. param["param_content"] = content
  645. param["type_time"] = 0
  646. param["type_list"] = 0
  647. param["type_content"] = 0
  648. //专家模式
  649. param["str_time"] = ""
  650. param["str_list"] = ""
  651. param["str_content"] = ""
  652. param["comeintime"] = time.Now().Unix()
  653. param["code"] = o["code"]
  654. param["createuser"] = one["s_name"]
  655. param["createuserid"] = id
  656. param["createuseremail"] = one["s_email"]
  657. param["modifyuser"] = one["s_name"]
  658. param["modifyuserid"] = id
  659. param["modifytime"] = time.Now().Unix()
  660. param["state"] = 0 //未完成
  661. if qu.IntAll(o["event"]) > 0 {
  662. param["event"] = qu.IntAll(o["event"])
  663. }
  664. s_model := "bid"
  665. configModel := util.Config.Model[s_model]
  666. model := map[string]interface{}{}
  667. for k, _ := range configModel {
  668. model[k] = ""
  669. }
  670. param["model"] = model
  671. param["next"] = o["next"]
  672. param["urgency"] = o["urgency"]
  673. param["isflow"] = o["isflow"]
  674. param["spidertype"] = "history"
  675. historyevent := qu.ObjToString(o["historyevent"])
  676. if movevent, ok := util.Config.Uploadevents[historyevent].(string); ok && movevent != "" {
  677. param["spidermovevent"] = movevent
  678. } else {
  679. param["spidermovevent"] = "7700"
  680. }
  681. param["historyevent"] = qu.IntAll(o["historyevent"])
  682. param["spiderhistorymaxpage"] = 1
  683. param["platform"] = o["platform"]
  684. //qu.Debug("param---", param)
  685. issave := spider.SaveSpider(o["code"].(string), param)
  686. return issave, one["s_name"].(string)
  687. }
  688. func (f *Front) Importdata() {
  689. auth := qu.IntAll(f.GetSession("auth"))
  690. if auth == role_admin {
  691. if f.Method() == "GET" {
  692. f.Render("import.html")
  693. } else {
  694. rss := *mgdb.Find("import", nil, `{"timestamp": -1}`, nil, false, -1, -1)
  695. f.ServeJson(map[string]interface{}{
  696. "data": rss,
  697. })
  698. }
  699. } else {
  700. f.Write("您没有导入脚本的权限")
  701. }
  702. }
  703. func Wlog(name, code, man, manid, types string, content map[string]interface{}) {
  704. obj := bson.M{
  705. "name": name,
  706. "code": code,
  707. "man": man,
  708. "manid": manid,
  709. "types": types,
  710. "time": time.Now().Unix(),
  711. "content": content,
  712. }
  713. mgdb.Save("lua_logs", obj)
  714. }
  715. func (f *Front) Oldedit() {
  716. if f.Method() == "GET" {
  717. f.Render("oldedit.html")
  718. } else {
  719. }
  720. }
  721. func (f *Front) FindName() {
  722. words := f.GetString("words")
  723. if words == "" {
  724. f.ServeJson(bson.M{"error": "null"})
  725. }
  726. query := bson.M{"$or": []interface{}{
  727. bson.M{"param_common.0": bson.M{"$regex": words}},
  728. bson.M{"param_common.1": bson.M{"$regex": words}},
  729. bson.M{"createuser": bson.M{"$regex": words}},
  730. }, "oldlua": bson.M{"$exists": false}}
  731. rs := *mgdb.Find("luaconfig", query, bson.M{"modifytime": -1}, bson.M{"param_common": 1}, false, -1, -1)
  732. if len(rs) > 0 {
  733. f.ServeJson(bson.M{"data": rs})
  734. } else {
  735. f.ServeJson(bson.M{"error": "data"})
  736. }
  737. }
  738. //分配爬虫
  739. func (f *Front) Assign() {
  740. auth := qu.IntAll(f.GetSession("auth"))
  741. if auth != role_admin {
  742. f.Write("n")
  743. return
  744. }
  745. ids := f.GetString("ids")
  746. codes := f.GetString("codes")
  747. email := f.GetString("email")
  748. idarr := strings.Split(ids, ",")
  749. codesarr := strings.Split(codes, ",")
  750. var idsinter []interface{}
  751. isemail := strings.Index(ids, "@")
  752. user := *mgdb.FindOne("user", bson.M{"s_email": email})
  753. if len(user) > 0 {
  754. userid := user["_id"].(bson.ObjectId).Hex()
  755. name := user["s_name"].(string)
  756. var query bson.M
  757. if isemail > -1 {
  758. query = bson.M{
  759. "createuseremail": bson.M{
  760. "$in": idarr,
  761. },
  762. }
  763. } else {
  764. idsinter = make([]interface{}, len(idarr))
  765. for k, v := range idarr {
  766. idsinter[k] = bson.ObjectIdHex(v)
  767. }
  768. query = bson.M{
  769. "_id": bson.M{
  770. "$in": idsinter,
  771. },
  772. }
  773. }
  774. set := bson.M{
  775. "$set": bson.M{
  776. "createuserid": userid,
  777. "createuser": user["s_name"],
  778. "createuseremail": user["s_email"],
  779. "modifyuser": user["s_name"],
  780. "modifyuserid": userid,
  781. },
  782. }
  783. b := mgdb.Update("luaconfig", query, set, false, true)
  784. if b {
  785. f.Write("y")
  786. editModify(codesarr, userid, name) //分配
  787. } else {
  788. f.Write("n")
  789. }
  790. } else {
  791. f.Write("null")
  792. }
  793. }
  794. //修改维护人
  795. func editModify(codesarr []string, userid, name string) {
  796. //修改modifyid和modify
  797. for _, v := range codesarr {
  798. query := bson.M{
  799. "s_code": v,
  800. "i_state": bson.M{
  801. "$ne": 4,
  802. },
  803. }
  804. task := *mgdb.Find("task", query, nil, nil, false, -1, -1)
  805. if len(task) > 0 {
  806. for _, v := range task { //循环 修改任务
  807. update := bson.M{
  808. "$set": bson.M{
  809. "s_modify": name,
  810. "s_modifyid": userid,
  811. },
  812. }
  813. queryT := bson.M{
  814. "_id": v["_id"],
  815. }
  816. flag := mgdb.Update("task", queryT, update, false, false)
  817. log.Println("分配修改任务维护人:", flag)
  818. }
  819. } else {
  820. continue
  821. }
  822. }
  823. }
  824. //清理Redis
  825. func (f *Front) DelRedis() {
  826. hrefs := f.GetString("href")
  827. hrefsarr := strings.Split(hrefs, ",")
  828. auth := qu.IntAll(f.GetSession("auth"))
  829. err := []string{}
  830. if auth == role_admin { //权限控制
  831. if len(hrefsarr) > 0 {
  832. for k1, h := range hrefsarr {
  833. href := Reg.FindString(h)
  834. if href != "" {
  835. href = "url_repeat_" + href + "*"
  836. res := redis.GetKeysByPattern("title_repeat_judgement", href)
  837. if res != nil {
  838. for _, v := range res {
  839. hf := string(v.([]uint8))
  840. b := redis.Del("title_repeat_judgement", hf)
  841. if !b {
  842. err = append(err, "第"+strconv.Itoa(k1+1)+"个")
  843. }
  844. }
  845. }
  846. } else {
  847. err = append(err, "第"+strconv.Itoa(k1+1)+"个")
  848. }
  849. }
  850. }
  851. } else {
  852. err = append(err, "没有权限")
  853. }
  854. f.ServeJson(err)
  855. }
  856. func (f *Front) UpdateEventOrState() {
  857. val := f.GetString("val")
  858. w := f.GetString("w")
  859. id := f.GetString("id")
  860. query := map[string]interface{}{
  861. "_id": qu.StringTOBsonId(id),
  862. }
  863. set := map[string]interface{}{}
  864. update := map[string]interface{}{
  865. "$set": set,
  866. }
  867. if w == "state" { //修改状态为待完成
  868. set["state"] = 0
  869. } else { //修改节点
  870. event, _ := strconv.Atoi(val)
  871. set["event"] = event
  872. set["historyevent"] = event
  873. //state := f.GetString("s")
  874. //if state == "5" { //已上架状态改为下架
  875. code := f.GetString("c")
  876. set["state"] = 6
  877. b, err := UpStateAndUpSpider(code, "", "", "", Sp_state_6) //线上爬虫下架
  878. if !b || err != nil {
  879. f.Write("n")
  880. return
  881. }
  882. //}
  883. }
  884. if mgdb.Update("luaconfig", query, update, false, false) {
  885. log.Println("Id:", id, " Update", w, val, "Success")
  886. f.Write("y")
  887. } else {
  888. log.Println("Id:", id, " Update", w, val, "Failed")
  889. f.Write("n")
  890. }
  891. f.Write("n")
  892. }
  893. func (f *Front) GetCity() {
  894. area := f.GetString("area")
  895. cityArr := []string{}
  896. cityArr = u.Province[area]
  897. f.ServeJson(cityArr)
  898. }