async def validate_training_data( background_tasks: BackgroundTasks, current_user: User = Depends(auth.get_current_user_and_bot), ): """ Validates bot training data. """ DataImporterLogProcessor.is_limit_exceeded(current_user.get_bot()) DataImporterLogProcessor.is_event_in_progress(current_user.get_bot()) Utility.make_dirs( os.path.join("training_data", current_user.get_bot(), str(datetime.utcnow()))) DataImporterLogProcessor.add_log(current_user.get_bot(), current_user.get_user(), is_data_uploaded=False) background_tasks.add_task(EventsTrigger.trigger_data_importer, current_user.get_bot(), current_user.get_user(), False, False) return {"message": "Event triggered! Check logs."}
def test_is_limit_exceeded_false(self, monkeypatch): monkeypatch.setitem(Utility.environment['model']['data_importer'], "limit_per_day", 6) bot = 'test' assert not DataImporterLogProcessor.is_limit_exceeded(bot)
def test_is_limit_exceeded_exception(self, monkeypatch): monkeypatch.setitem(Utility.environment['model']['data_importer'], "limit_per_day", 3) bot = 'test' with pytest.raises(AppException): assert DataImporterLogProcessor.is_limit_exceeded(bot)