def test_build_event_request(self):
request = {'BOT': 'mood_bot', "USER": "******"}
request_body = Utility.build_event_request(request)
assert isinstance(request_body, list)
assert request_body[0]['name'] == 'BOT'
assert request_body[0]['value'] == 'mood_bot'
assert request_body[0]['name'] == 'USER'
assert request_body[0]['value'] == 'bot_user'
assert len(request_body) == 2
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
try:
if Utility.get_event_url("DATA_IMPORTER"):
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()
is_data_valid = all(
[not summary[key] for key in summary.keys()])
validation_status = 'Success' if is_data_valid else 'Failure'
DataImporterLogProcessor.update_summary(
bot,
user,
component_count,
summary,
status=validation_status,
event_status=EVENT_STATUS.SAVE.value)
if is_data_valid:
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='Failed to trigger the event.',
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)
def test_build_event_request_empty(self):
request_body = Utility.build_event_request({})
assert isinstance(request_body, list)
assert not request_body
