conf_task.py 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. import json
  2. from flask import Flask, request
  3. from flask import jsonify
  4. from common.db.mysql_help import MysqlHelper
  5. app = Flask(__name__)
  6. app.config['JSON_AS_ASCII'] = False
  7. # 只接受get方法访问
  8. @app.route("/v1/source/getinfo", methods=["GET"])
  9. def getSource():
  10. # 获取传入的params参数
  11. get_data = request.args.to_dict()
  12. fields = get_data.get('fields')
  13. # # 对参数进行操作
  14. sql = 'select source, task_type, spider_name from crawler_source'
  15. result = MysqlHelper.get_values(sql)
  16. if not result:
  17. return jsonify({'return_code': '200', 'result': [], 'message': 'no data'})
  18. source_list = list()
  19. for source, task_type, spider_name in result:
  20. data = dict(
  21. source=source,
  22. task_type=task_type,
  23. spider_name=spider_name
  24. )
  25. source_list.append(data)
  26. return jsonify({'return_code': '200', 'result': source_list})
  27. @app.route("/v1/task/insert", methods=["POST"])
  28. def inerttask():
  29. pass
  30. # 获取传入的参数
  31. # get_data = request.args.to_dict()
  32. # 传入的参数为bytes类型,需要转化成json
  33. # return json.dumps(return_dict, ensure_ascii=False)
  34. @app.route("/v1/task/gettask", methods=["GET"])
  35. def getTask():
  36. get_data = request.args.to_dict()
  37. sql = 'select task_id, task_name from crawler_task'
  38. result = MysqlHelper.get_values(sql)
  39. if not result:
  40. return jsonify({'return_code': '200', 'result': [], 'message': 'no data'})
  41. source_list = list()
  42. for task_id, task_name in result:
  43. data = dict(
  44. task_id=task_id,
  45. task_name=task_name,
  46. )
  47. source_list.append(data)
  48. return jsonify({'return_code': '200', 'result': source_list})
  49. @app.route("/v1/author/getuser", methods=["POST"])
  50. def createUser():
  51. data = request.form.get('author')
  52. print(eval(data))
  53. for i in eval(data):
  54. print(i)
  55. return jsonify({'data':data})
  56. if __name__ == "__main__":
  57. app.run(debug=True)