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')