def upload_file_to_region_center(self, team_name, user_name, region, upload_file):
url, token = region_services.get_region_access_info(team_name, region)
headers = {"Authorization": token}
logger.debug("request header : {0}".format(headers))
files = {'appTarFile': upload_file}
event_id = make_uuid()
import_record_params = {
"event_id": event_id,
"status": "uploading",
"team_name": team_name,
"region": region,
"user_name": user_name
}
import_record = app_import_record_repo.create_app_import_record(**import_record_params)
data = {"eventId": event_id}
url += "/v2/app/upload"
logger.debug("upload url : {0}".format(url))
response = requests.post(url, data=data, files=files, headers=headers, verify=False)
if response.status_code == 200:
logger.debug("file upload success !")
import_record.status = "upload_success"
import_record.save()
upload_file.close()
return 200, "上传成功", import_record
else:
logger.debug("file upload failed !")
import_record.delete()
upload_file.close()
return 500, "上传失败", None
def upload_file_to_region_center_by_enterprise_id(self, enterprise_id, user_name, upload_file):
rst_list = []
regions = region_services.get_regions_by_enterprise_id(enterprise_id)
for region in regions:
url, token = region_services.get_region_access_info_by_enterprise_id(enterprise_id, region.region_name)
headers = {"Authorization": token}
logger.debug("request header : {0}".format(headers))
files = {'appTarFile': upload_file}
event_id = make_uuid()
import_record_params = {
"event_id": event_id,
"status": "uploading",
# "team_name": team_name,
"region": region.region_name,
"user_name": user_name
}
import_record = app_import_record_repo.create_app_import_record(**import_record_params)
data = {"eventId": event_id}
url += "/v2/app/upload"
logger.debug("upload url : {0}".format(url))
response = requests.post(url, data=data, files=files, headers=headers, verify=False)
if response.status_code == 200:
logger.debug("file upload success !")
import_record.status = "upload_success"
import_record.save()
upload_file.close()
rst_list.append({"status": 200, "msg": "上传成功", "data": import_record, "region": region.region_name})
else:
logger.debug("file upload failed !")
import_record.delete()
upload_file.close()
rst_list.append({"status": 500, "msg": "上传失败", "data": None, "region": region.region_name})
return rst_list
def create_import_app_dir(self, tenant, region):
"""创建一个应用包"""
event_id = make_uuid()
res, body = region_api.create_import_file_dir(region, tenant.tenant_name, event_id)
path = body["bean"]["path"]
import_record_params = {"event_id": event_id, "status": "created_dir", "source_dir": path,
"team_name": tenant.tenant_name, "region": region}
import_record = app_import_record_repo.create_app_import_record(**import_record_params)
return import_record
def create_app_import_record(self, team_name, user_name, region):
event_id = make_uuid()
import_record_params = {
"event_id": event_id,
"status": "uploading",
"team_name": team_name,
"region": region,
"user_name": user_name
}
return app_import_record_repo.create_app_import_record(**import_record_params)
def create_app_import_record(self, team_name, user_name, region):
event_id = make_uuid()
import_record_params = {
"event_id": event_id,
"status": "uploading",
"team_name": team_name,
"region": region,
"user_name": user_name,
"source_dir": "/grdata/app/import/{0}".format(event_id)
}
return app_import_record_repo.create_app_import_record(
**import_record_params)
def create_app_import_record_2_enterprise(self, eid, user_name):
event_id = make_uuid()
region = self.select_handle_region(eid)
import_record_params = {
"event_id": event_id,
"status": "uploading",
"enterprise_id": eid,
"region": region.region_name,
"user_name": user_name
}
return app_import_record_repo.create_app_import_record(
**import_record_params)
def create_app_import_record_2_enterprise(self, eid, user_name):
event_id = make_uuid()
try:
region = region_repo.get_region_by_enterprise_id(eid)
except TenantRegionInfo.DoesNotExist:
raise RegionNotFound("region not found")
if not region:
raise RegionNotFound("region not found")
import_record_params = {
"event_id": event_id,
"status": "uploading",
"enterprise_id": eid,
"region": region.region_name,
"user_name": user_name
}
return app_import_record_repo.create_app_import_record(**import_record_params)
