海峡

-import requests
-import time
-import oss2
-from common.log import *
-
-logger = BotLogger(os.path.basename(__file__), logname=1).getLogger()
-
-
-def handle_error(res, normal_status_code):
-    """
-    统一处理workai平台返回的error
-    :param res: 调用接口的返回
-    :param normal_status_code: 正常返回的status code
-    :return: 返回错误信息
-    """
-    if res is None or res.status_code == normal_status_code:
-        return ''
-
-    logger.info('The res is %s' % res)
-    err = '网络或服务器繁忙！'
-    try:
-        if 'message' in res.json():
-            err = res.json()['message']
-    except Exception as e:
-        logger.error(str(e))
-
-    return err
-
-
-def get_token_by_password(username, password, tenant_id, tenant_type):
-    """
-    密码方式获取access_token
-    :param username: 系统管理员用户
-    :param password: 系统管理员密码
-    :param tenant_id: 客户在workai云平台唯一表示
-    :return: 返回access_token, refresh_token 以及error, access_token用于访问接口的令牌, refresh_token
-    用于刷新获取新的access_token, error为返回的错误信息
-    """
-    data = {
-        'grant_type': 'password',
-        'username': username,
-        'password': password,
-        'scope': 'global_access:tenant_admin,tenant:%s,tenant_type:%s' % (tenant_id, tenant_type),
-    }
-
-    path = '%s/uaa/v1/auth/tokens' % WACLOUD_API_ENDPOINT
-    try:
-        res = requests.post(path, json=data, timeout=10)
-        err = handle_error(res, 200)
-        if err != '':
-            logger.error('Could not get the token by password for user: %s, tenant: %s' % (username, tenant_id))
-            return '', '', err, None
-
-        ret = res.json()
-        company, username, play_type = '', '', ''
-        if tenant_type == '0':
-            play_type = 'HRO'
-        elif tenant_type == '1':
-            play_type = 'HR'
-
-        try:
-            company = ret['tenant']['name']
-            username = ret['user']['name']
-        except Exception as e:
-            logger.error(str(e))
-
-        return ret['access_token'], ret['refresh_token'], '', (company, username, play_type)
-    except Exception as e:
-        logger.error(str(e))
-        return '', '', '网络或服务器异常，请稍后再试！', None
-
-
-def refresh_token(rtoken, tenant_id):
-    """
-    刷新token方式获取access_token
-    :return: 返回access_token, refresh_token 以及error, access_token用于访问接口的令牌, refresh_token
-    用于刷新获取新的access_token, error为返回的错误信息
-    """
-    data = {
-        'grant_type': 'refresh_token',
-        'refresh_token': rtoken,
-        'scope': 'global_access:tenant_admin,tenant:%s' % tenant_id,
-    }
-
-    path = '%s/uaa/v1/auth/tokens' % WACLOUD_API_ENDPOINT
-    res = requests.post(path, json=data)
-
-    err = handle_error(res, 200)
-    if err != '':
-        logger.error('Could not get the token by refresh token for tenant: %s' % (tenant_id,))
-        return '', '', err
-
-    ret = res.json()
-    return ret['access_token'], ret['refresh_token'], ''
-
-
-def upload_file(ctx, local_path, tenant_type, file_name):
-    """上传附件"""
-    object_path = ""
-    if ctx is None or local_path is None:
-        return 401, object_path
-
-    headers = {'Authorization': 'Bearer %s' % ctx['access_token']}
-    path = '%s/filemeta/v2/inits' % (WACLOUD_API_ENDPOINT,)
-    if tenant_type == "0":
-        data = {"object_type": "hro", "access_type": "sdk"}
-    elif tenant_type == "1":
-        data = {"object_type": "hr", "access_type": "sdk"}
-    else:
-        return 401, '配置文件设置tenant_type的值有误'
-
-    res = None
-    try:
-        res = requests.post(path, json=data, headers=headers, timeout=120)
-        if res.status_code == 401:
-            access_token, _refresh_token, err = refresh_token(ctx['refresh_token'], ctx['tenant_id'])
-            if err != '':
-                return 401, object_path
-            ctx['access_token'] = access_token
-            ctx['refresh_token'] = _refresh_token
-            headers = {'Authorization': 'Bearer %s' % ctx['access_token']}
-            res = requests.post(path, json=data, headers=headers, timeout=120)
-
-        err = handle_error(res, 200)
-        if err != '':
-            return res.status_code, object_path
-    except Exception as e:
-        logger.error(str(e))
-        return 401, str(e)
-
-    try:
-        datas = res.json()
-        # 使用临时token中的认证信息初始化StsAuth实例。
-        # auth = oss2.Auth(datas['access_key_id'], datas['access_key_secret'])
-        auth = oss2.StsAuth(datas['access_key_id'], datas['access_key_secret'], datas['security_token'])
-        # 使用StsAuth实例初始化存储空间。
-        bucket = oss2.Bucket(auth, datas['domain'], datas['bucket'])
-        object_path = datas['object_path'] + "/" + file_name
-        result = bucket.put_object_from_file(object_path, local_path)
-        return result.status, object_path
-    except Exception as e:
-        logger.error(str(e))
-        return 403, str(e)
-
-
-def update_object_path(ctx, tenant_type, object_path, types, file_name):
-    """上传文件路径到saas平台"""
-    if tenant_type == "0" or tenant_type == "1":
-        path = '%s/payroll/v1/do-import-data?type=%s' % (WACLOUD_API_ENDPOINT, types)
-    else:
-        return False, '平台类型设置有误'
-
-    headers = {'Authorization': 'Bearer %s' % ctx['access_token']}
-    bucket = '/hro/'
-    if 'hx' in WACLOUD_API_ENDPOINT:
-        bucket = '/hxhro/'
-    elif "xsb" in WACLOUD_API_ENDPOINT:
-        bucket = "/xsbhro/"
-    elif "engma" in WACLOUD_API_ENDPOINT:
-        bucket = "/engmahro/"
-    elif "anbo" in WACLOUD_API_ENDPOINT:
-        bucket = "/anbohro/"
-    elif "teda" in WACLOUD_API_ENDPOINT:
-        bucket = "/tedahro/"
-    elif "hwapi" in WACLOUD_API_ENDPOINT:
-        bucket = "/hro/"
-    elif "jincao" in WACLOUD_API_ENDPOINT:
-        bucket = "/jchro/"
-    elif 'jincao' in WACLOUD_API_ENDPOINT:
-        bucket = '/jchro/'
-    elif 'bhgn' in WACLOUD_API_ENDPOINT:
-        bucket = '/bhgnhro/'
-    elif 'gongyuanhr' in WACLOUD_API_ENDPOINT:
-        bucket = '/szgyhro/'
-
-
-    object_path = bucket + object_path
-    data = {
-        "object_path": object_path,
-        "file_name": file_name
-    }
-
-    try:
-        res = requests.post(path, json=data, headers=headers, timeout=120)
-        if res.status_code == 401:
-            time.sleep(2)
-            res = requests.post(path, json=data, headers=headers, timeout=120)
-
-        err = handle_error(res, 200)
-        if err != '':
-            logger.error("上传失败：{file}, 原因：{reason}".format(file=object_path, reason=err))
-            if 'Internal Server Error' in err:
-                err = '数据助手sheet附表内容错误'
-            return False, err
-        return True, '任务办理成功'
-    except Exception as e:
-        logger.error(str(e))
+import requests
+import time
+import oss2
+from common.log import *
+
+logger = BotLogger(os.path.basename(__file__), logname=1).getLogger()
+
+
+def handle_error(res, normal_status_code):
+    """
+    统一处理workai平台返回的error
+    :param res: 调用接口的返回
+    :param normal_status_code: 正常返回的status code
+    :return: 返回错误信息
+    """
+    if res is None or res.status_code == normal_status_code:
+        return ''
+
+    logger.info('The res is %s' % res)
+    err = '网络或服务器繁忙！'
+    try:
+        if 'message' in res.json():
+            err = res.json()['message']
+    except Exception as e:
+        logger.error(str(e))
+
+    return err
+
+
+def get_token_by_password(username, password, tenant_id, tenant_type):
+    """
+    密码方式获取access_token
+    :param username: 系统管理员用户
+    :param password: 系统管理员密码
+    :param tenant_id: 客户在workai云平台唯一表示
+    :return: 返回access_token, refresh_token 以及error, access_token用于访问接口的令牌, refresh_token
+    用于刷新获取新的access_token, error为返回的错误信息
+    """
+    data = {
+        'grant_type': 'password',
+        'username': username,
+        'password': password,
+        'scope': 'global_access:tenant_admin,tenant:%s,tenant_type:%s' % (tenant_id, tenant_type),
+    }
+
+    path = '%s/uaa/v1/auth/tokens' % WACLOUD_API_ENDPOINT
+    try:
+        res = requests.post(path, json=data, timeout=10)
+        err = handle_error(res, 200)
+        if err != '':
+            logger.error('Could not get the token by password for user: %s, tenant: %s' % (username, tenant_id))
+            return '', '', err, None
+
+        ret = res.json()
+        company, username, play_type = '', '', ''
+        if tenant_type == '0':
+            play_type = 'HRO'
+        elif tenant_type == '1':
+            play_type = 'HR'
+
+        try:
+            company = ret['tenant']['name']
+            username = ret['user']['name']
+        except Exception as e:
+            logger.error(str(e))
+
+        return ret['access_token'], ret['refresh_token'], '', (company, username, play_type)
+    except Exception as e:
+        logger.error(str(e))
+        return '', '', '网络或服务器异常，请稍后再试！', None
+
+
+def refresh_token(rtoken, tenant_id):
+    """
+    刷新token方式获取access_token
+    :return: 返回access_token, refresh_token 以及error, access_token用于访问接口的令牌, refresh_token
+    用于刷新获取新的access_token, error为返回的错误信息
+    """
+    data = {
+        'grant_type': 'refresh_token',
+        'refresh_token': rtoken,
+        'scope': 'global_access:tenant_admin,tenant:%s' % tenant_id,
+    }
+
+    path = '%s/uaa/v1/auth/tokens' % WACLOUD_API_ENDPOINT
+    res = requests.post(path, json=data)
+
+    err = handle_error(res, 200)
+    if err != '':
+        logger.error('Could not get the token by refresh token for tenant: %s' % (tenant_id,))
+        return '', '', err
+
+    ret = res.json()
+    return ret['access_token'], ret['refresh_token'], ''
+
+
+def upload_file(ctx, local_path, tenant_type, file_name):
+    """上传附件"""
+    object_path = ""
+    if ctx is None or local_path is None:
+        return 401, object_path
+
+    headers = {'Authorization': 'Bearer %s' % ctx['access_token']}
+    path = '%s/filemeta/v2/inits' % (WACLOUD_API_ENDPOINT,)
+    if tenant_type == "0":
+        data = {"object_type": "hro", "access_type": "sdk"}
+    elif tenant_type == "1":
+        data = {"object_type": "hr", "access_type": "sdk"}
+    else:
+        return 401, '配置文件设置tenant_type的值有误'
+
+    res = None
+    try:
+        res = requests.post(path, json=data, headers=headers, timeout=120)
+        if res.status_code == 401:
+            access_token, _refresh_token, err = refresh_token(ctx['refresh_token'], ctx['tenant_id'])
+            if err != '':
+                return 401, object_path
+            ctx['access_token'] = access_token
+            ctx['refresh_token'] = _refresh_token
+            headers = {'Authorization': 'Bearer %s' % ctx['access_token']}
+            res = requests.post(path, json=data, headers=headers, timeout=120)
+
+        err = handle_error(res, 200)
+        if err != '':
+            return res.status_code, object_path
+    except Exception as e:
+        logger.error(str(e))
+        return 401, str(e)
+
+    try:
+        datas = res.json()
+        # 使用临时token中的认证信息初始化StsAuth实例。
+        # auth = oss2.Auth(datas['access_key_id'], datas['access_key_secret'])
+        auth = oss2.StsAuth(datas['access_key_id'], datas['access_key_secret'], datas['security_token'])
+        # 使用StsAuth实例初始化存储空间。
+        bucket = oss2.Bucket(auth, datas['domain'], datas['bucket'])
+        object_path = datas['object_path'] + "/" + file_name
+        result = bucket.put_object_from_file(object_path, local_path)
+        return result.status, object_path
+    except Exception as e:
+        logger.error(str(e))
+        return 403, str(e)
+
+
+def update_object_path(ctx, tenant_type, object_path, types, file_name):
+    """上传文件路径到saas平台"""
+    if tenant_type == "0" or tenant_type == "1":
+        path = '%s/payroll/v1/do-import-data?type=%s' % (WACLOUD_API_ENDPOINT, types)
+    else:
+        return False, '平台类型设置有误'
+
+    headers = {'Authorization': 'Bearer %s' % ctx['access_token']}
+    bucket = '/hro/'
+    if 'hx' in WACLOUD_API_ENDPOINT:
+        bucket = '/new-hxhro/'
+    elif "xsb" in WACLOUD_API_ENDPOINT:
+        bucket = "/xsbhro/"
+    elif "engma" in WACLOUD_API_ENDPOINT:
+        bucket = "/engmahro/"
+    elif "anbo" in WACLOUD_API_ENDPOINT:
+        bucket = "/anbohro/"
+    elif "teda" in WACLOUD_API_ENDPOINT:
+        bucket = "/tedahro/"
+    elif "hwapi" in WACLOUD_API_ENDPOINT:
+        bucket = "/hro/"
+    elif "jincao" in WACLOUD_API_ENDPOINT:
+        bucket = "/jchro/"
+    elif 'jincao' in WACLOUD_API_ENDPOINT:
+        bucket = '/jchro/'
+    elif 'bhgn' in WACLOUD_API_ENDPOINT:
+        bucket = '/bhgnhro/'
+    elif 'gongyuanhr' in WACLOUD_API_ENDPOINT:
+        bucket = '/szgyhro/'
+
+
+    object_path = bucket + object_path
+    data = {
+        "object_path": object_path,
+        "file_name": file_name
+    }
+
+    try:
+        res = requests.post(path, json=data, headers=headers, timeout=120)
+        if res.status_code == 401:
+            time.sleep(2)
+            res = requests.post(path, json=data, headers=headers, timeout=120)
+
+        err = handle_error(res, 200)
+        if err != '':
+            logger.error("上传失败：{file}, 原因：{reason}".format(file=object_path, reason=err))
+            if 'Internal Server Error' in err:
+                err = '数据助手sheet附表内容错误'
+            return False, err
+        return True, '任务办理成功'
+    except Exception as e:
+        logger.error(str(e))
         return False, '请求服务器异常：{}'.format(str(e))
         return False, '请求服务器异常：{}'.format(str(e))