def send_director(url, device_id, data_type, header, data, retract=0): """ 发送数据到指挥中心 :param url: :param device_id: 如果发送的是检测其数据,则为检测器ID,如果是管理中心数据则为管理中心ID,用户记录日志 :param data_type: 用于标识发送的数据类型 :param header: 请求头 :param data: 请求数据,可能是文件,如果是文件data格式是tuple(filename,filepath) :param retract 打印控制台缩进数,以"tab"为单位 :return: """ try: pu.print_format_header('发送至指挥中心', retract=retract) cookies = check_director_cookies() # 携带cookie print "header:", pu.pretty_print_format(header) print "data:", pu.pretty_print_format(json.loads(data)) if header.get('Data-Type') == 'file': boundary = '---------------------------' + (''.join(map(lambda xx: (hex(ord(xx))[2:]), os.urandom(6))))[0:10] header['Content-Type'] = 'multipart/form-data;boundary=' + boundary data = json.loads(data) print "###file config", data[0], data[1] data = MultipartEncoder( fields={ "file": (data[0], open(data[1], "rb"), "multipart/form-data") # 文件流 }, boundary=boundary ) # 文件实体对象 pu.print_with_retract('send %s type data to director' % data_type) resp = requests.post(url, headers=header, data=data, cookies=cookies) # pu.print_with_retract(resp.status_code, retract=retract+1) # 响应码 # pu.print_with_retract(resp.headers, retract=retract+1) # 响应头 pu.print_with_retract(resp.text.encode('utf-8'), retract=retract+1) # 响应消息正文 log_str = '%s %s %d %s' % (device_id, data_type, resp.status_code, resp.text.encode('utf-8')) fu.string2log_append_per_day(content=log_str, path=common.LOG_PATH + 'director_log/', prefix='send_director.', suffix='log') if resp.status_code == 400: # cookie失效或者不合法 pu.print_with_retract('管理中心的Cookie失效, 重新认证') res = detect_center_reg_auth.send_auth_login_request() if res['code'] == 200: resp = requests.post(url, headers=header, data=data, cookies=check_director_cookies()) # pu.print_with_retract(resp.status_code, retract=retract+1) # 响应码 # pu.print_with_retract(resp.headers, retract=retract+1) # 响应头 pu.print_with_retract(resp.text.encode('utf-8'), retract=retract+1) # 响应消息正文 pu.print_format_tail('发送至指挥中心', retract=retract) except Exception as e: # print e.args, "##", e.message traceback.print_exc() log_str = '%s %s %s' % (device_id, data_type, 'send_exception') fu.string2log_append_per_day(content=log_str, path=common.LOG_PATH + 'director_log/', prefix='send_director.', suffix='log')
def send_business_data(logger_str, flag, device_id, send_data, url=config.const.BUSINESS_URL): pu.print_format_header('\n发送json数据到业务系统') try: # logger = logging.getLogger(logger_str) for d in send_data: send_header = d[0] if config.const.OBTAIN_AUTH_KEY: print '获取业务系统授权Key' send_header['Authorization'] = common.get_auth_key() print 'Authorization_Key:', send_header['Authorization'] # send_header['Authorization'] = '96f9047831be4bb8a2bbd1417f58e94b' # send_header['Connection'] = 'keep-alive' print 'header:', pu.pretty_print_format(send_header), '\ndata:', d[1] print '发送%s数据到业务系统' % flag r = requests.post(url, headers=send_header, data=json.dumps(d[1])) # pu.print_with_retract(r.status_code, retract + 1) # 响应码 pu.print_with_retract(r.headers, 1) # 响应头 pu.print_with_retract(r.text.encode('utf-8'), 1) # 响应消息正文 log_str = '%s %s %d %s' % (device_id, flag, r.status_code, r.text.encode('utf-8')) fu.string2log_append_per_day(content=log_str, path=common.LOG_PATH + 'send_business/', prefix='send_business.', suffix='log') except Exception: traceback.print_exc() log_str = '%s %s %s' % (device_id, flag, 'send_exception') fu.string2log_append_per_day(content=log_str, path=common.LOG_PATH + 'send_business/', prefix='send_business.', suffix='log') pu.print_format_tail('发送json数据到业务系统\n')
def send_business_file(logger_str, flag, device_id, header, file_path, filename, url=config.const.BUSINESS_URL): pu.print_format_header('\n发送文件到业务系统') try: # logger = logging.getLogger(logger_str) if config.const.OBTAIN_AUTH_KEY: print '获取业务系统授权Key' header['Authorization'] = common.get_auth_key() print 'Authorization_Key:', header['Authorization'] header['Connection'] = 'keep-alive' # 长连接 boundary = '---------------------------' + (''.join(map(lambda xx: (hex(ord(xx))[2:]), os.urandom(6))))[0:10] header['Content-Type'] = 'multipart/form-data;boundary=' + boundary files = MultipartEncoder( fields={ "file": (filename, open(file_path, "rb"), "multipart/form-data") # 文件流 }, boundary=boundary ) # 文件实体对象 print 'header:', pu.pretty_print_format(header) print '发送%s数据到业务系统' % flag r = requests.post(url, headers=header, data=files) # r = requests.post(url, headers=header, files={'file': (filename, open(file_path, "rb"))}) # pu.print_with_retract(r.status_code, retract + 1) # 响应码 pu.print_with_retract(r.headers, 1) # 响应头 pu.print_with_retract(r.text.encode('utf-8'), 1) # 响应消息正文 log_str = '%s %s %d %s' % (device_id, flag, r.status_code, r.text.encode('utf-8')) fu.string2log_append_per_day(content=log_str, path=common.LOG_PATH + 'send_business/', prefix='send_business.', suffix='log') except Exception: traceback.print_exc() log_str = '%s %s %s' % (device_id, flag, 'send_exception') fu.string2log_append_per_day(content=log_str, path=common.LOG_PATH + 'send_business/', prefix='send_business.', suffix='log') pu.print_format_tail('发送文件到业务系统\n')