def delete_algorithm_model(identifier, request_id, params):
response = AlgorithmModelDeployResultEvent(
AlgorithmEventErrorCodes.ALGO_EVENT_SUCCESS,
AlgorithmDeployEventType.ALGO_DELETE_ALGO_MODEL)
if "AlgoID" not in params.keys():
response.Result = AlgorithmEventErrorCodes.ALGO_EVENT_PARAM_ERR
return response.to_json()
algo_id = params["AlgoID"]
"""判断该algo_id是否有任务在运行"""
response.AlgoID = algo_id
if algo_id in dict_cloud_task_id_algo_id.values():
response.Result = AlgorithmEventErrorCodes.ALGO_EVENT_DELETE_ERR
return response.to_json()
module_path = conf.get_string("Default", "algoModuleDir")
algo_path = module_path + "/" + algo_id + "/"
if os.path.exists(algo_path):
shutil.rmtree(algo_path)
else:
response.Result = AlgorithmEventErrorCodes.ALGO_EVENT_INVALID_ALGO_ID_ERR
return response.to_json()
return response.to_json()
def __init__(self):
if "ACCESS_KEY_ID" in os.environ:
self.access_key_id = os.environ["ACCESS_KEY_ID"]
else:
self.access_key_id = conf.get_string("OSS_CFG", "access_key_id")
if "ACCESS_KEY_SECRET" in os.environ:
self.access_key_secret = os.environ["ACCESS_KEY_SECRET"]
else:
self.access_key_secret = conf.get_string("OSS_CFG",
"access_key_secret")
if "BUCKET_NAME" in os.environ:
self.bucket_name = os.environ["BUCKET_NAME"]
else:
self.bucket_name = conf.get_string("OSS_CFG", "bucket_name")
if "ENDPOINT" in os.environ:
self.endpoint = os.environ["ENDPOINT"]
else:
self.endpoint = conf.get_string("OSS_CFG", "endpoint")
self.bucket = oss2.Bucket(
oss2.Auth(self.access_key_id, self.access_key_secret),
self.endpoint, self.bucket_name)
return
def download_model(identifier, request_id, params):
"""
:param identifier:
:param request_id:
:param params:
:return:
"""
log.info("download_model params:%s" % params)
response = AlgorithmModelDeployResultEvent(
AlgorithmEventErrorCodes.ALGO_EVENT_SUCCESS,
AlgorithmDeployEventType.ALGO_DEPLOY_ALGO_MODEL)
if "AlgoURL" not in params.keys() or "AlgoID" not in params.keys():
response.Result = AlgorithmEventErrorCodes.ALGO_EVENT_PARAM_ERR
return response.to_json()
url = params["AlgoURL"]
algo_id = params["AlgoID"]
response.AlgoID = algo_id
module_path = conf.get_string("Default", "algoModuleDir")
algo_path = module_path + "/" + algo_id + "/"
if os.path.isdir(algo_path) is False:
os.mkdir(algo_path)
algo_name = os.path.join(module_path, "temp_" + algo_id)
try:
urllib.request.urlretrieve(url, algo_name, schedule_hook)
file_zip = zipfile.ZipFile(algo_name, 'r')
for file in file_zip.namelist():
file_zip.extract(file, algo_path)
file_zip.close()
os.remove(algo_name)
except Exception as e:
response.Result = AlgorithmEventErrorCodes.ALGO_EVENT_DOWNLOAD_ERR
if os.path.exists(algo_name):
os.remove(algo_name)
return response.to_json()
return response.to_json()
def __init__(self):
self.path = conf.get_string("Default", "algoModuleDir")
self.dict_modules = {}
self.dict_single_modules = {}
# self.dict_module_info = {}
return
code,
str(data),
message))
def on_thing_event_post(event, request_id, code, data, reply_message, user_data):
log.info("on_thing_event_post event: {} request_id:{}, code:{}, data:{}, reply_message:{}".format(event,
request_id,
code,
str(data),
reply_message))
if "LINKKIT_LOADED" not in os.environ:
os.environ.setdefault("LINKKIT_LOADED", "TRUE")
host_name = conf.get_string("LinkKit", "host_name")
if "PRODUCT_KEY" in os.environ:
product_key = os.environ["PRODUCT_KEY"]
else:
product_key = conf.get_string("LinkKit", "product_key")
log.info("product_key={}".format(product_key))
if "DEVICE_NAME" in os.environ:
device_name = os.environ["DEVICE_NAME"]
else:
device_name = conf.get_string("LinkKit", "device_name")
log.info("device_name={}".format(device_name))
if "DEVICE_SECRET" in os.environ:
device_secret = os.environ["DEVICE_SECRET"]
else:
def query_algo_model_list():
module_path = conf.get_string("Default", "algoModuleDir")
return os.listdir(module_path)