ai_callback_add_group_handler.py 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. import json
  2. import traceback
  3. import tornado.web
  4. from db_config import *
  5. class AiCallBackAddGroup(tornado.web.RequestHandler):
  6. def initialize(self, logger):
  7. self.logger = logger
  8. def post(self):
  9. res = {'code': 0,
  10. 'message': "SUCCESS"}
  11. data = self.request.body
  12. data = str(data, 'utf8')
  13. data = json.loads(data, encoding='utf8')
  14. self.logger.info("*************************************** NEW REQUEST ***************************************")
  15. self.logger.info("raw data from request is %s" % data)
  16. try:
  17. inst = UpdateGroupLevelRecordTable(data, self.logger)
  18. inst.update_table_info()
  19. except Exception:
  20. res['code'] = -1
  21. res['message'] = 'FAIL'
  22. self.logger.error(traceback.format_exc())
  23. # ·µ»Ø½Ó¿Ú½á¹û
  24. self.write(json.dumps(res))
  25. self.flush()
  26. class UpdateGroupLevelRecordTable(object):
  27. def __init__(self, request_data, logger):
  28. self.data = request_data
  29. self.logger = logger
  30. def update_table_info(self):
  31. db_con = pymysql.connect(host=host,
  32. port=port,
  33. user=user,
  34. password=passwd,
  35. database=db,
  36. charset=charset)
  37. cursor = db_con.cursor()
  38. for item in self.data['callbackData']:
  39. if item['code'] == 0:
  40. group_uuid = item['group_uuid']
  41. unit_id = item['unit_id']
  42. unit_create_time = item['group_create_time']
  43. self.logger.info("on single update callback info: %s" % item)
  44. cursor.execute("""UPDATE ctop_ai_kuaishou_unit_level_operation_record
  45. SET unit_id= %s, unit_create_time=%s
  46. WHERE group_uuid = %s""", (unit_id, unit_create_time, group_uuid))
  47. else:
  48. self.logger.error("on single update callback info: %s" % item)
  49. db_con.commit()
  50. db_con.close()