def _copy_lib(): """ Copy lib file to project path """ from_path = os.path.realpath(os.path.join(EXTRACT_DIR, Constant.AI_LIB_PATH)) to_path = PYTHON_PATH CommonTools.copy_file_to_dest_path(from_path, to_path) g.logger.info('Successfully to copy lib files.')
def _copy_manager_files(self): """ Copy manager files to manager dir """ from_path = os.path.join(os.path.dirname(os.path.dirname( os.path.realpath(__file__))), Constant.AI_MANAGER_PATH) to_path = self.ai_manager_path CommonTools.copy_file_to_dest_path(from_path, to_path) g.logger.info('Successfully to copy files to manager path.')
def restore_db_file(status, db_cabin, back_up_path): """ Restore db file. """ if status: CommonTools.mkdir_with_mode(os.path.dirname(db_cabin), Constant.AUTH_COMMON_DIR_STR) CommonTools.copy_file_to_dest_path(back_up_path, db_cabin) else: g.logger.info('No need Restore db file.')
def prepare_ca_certificates(self): """ Generate server ca certificates. """ ca_root_file_path, ca_root_key_path, server_cert_path, server_key_path, agent_cert_path, \ agent_key_path = self.read_ca_cert_path() self.ca_info = self.param_dict.pop(Constant.CA_INFO) if not self.ca_info: raise Exception(Errors.PARAMETER['gauss_0201'] % 'ca root cert information') get_ca_root_cert_path = self.ca_info.get(Constant.CA_CERT_PATH) get_ca_root_key_path = self.ca_info.get(Constant.CA_KEY_PATH) get_ca_root_password = self.ca_info.get(Constant.CA_PASSWORD) if not all([ get_ca_root_cert_path, get_ca_root_key_path, get_ca_root_password ]): raise Exception(Errors.PARAMETER['gauss_0201'] % 'items info of ca root cert') # copy ca root cert and key files to path of configured g.logger.info('Start deploy ca root files.') CommonTools.remove_files([ca_root_file_path, ca_root_key_path]) CommonTools.mkdir_with_mode(os.path.dirname(ca_root_file_path), Constant.AUTH_COMMON_DIR_STR) CommonTools.copy_file_to_dest_path(get_ca_root_cert_path, ca_root_file_path) CommonTools.mkdir_with_mode(os.path.dirname(ca_root_key_path), Constant.AUTH_COMMON_DIR_STR) CommonTools.copy_file_to_dest_path(get_ca_root_key_path, ca_root_key_path) # get ssl password ssl_password = CertGenerator.get_rand_str() ca_config_path = os.path.join( os.path.dirname(os.path.realpath(__file__)), Constant.CA_CONFIG) server_ip = CommonTools.get_local_ip(ignore=True) g.logger.info('Get server ip:[%s].' % server_ip) # create server cert CertGenerator.create_ca_certificate_with_script(get_ca_root_password, ssl_password, ca_root_file_path, ca_root_key_path, ca_config_path, server_cert_path, server_key_path, server_ip, crt_type='server') g.logger.info('Successfully generate server ca certificate.') return get_ca_root_password, ssl_password, ca_root_file_path, ca_root_key_path, \ ca_config_path, agent_cert_path, agent_key_path
def deploy_module_files(self): """ Copy files to module path. """ from_path = os.path.join( os.path.dirname( os.path.dirname( os.path.dirname(os.path.dirname( os.path.realpath(__file__))))), self.module_name) to_path = os.path.realpath( os.path.join(self.install_path, self.module_name)) CommonTools.copy_file_to_dest_path(from_path, to_path) g.logger.info('Successfully to copy files to package path.')
def deploy_agent_certs(self): """ Copy file from temp dir to config path. """ file_list = self.read_ca_cert_path(agent_only=True) file_list.append(os.path.join(self.module_path, Constant.PWF_PATH)) temp_cert_list = os.listdir(TMP_CA_FILE) for dest_path in file_list: CommonTools.mkdir_with_mode(os.path.dirname(dest_path), Constant.AUTH_COMMON_DIR_STR) CommonTools.remove_files(file_list) g.logger.info('Successfully prepare the path:%s' % str(file_list)) for file_name in temp_cert_list: for dest_file in file_list: if file_name in dest_file: from_path = os.path.join(TMP_CA_FILE, file_name) CommonTools.copy_file_to_dest_path(from_path, dest_file) CommonTools.clean_dir(TMP_CA_FILE)
def backup_db_file(self): """ Backup data file. """ ad_config_path = os.path.join(self.install_path, Constant.ANOMALY_DETECTION_CONFIG_PATH) if not os.path.isfile(ad_config_path): g.logger.info('Config file not exist, can not backup db file') return False, None, None db_cabin = CommonTools.read_info_from_config_file( ad_config_path, Constant.AD_CONF_SECTION_DATABASE, Constant.AD_CONF_DATABASE_PATH, self.module_path) if os.path.isdir(db_cabin): g.logger.info('Start backup db file.') back_up_path = os.path.join(EXTRACT_DIR, os.path.basename(db_cabin)) CommonTools.copy_file_to_dest_path(db_cabin, back_up_path) return True, db_cabin, back_up_path else: g.logger.info('No need backup db file.') return False, None, None