def test_update_log_create_new(self):
bot = 'test'
user = '******'
count = COMPONENT_COUNT.copy()
count['http_action'] = 6
count['domain']['intents'] = 12
summary = {
'intents': ['Intent not added to domain'],
'config': ['Invalid component']
}
DataImporterLogProcessor.update_summary(bot, user, count, summary)
log = next(DataImporterLogProcessor.get_logs(bot))
assert log.get('intents').get('data') == ['Intent not added to domain']
assert not log.get('stories').get('data')
assert not log.get('utterances').get('data')
assert not log.get('actions')
assert not log.get('training_examples').get('data')
assert not log.get('domain').get('data')
assert log.get('config').get('data') == ['Invalid component']
assert not log.get('exception')
assert not log['is_data_uploaded']
assert log['start_timestamp']
assert log.get('end_timestamp')
assert not log.get('validation_status')
assert log['event_status'] == EVENT_STATUS.COMPLETED.value
async def trigger_data_importer(bot: Text,
user: Text,
save_data: bool,
overwrite: bool = True):
"""
Triggers data importer event which validates and imports data into kairon.
@param bot: bot id.
@param user: kairon username.
@param save_data: Flag to import data into kairon. If set to false, then only validation is run.
Otherwise, both validation and import is done.
@param overwrite: Overwrite existing data(if set to true) or append (if set to false).
@return:
"""
validation_status = 'Failure'
path = None
event_url = Utility.get_event_url("DATA_IMPORTER")
try:
if not Utility.check_empty_string(event_url):
import_flag = '--import-data' if save_data else ''
overwrite_flag = '--overwrite' if overwrite else ''
env_var = {
'BOT': bot,
'USER': user,
"IMPORT_DATA": import_flag,
"OVERWRITE": overwrite_flag
}
event_request = Utility.build_event_request(env_var)
Utility.http_request(
"POST", Utility.environment['model']['data_importer'].get(
'event_url'), None, user, event_request)
DataImporterLogProcessor.add_log(
bot, user, event_status=EVENT_STATUS.TASKSPAWNED.value)
else:
path = Utility.get_latest_file(
os.path.join('training_data', bot))
files_received = DataImporterLogProcessor.get_files_received_for_latest_event(
bot)
DataImporterLogProcessor.add_log(
bot, user, event_status=EVENT_STATUS.PARSE.value)
data_importer = DataImporter(path, bot, user, files_received,
save_data, overwrite)
DataImporterLogProcessor.add_log(
bot, user, event_status=EVENT_STATUS.VALIDATING.value)
summary, component_count = await data_importer.validate()
initiate_import = Utility.is_data_import_allowed(
summary, bot, user)
status = 'Success' if initiate_import else 'Failure'
DataImporterLogProcessor.update_summary(
bot,
user,
component_count,
summary,
status=status,
event_status=EVENT_STATUS.SAVE.value)
if initiate_import:
data_importer.import_data()
DataImporterLogProcessor.add_log(
bot, user, event_status=EVENT_STATUS.COMPLETED.value)
except exceptions.ConnectionError as e:
logger.error(str(e))
DataImporterLogProcessor.add_log(
bot,
user,
exception=f'Failed to trigger the event. {e}',
status=validation_status,
event_status=EVENT_STATUS.FAIL.value)
except Exception as e:
logger.error(str(e))
DataImporterLogProcessor.add_log(
bot,
user,
exception=str(e),
status=validation_status,
event_status=EVENT_STATUS.FAIL.value)
if path:
Utility.delete_directory(path)